arXiv AI

On the Statistical Mechanics of Attention

arXiv AI
Sep 2

Energy-Based Transformers as Predictors of Reading Difficulty

The paper investigates energy-based transformers as predictors of reading difficulty, extending the use of transformer language models in psycholinguistics. It demonstrates that the energy measure from these models robustly predicts reading times across multiple corpora, outperforming traditional metrics like surprisal and attention entropy. In a controlled experiment on relative clause processing, energy captures known asymmetries, suggesting it may unify previously complementary predictors.

By Jakub Dotlacil, Ece Takmaz
arXiv Machine Learning
Oct 19

Anisotropic Representation Dynamics in Transformers

The paper introduces a Mixture‑Gated Theory of Anisotropic Representation Dynamics that decomposes Transformer self‑attention into value transport, query gating, and key gating. It shows that the geometry of attended representations can cause a representation difference to contract or expand, a phenomenon termed gating‑induced regime switching. The authors derive a spectral criterion to predict switching, quantify its strength, and identify subspaces where switching occurs, validating these predictions experimentally across four Transformer architectures.

By Nguyen Do, My T. Thai
arXiv Machine Learning
Oct 22

Kinetic theory for Transformers and the lost-in-the-middle phenomenon

The paper investigates causal self‑attention dynamics in decoder Transformers, modeling them as a non‑exchangeable interacting particle system. Using cumulant expansions and Glauber calculus, the authors prove a quantitative mean‑field limit and a next‑order correlation characterization. For i.i.d. uniformly distributed tokens, they solve the limiting correlation equation in closed form and rigorously explain the empirically observed lost‑in‑the‑middle phenomenon, showing a U‑shaped token retrieval profile with primacy, recency, and a unique interior minimum under a smallness condition.

By Mitia Duerinckx, Borjan Geshkovski, Stefano Rossi
arXiv Machine Learning
Jul 7

Incremental Learning of Sparse Attention Patterns in Transformers

arXiv:2602. 19143v2 Announce Type: replace Abstract: This paper studies simple transformers trained on a high-order Markov chain, where the model must incorporate information from multiple past positions, each with different statistical importance.

By O\u{g}uz Kaan Y\"uksel, Rodrigo Alvarez Lucendo, Nicolas Flammarion
arXiv Machine Learning
Sep 10

Conditioned Initialization for Attention

arXiv:2609.07086v1 Announce Type: new Abstract: Transformers are a dominant architecture in modern machine learning, powering applications across vision, language, and beyond. At the core of their su...

By Hemanth Saratchandran, Simon Lucey
arXiv AI
Oct 31

Emergence of the transformer architecture as the score of entropic laws of random measures

The paper demonstrates that the transformer architecture naturally arises as the score of last‑token laws in a statistical model of language where sentences are treated as empirical measures of tokens. By applying the maximum‑entropy principle to a prior that assumes no inherent sentence structure, the authors derive a statistic capturing token interactions via Luce’s choice axiom, leading to the softmax choice rule and attention mechanisms. The resulting framework explains key transformer components—query‑key and value matrices as Lagrange multipliers, causal masking, multi‑head attention, and residual blocks—as consequences of entropic optimization.

By Giacomo Greco