mlsanityData to 7 Oct 2026

Which Question Is Your Attention Metric Answering? Attention Rows as Compositional Data

arXiv 2608.14712 ↗First posted 11 Aug 2026cs.CL cs.AI cs.LGcited by 1

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

  1. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
  2. 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 994
  3. 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
  4. 2023Vision Transformers Need Registerscited by 844
  5. 2019Are Sixteen Heads Really Better than One?cited by 803
  6. 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 411
  7. 2019Revealing the Dark Secrets of BERTcited by 300
  8. 2024Massive Activations in Large Language Modelscited by 277
  9. 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 200
  10. 2023Stabilizing Transformer Training by Preventing Attention Entropy Collapsecited by 154
  11. 2025Why do LLMs attend to the first token?cited by 122
  12. 2020DeepCoDA: personalized interpretability for compositional health datacited by 7
4 more
  1. 2022Aitchison's Compositional Data Analysis 40 Years On: A Reappraisalcited by 6
  2. 2026On the Invariants of Softmax Attentioncited by 3
  3. 2026Aitchison Embeddings for Learning Compositional Graph Representationscited by 2
  4. 2026Tree-Structured Orthonormal Decomposition of the Aitchison Simplexcited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.