How Does Attention Help? Insights from Random Matrices on Signal Recovery from Sequence Models
Not on the map. Why?
Abstract (arXiv)
We study the spectral properties of sample covariance matrices constructed from pooled sequence representations, where token embeddings are drawn from a fixed two-class Gaussian mixture table and pooled via (fixed) attention weights. Working in the high-dimensional regime d,V,N→∞ with d/V→δ and d/N→γ, we derive exact characterizations of the limiting eigenvalue distribution, outlier eigenvalues, and eigenvector alignment with the hidden signal. The bulk spectrum follows a non-Marchenko--Pastur law given by the free multiplicative convolution kappa(MPδboxtimes MPγ), reflecting the finite vocabulary structure. Signal recovery undergoes two successive BBP-type phase transitions characterized by the scalars: δ,γ,α=wtop R w and kappa=|w|2, where w denotes the attention pooling weights and R the positional correlation matrix. An aftermath of our analysis demonstrates that the optimal attention weights maximizing the signal-to-noise ratio α/kappa are given by the (normalized) top eigenvector of R, and we show (as a particular case of our analysis) that parameter-free causal self-attention with τ/d score scaling yields deterministic harmonic weights that improve signal recovery over mean pooling whenever early tokens carry more signal. Extensive simulations confirm sharp agreement between theory and finite-dimensional experiments.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 19 of its 36 references that are in the map. First posted on arXiv 7 May 2026.
Builds on
19 of 36 references.
References · 19
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2019How Contextual are Contextualized Word Representations? Comparing the Geometry of BERT, ELMo, and GPT-2 Embeddingscited by 784
- 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
- 2019Are Transformers universal approximators of sequence-to-sequence functions?cited by 406
- 2019Representation Degeneration Problem in Training Natural Language Generation Modelscited by 255
- 2017All-but-the-Top: Simple and Effective Postprocessing for Word Representationscited by 212
- 2021Inductive Biases and Variable Creation in Self-Attention Mechanismscited by 144
- 2021Learning curves of generic features maps for realistic datasets with a teacher-student modelcited by 126
- 2020Infinite attention: NNGP and NTK for deep attention networkscited by 121
7 more
- 2020Random Matrix Theory Proves that Deep Learning Representations of GAN-data Behave as Gaussian Mixturescited by 47
- 2022Unraveling Attention via Convex Duality: Analysis and Interpretations of Vision Transformerscited by 30
- 2025Statistical Advantage of Softmax Attention: Insights from Single-Location Regressioncited by 7
- 2025Single-Head Attention in High Dimensions: A Theory of Generalization, Weights Spectra, and Scaling Lawscited by 6
- 2025Asymptotic Analysis of Two-Layer Neural Networks after One Gradient Step under Gaussian Mixtures Data with Structurecited by 4
- 2024High-dimensional Learning with Noisy Labelscited by 2
- 2025A Random Matrix Theory Perspective on the Learning Dynamics of Multi-head Latent Attentioncited by 1
Built on by
Not cited yet.