From Self-Attention to Connection Laplacian: A Unified Operator View of Transformers
Not on the map. Why?
Abstract (arXiv)
Self-attention is a ubiquitous primitive in modern sequence models, yet its operator-level geometry is only partially understood. We view a token sequence as a vector field over the token-position graph and identify attention as a connection walk: messages are aggregated by a nonnegative walk matrix while being transported along each edge by a learned linear map. Within this framework, we prove that single-head attention (SHA) is exactly a connection propagation step with constant transport, and that multi-head attention (MHA) is exactly a single edge-dependent connection walk whose effective transport is an attention-gated mixture of headwise transports. We further clarify the conditions under which the corresponding generator reduces to a random-walk connection Laplacian, highlighting the roles of stochasticity, reversibility, and metric-compatible transports. Empirically, we find that trained Transformers across scales (from 124M to 8B) and structures (encoder/decoder) exhibit geometric structure consistent with our theory: effective attention graphs converge to stable geometric operators in deeper layers, learned transports self-organize into approximate scaled isometries, and both phenomena strengthen consistently with scale. Overall, the paper provides a precise connection-walk formalism that links self-attention to classical geometric operators, along with a set of operator-level tools for analyzing transformer models from a geometric perspective.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 18 of its 26 references that are in the map. First posted on arXiv 12 Jul 2026.
Builds on
18 of 26 references.
References · 18
- 2017Attention Is All You Needcited by 59,932
- 2020Rethinking Attention with Performerscited by 1,584
- 2021Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gaugescited by 1,131
- 2020Efficient Transformers: A Surveycited by 784
- 2019Graph Transformer Networkscited by 520
- 2021Rethinking Graph Transformers with Spectral Attentioncited by 468
- 2020Hopfield Networks is All You Needcited by 461
- 2019Gauge Equivariant Convolutional Networks and the Icosahedral CNNcited by 262
- 2022Neural Sheaf Diffusion: A Topological Perspective on Heterophily and Oversmoothing in GNNscited by 208
- 2019A Tensorized Transformer for Language Modelingcited by 85
- 2022Sheaf Neural Networks with Connection Laplacianscited by 64
- 2024On the Role of Attention Masks and LayerNorm in Transformerscited by 45
6 more
- 2025Tensor Product Attention Is All You Needcited by 27
- 2021Transformers are Deep Infinite-Dimensional Non-Mercer Binary Kernel Machinescited by 22
- 2021On Learning the Transformer Kernelcited by 14
- 2025Neural ODE Transformers: Analyzing Internal Dynamics and Adaptive Fine-tuningcited by 12
- 2025OT-Transformer: A Continuous-time Transformer Architecture with Optimal Transport Regularizationcited by 7
- 2025A Framework for Non-Linear Attention via Modern Hopfield Networkscited by 1
Built on by
Not cited yet.