mlsanityData to 7 Oct 2026

Tracing distinguishability through transformer processing with stochastic LayerNorm

arXiv 2608.30720 ↗First posted 31 Aug 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Representational similarity is foundational to analyses of deep networks, yet distances between point-valued representations are not intrinsically tied to downstream function: nearby states may produce different behaviors, while distant states may behave similarly. We instead give representations volume, turning similarity into statistical distinguishability. Overlapping stochastic representations necessarily induce overlapping downstream distributions, grounding latent comparison in model function and bringing it under information-theoretic tools such as the data-processing inequality. We realize this idea in pretrained transformers through a light-touch modification to LayerNorm: at each residual-stream read, we normalize the state, add isotropic Gaussian noise, and renormalize. During distillation fine-tuning, one learned allocation parameter per residual-stream read distributes a fixed global rate budget across the processing stack. The resulting model can be viewed as transformer blocks reading the residual stream with learned finite precision under a shared global rate budget. Using the Bhattacharyya coefficient, we trace which counterfactual distinctions are preserved through MLP blocks or selectively exposed to the query, key, and value computations of individual attention heads. Experiments on ViT-S and GPT-2 small reveal the depthwise propagation of continuous visual perturbations and head-specific sensitivity to token distinctions aligned with known attention motifs. These results establish distinguishability as a functionally grounded lens on transformer computation that complements existing interpretability approaches.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 11 of its 26 references that are in the map. First posted on arXiv 31 Aug 2026.

Builds on

11 of 26 references.

References · 11

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  2. 2016Layer Normalizationcited by 5,579
  3. 2013Auto-Encoding Variational Bayescited by 3,980
  4. 2019Similarity of Neural Network Representations Revisitedcited by 1,695
  5. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
  6. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
  7. 2022In-context Learning and Induction Headscited by 930
  8. 2016Deep Variational Information Bottleneckcited by 915
  9. 2022Reliability of CKA as a Similarity Measure in Deep Learningcited by 82
  10. 2022Deconfounded Representation Similarity for Comparison of Neural Networkscited by 18
  11. 2026From independent patches to coordinated attention: Controlling information flow in vision transformerscited by 1

Built on by

Not cited yet.