mlsanityData to 7 Oct 2026

A Unified Framework for Critical Scaling of Inverse Temperature in Self-Attention

arXiv 2605.12697 ↗First posted 12 May 2026stat.ML cs.LG math.PRcited by 0

Not on the map. Why?

Abstract (arXiv)

Length-dependent logit rescaling is widely used to stabilize long-context self-attention, but existing analyses and methods suggest conflicting inverse-temperature laws for the context length n, ranging from (log n)1/2 to log n and (log n)2. We provide a general theory showing that the desirable scale is determined by the gap-counting function Nn of each attention row. Counting how many competitors lie within each gap from the maximum, we define an upper-tail accumulation scale and prove that it gives the critical inverse-temperature scale for softmax concentration: below this scale, the top competitors remain unseparated, whereas above it, the attention entropy collapses. This framework unifies prior scaling laws as different Nn and yields a direct diagnostic for attention-score families, from idealized theoretical models to more practical transformers.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 26 of its 32 references that are in the map. First posted on arXiv 12 May 2026.

Builds on

26 of 32 references.

References · 26

  1. 2017Attention Is All You Needcited by 59,932
  2. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  3. 2023Qwen Technical Reportcited by 3,273
  4. 2019Compressive Transformers for Long-Range Sequence Modellingcited by 699
  5. 2023YaRN: Efficient Context Window Extension of Large Language Modelscited by 584
  6. 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 411
  7. 2023Llemma: An Open Language Model For Mathematicscited by 385
  8. 2023A mathematical perspective on Transformerscited by 160
  9. 2022Signal Propagation in Transformers: Theoretical Perspectives and the Role of Rank Collapsecited by 136
  10. 2023The emergence of clusters in self-attention dynamicscited by 133
  11. 2020Infinite attention: NNGP and NTK for deep attention networkscited by 117
  12. 2023How Do Transformers Learn Topic Structure: Towards a Mechanistic Understandingcited by 85
14 more
  1. 2024Dynamic metastability in the self-attention modelcited by 42
  2. 2024Infinite Limits of Multi-head Transformer Dynamicscited by 38
  3. 2025Scalable-Softmax Is Superior for Attentioncited by 38
  4. 2024Clustering in Causal Attention Maskingcited by 31
  5. 2024Geometric Dynamics of Signal Propagation Predict Trainability of Transformerscited by 24
  6. 2025A multiscale analysis of mean-field transformers in the moderate interaction regimecited by 24
  7. 2025The Mean-Field Dynamics of Transformerscited by 24
  8. 2024How transformers learn structured data: insights from hierarchical filteringcited by 18
  9. 2025Critical attention scaling in long-context transformerscited by 17
  10. 2025Two failure modes of deep transformers and how to avoid them: a unified theory of signal propagation at initialisationcited by 15
  11. 2025Asymptotics of SGD in Sequence-Single Index Models and Single-Layer Attention Networkscited by 9
  12. 2025Scale-invariant Attentioncited by 5
  13. 2025Infinite-Width Limit of a Single Attention Layer: Analysis via Tensor Programscited by 5
  14. 2025Recurrent Self-Attention Dynamics: An Energy-Agnostic Perspective from Jacobianscited by 4

Built on by

Not cited yet.