Which Question Is Your Attention Metric Answering? Attention Rows as Compositional Data
Not on the map. Why?
Abstract (arXiv)
Each row of a transformer's attention matrix is a probability distribution over tokens, and in trained models most of that probability lands on a single sink token, usually the first. Standard tools for comparing attention rows (cosine similarity, Jensen--Shannon divergence, Shannon entropy) therefore hinge on a choice papers rarely report: keep the sink, or drop it and renormalize. This choice can reverse conclusions. On ten pretrained models from five families, 17--47% of verdicts about which of two heads is more similar flip with the convention, and the most prominent structure in a standard BERT head-clustering pipeline is an artifact of it. The reason is that one-number summaries mix two questions: how much attention the sink takes, and how the rest is divided among the content tokens. Treating rows as compositional data separates them exactly: the Aitchison distance splits orthogonally into a sink term and a content term, entropy splits by an exact identity, and the content distance is characterized by invariances the transformer itself possesses. The separation matters in practice: most measured entropy collapse during training is the sink growing, not attention sharpening (30% of the drop at 70M parameters, 95% at 1B, 79% at 1.4B), and pruning heads with the wrong channel can inflate perplexity more than a hundredfold. We map where each convention is safe, test a frozen out-of-sample predictor (one confirmation, one abstention, one failure), and release code regenerating every number.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 16 of its 27 references that are in the map. First posted on arXiv 11 Aug 2026.
Builds on
16 of 27 references.
References · 16
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
- 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 994
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
- 2023Vision Transformers Need Registerscited by 844
- 2019Are Sixteen Heads Really Better than One?cited by 803
- 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 411
- 2019Revealing the Dark Secrets of BERTcited by 300
- 2024Massive Activations in Large Language Modelscited by 277
- 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 200
- 2023Stabilizing Transformer Training by Preventing Attention Entropy Collapsecited by 154
- 2025Why do LLMs attend to the first token?cited by 122
- 2020DeepCoDA: personalized interpretability for compositional health datacited by 7
4 more
Built on by
Citing papers per year: 2026: 1.
1 paper.