acceptodds
Under review as a conference paper at ICLR 2027

Recovering Clean Representation Similarity with Independent Replicas

Abstract

Sampling debiasing does not necessarily remove the effect of observation noise on centered kernel alignment (CKA). Even when noise leaves the cross-model HSIC numerator unbiased, it can inflate the two self factors and distort both similarity scores and layer rankings. We introduce Independent-Replica CKA (IR-CKA), which pairs Gram matrices from two conditionally independent observations inside each minibatch factor and averages the factors before normalization. Under the stated mean-preserving noise assumptions, all three factors are unbiased for their clean finite-pool counterparts. Comparisons with feature averaging and -replica cross-Grams identify the corrections retained when independence is used elsewhere in the construction. We establish concentration under random partitions of a fixed input pool and mean-square error for PSD-stabilized IR with ridge , under moment and positive self-factor margin conditions. Experiments with frozen ViT and ResNet representations show that IR and its variants recover clean similarity matrices and bidirectional layer relationships more accurately than shared-replica, feature-averaging, and cross-Gram baselines under moderate feature dropout. Under stronger corruption, PSD stabilization restores complete estimates and improves several retrieval measures, although accurate recovery is not uniform across layers or metrics. These results demonstrate the practical value of placing replica independence within all three HSIC factors for recovering clean representation similarity from noisy observations.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.