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