mlsanityData to 14 Oct 2026

The underlying structures of self-attention: symmetry, directionality, and emergent dynamics in Transformer training

arXiv 2502.10927 ↗First posted 15 Feb 2025cs.LGcited by 9

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  4. 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
  5. 2022Robust Speech Recognition via Large-Scale Weak Supervisioncited by 4,057
  6. 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,884
  7. 2019XLNet: Generalized Autoregressive Pretraining for Language Understandingcited by 2,811
  8. 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,790
  9. 2022Locating and Editing Factual Associations in GPTcited by 2,691
  10. 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
  11. 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
  12. 2024Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phonecited by 1,917
21 more
  1. 2020mT5: A massively multilingual pre-trained text-to-text transformercited by 1,818
  2. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
  3. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  4. 2019Is BERT Really Robust? A Strong Baseline for Natural Language Attack on Text Classification and Entailmentcited by 671
  5. 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 623
  6. 2024Smarter, Better, Faster, Longer: A Modern Bidirectional Encoder for Fast, Memory Efficient, and Long Context Finetuning and Inferencecited by 554
  7. 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 532
  8. 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 449
  9. 2023A Survey on Large Language Model (LLM) Security and Privacy: The Good, the Bad, and the Uglycited by 387
  10. 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 334
  11. 2023Localizing Model Behavior with Path Patchingcited by 195
  12. 2023Does Circuit Analysis Interpretability Scale? Evidence from Multiple Choice Capabilities in Chinchillacited by 149
  13. 2022Analyzing Transformers in Embedding Spacecited by 132
  14. 2018What Is One Grain of Sand in the Desert? Analyzing Individual Neurons in Deep NLP Modelscited by 124
  15. 2024A Practical Review of Mechanistic Interpretability for Transformer-Based Language Modelscited by 118
  16. 2020Perturbed Masking: Parameter-free Probing for Analyzing and Interpreting BERTcited by 109
  17. 2022A Rigorous Study of Integrated Gradients Method and Extensions to Internal Neuron Attributionscited by 64
  18. 2023Mimetic Initialization of Self-Attention Layerscited by 54
  19. 2023AttentionViz: A Global View of Transformer Attentioncited by 50
  20. 2022Grad-SAM: Explaining Transformers via Gradient Self-Attention Mapscited by 42
  21. 2022SemAttack: Natural Textual Attacks via Different Semantic Spacescited by 42

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 2; 2026: 7.

9 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025From Condensation to Rank Collapse: A Two-Stage Analysis of Transformer Training Dynamics
  2. 2024Self-Attention Limits Working Memory Capacity of Transformer-Based Models
  3. 2026Symmetry Breaking in Transformers for Efficient and Interpretable Training
  4. 2025Sigmoid Self-Attention has Lower Sample Complexity than Softmax Self-Attention: A Mixture-of-Experts Perspective
  5. 2023Rethinking Attention: Exploring Shallow Feed-Forward Neural Networks as an Alternative to Attention Layers in Transformers
  6. 2025Dynamical Properties of Tokens in Self-Attention and Effects of Positional Encoding
  7. 2024The Asymptotic Behavior of Attention in Transformers
  8. 2024An Evolved Universal Transformer Memory