Quantifying LLM Attention-Head Stability: Implications for Circuit Universality
Not on the map. Why?
Abstract (arXiv)
In mechanistic interpretability, recent work scrutinizes transformer "circuits" - sparse, mono or multi layer sub computations, that may reflect human understandable functions. Yet, these network circuits are rarely acid-tested for their stability across different instances of the same deep learning architecture. Without this, it remains unclear whether reported circuits emerge universally across labs or turn out to be idiosyncratic to a particular estimation instance, potentially limiting confidence in safety-critical settings. Here, we systematically study stability across-refits in increasingly complex transformer language models of various sizes. We quantify, layer by layer, how similarly attention heads learn representations across independently initialized training runs. Our rigorous experiments show that (1) middle-layer heads are the least stable yet the most representationally distinct; (2) deeper models exhibit stronger mid-depth divergence; (3) unstable heads in deeper layers become more functionally important than their peers from the same layer; (4) applying weight decay optimization substantially improves attention-head stability across random model initializations; and (5) the residual stream is comparatively stable. Our findings establish the cross-instance robustness of circuits as an essential yet underappreciated prerequisite for scalable oversight, drawing contours around possible white-box monitorability of AI systems.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 24 of its 41 references that are in the map. First posted on arXiv 17 Feb 2026.
Builds on
24 of 41 references.
References · 24
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2019Similarity of Neural Network Representations Revisitedcited by 1,695
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
- 2020On Layer Normalization in the Transformer Architecturecited by 906
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
- 2024Sparse Feature Circuits: Discovering and Editing Interpretable Causal Graphs in Language Modelscited by 412
- 2014The Loss Surfaces of Multilayer Networkscited by 278
- 2023Small-scale proxies for large-scale Transformer training instabilitiescited by 182
12 more
- 2022DeepNet: Scaling Transformers to 1,000 Layerscited by 176
- 2023A Toy Model of Universality: Reverse Engineering How Networks Learn Group Operationscited by 141
- 2018Three Mechanisms of Weight Decay Regularizationcited by 112
- 2021The MultiBERTs: BERT Reproductions for Robustness Analysiscited by 89
- 2022Reliability of CKA as a Similarity Measure in Deep Learningcited by 82
- 2024Disentangling the Causes of Plasticity Loss in Neural Networkscited by 82
- 2023Why Do We Need Weight Decay in Modern Deep Learning?cited by 70
- 2021Which Shortcut Cues Will DNNs Choose? A Study from the Parameter-Space Perspectivecited by 56
- 2023Analyzing Feed-Forward Blocks in Transformers through the Lens of Attention Mapscited by 33
- 2025PolyPythias: Stability and Outliers across Fifty Language Model Pre-Training Runscited by 33
- 2023Subspace Chronicles: How Linguistic Information Emerges, Shifts and Interacts during Language Model Trainingcited by 23
- 2022B2T Connection: Serving Stability and Performance in Deep Transformerscited by 16
Built on by
Citing papers per year: 2026: 2.
2 papers.