On the Invariants of Softmax Attention
Not on the map. Why?
Abstract (arXiv)
Softmax attention maps every query--key interaction into a probability distribution, but the underlying structure remains largely unexplored. We define the energy field, the row-centered attention logit, and show that it exhibits invariant properties across models, architectures, and inputs. Two classes of invariants emerge. Mechanism-level invariants follow from the algebraic structure of softmax attention. They include a per-row zero-sum constraint, a rank bound determined by the head dimension, and spectral signatures that follow from them. Model-level regularities are not required by the mechanism, yet hold in every autoregressive language model we test, spanning several architecture families. The energy field distributes its variance over key positions without concentrating at a few. This delocalization traces to a property of the key matrix we call key incoherence. These invariants have practical consequences. The rank bound confines the energy field to a low-dimensional subspace. Key incoherence yields a per-head training monitor. All results are verified at multiple context lengths and input texts.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 11 of its 17 references that are in the map. First posted on arXiv 6 Apr 2026.
Builds on
11 of 17 references.
References · 11
- 2017Attention Is All You Needcited by 60,361
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
- 2023Mistral 7Bcited by 3,164
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
- 2024Qwen2 Technical Reportcited by 2,120
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2020Rethinking Attention with Performerscited by 1,601
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,346
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Which Question Is Your Attention Metric Answering? Attention Rows as Compositional Datacited by 1
- 2026Intrinsic Interaction Geometry Controls the Low-Rank Complexity of Softmax Attentioncited by 1
- 2026What Does Chain-of-Thought Entropy Measure? A Channel Audit of Scaffolding, Routing, and Contentcited by 0
- 2026Intrinsic Interaction Geometry Controls the Low-Rank Complexity of Softmax Attention28 Aug 2026
- 2026What Does Chain-of-Thought Entropy Measure? A Channel Audit of Scaffolding, Routing, and Content26 Aug 2026
- 2026Which Question Is Your Attention Metric Answering? Attention Rows as Compositional Data11 Aug 2026