The underlying structures of self-attention: symmetry, directionality, and emergent dynamics in Transformer training
Not on the map. Why?
Abstract (arXiv)
Self-attention is essential to Transformer architectures, yet how information is embedded in the self-attention matrices and how different objective functions impact this process remains unclear. We present a mathematical framework to analyze self-attention matrices by deriving the structures governing their weight updates. Using this framework, we demonstrate that bidirectional training induces symmetry in the weight matrices, while autoregressive training results in directionality and column dominance. Our theoretical findings are validated across multiple Transformer models - including ModernBERT, GPT, LLaMA3, and Mistral - and input modalities like text, vision, and audio. Finally, we apply these insights by showing that symmetric initialization improves the performance of encoder-only models on language tasks. This mathematical analysis offers a novel theoretical perspective on how information is embedded through self-attention, thereby improving the interpretability of Transformer models.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 33 of its 66 references that are in the map. First posted on arXiv 15 Feb 2025.
Builds on
33 of 66 references.
References · 33
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
- 2022Robust Speech Recognition via Large-Scale Weak Supervisioncited by 4,057
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,884
- 2019XLNet: Generalized Autoregressive Pretraining for Language Understandingcited by 2,811
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,790
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
- 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
- 2024Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phonecited by 1,917
21 more
- 2020mT5: A massively multilingual pre-trained text-to-text transformercited by 1,818
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2019Is BERT Really Robust? A Strong Baseline for Natural Language Attack on Text Classification and Entailmentcited by 671
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 623
- 2024Smarter, Better, Faster, Longer: A Modern Bidirectional Encoder for Fast, Memory Efficient, and Long Context Finetuning and Inferencecited by 554
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 532
- 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 449
- 2023A Survey on Large Language Model (LLM) Security and Privacy: The Good, the Bad, and the Uglycited by 387
- 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 334
- 2023Localizing Model Behavior with Path Patchingcited by 195
- 2023Does Circuit Analysis Interpretability Scale? Evidence from Multiple Choice Capabilities in Chinchillacited by 149
- 2022Analyzing Transformers in Embedding Spacecited by 132
- 2018What Is One Grain of Sand in the Desert? Analyzing Individual Neurons in Deep NLP Modelscited by 124
- 2024A Practical Review of Mechanistic Interpretability for Transformer-Based Language Modelscited by 118
- 2020Perturbed Masking: Parameter-free Probing for Analyzing and Interpreting BERTcited by 109
- 2022A Rigorous Study of Integrated Gradients Method and Extensions to Internal Neuron Attributionscited by 64
- 2023Mimetic Initialization of Self-Attention Layerscited by 54
- 2023AttentionViz: A Global View of Transformer Attentioncited by 50
- 2022Grad-SAM: Explaining Transformers via Gradient Self-Attention Mapscited by 42
- 2022SemAttack: Natural Textual Attacks via Different Semantic Spacescited by 42
Built on by
Citing papers per year: 2025: 2; 2026: 7.
9 papers.
- 2026Uncovering the Latent Potential of Deep Intermediate Representationscited by 3
- 2025GateTS: Versatile and Efficient Forecasting via Attention-Inspired routed Mixture-of-Expertscited by 1
- 2026Pseudo-Unification: Entropy Probing Reveals Divergent Information Patterns in Unified Multimodal Modelscited by 1
- 2025Reversal Invariance in Autoregressive Language Modelscited by 0
- 2026Self-Attention as Transport: Limits of Symmetric Spectral Diagnosticscited by 0
- 2026Fingerprint, Not Blueprint: How Positional Schemes Set the Default Spectral Algebra of Attentioncited by 0
- 2026Task- and dataset-specific information in protein language modelscited by 0
- 2026On attention heads and bilinear formscited by 0
- 2026A Mesoscopic View of Transformer Weights Through Row and Column Scale Fieldscited by 0
- 2026A Mesoscopic View of Transformer Weights Through Row and Column Scale Fields25 Sep 2026
- 2026On attention heads and bilinear forms19 Sep 2026
- 2026Task- and dataset-specific information in protein language models12 Aug 2026
- 2026Fingerprint, Not Blueprint: How Positional Schemes Set the Default Spectral Algebra of Attention7 Jul 2026
- 2026Uncovering the Latent Potential of Deep Intermediate Representations21 May 2026
- 2026Self-Attention as Transport: Limits of Symmetric Spectral Diagnostics6 May 2026
- 2026Pseudo-Unification: Entropy Probing Reveals Divergent Information Patterns in Unified Multimodal Models13 Apr 2026
- 2025Reversal Invariance in Autoregressive Language Models1 Nov 2025
- 2025GateTS: Versatile and Efficient Forecasting via Attention-Inspired routed Mixture-of-Experts24 Aug 2025
Similar, not cited
Close in wording, no citation either way
- 2025From Condensation to Rank Collapse: A Two-Stage Analysis of Transformer Training Dynamics
- 2024Self-Attention Limits Working Memory Capacity of Transformer-Based Models
- 2026Symmetry Breaking in Transformers for Efficient and Interpretable Training
- 2025Sigmoid Self-Attention has Lower Sample Complexity than Softmax Self-Attention: A Mixture-of-Experts Perspective
- 2023Rethinking Attention: Exploring Shallow Feed-Forward Neural Networks as an Alternative to Attention Layers in Transformers
- 2025Dynamical Properties of Tokens in Self-Attention and Effects of Positional Encoding
- 2024The Asymptotic Behavior of Attention in Transformers
- 2024An Evolved Universal Transformer Memory