English

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

Machine Learning 2026-05-14 v1 Machine Learning Probability

Abstract

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 nn, ranging from (logn)1/2(\log n)^{1/2} to logn\log n and (logn)2(\log n)^2. We provide a general theory showing that the desirable scale is determined by the gap-counting function NnN_n 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 NnN_n and yields a direct diagnostic for attention-score families, from idealized theoretical models to more practical transformers.

Keywords

Cite

@article{arxiv.2605.12697,
  title  = {A Unified Framework for Critical Scaling of Inverse Temperature in Self-Attention},
  author = {Tomohiro Hayase and Ryo Karakida},
  journal= {arXiv preprint arXiv:2605.12697},
  year   = {2026}
}