Gating Enables Curvature: A Geometric Expressivity Gap in Attention
Not on the map. Why?
Abstract (arXiv)
Multiplicative gating is widely used in neural architectures and has recently been applied to attention layers to improve performance and training stability in large language models. Despite the success of gated attention, the mathematical implications of gated attention mechanisms remain poorly understood. We study attention through the geometry of its representations by modeling outputs as mean parameters of Gaussian distributions and analyzing the induced Fisher--Rao geometry. We show that ungated attention operator is restricted to intrinsically flat statistical manifolds due to its affine structure, while multiplicative gating enables non-flat geometries, including positively curved manifolds that are unattainable in the ungated setting. These results establish a geometric expressivity gap between ungated and gated attention. Empirically, we show that gated models exhibit higher representation curvature and improved performance on tasks requiring nonlinear decision boundaries whereas they provide no consistent advantage on tasks with linear decision boundaries. Furthermore, we identify a structured regime in which curvature accumulates under composition, yielding a systematic depth amplification effect.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 15 of its 20 references that are in the map. First posted on arXiv 16 Apr 2026.
Builds on
15 of 20 references.
References · 15
- 2017Attention Is All You Needcited by 59,932
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2016Language Modeling with Gated Convolutional Networkscited by 688
- 2019Are Transformers universal approximators of sequence-to-sequence functions?cited by 404
- 2014On the Number of Linear Regions of Deep Neural Networkscited by 378
- 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 266
- 2019On the Turing Completeness of Modern Neural Network Architecturescited by 129
- 2017Fisher-Rao Metric, Geometry, and Complexity of Neural Networkscited by 99
- 2022Fisher SAM: Information Geometry and Sharpness Aware Minimisationcited by 84
- 2024Gated Slot Attention for Efficient Linear-Time Sequence Modelingcited by 73
- 2020The Depth-to-Width Interplay in Self-Attentioncited by 39
- 2016Associative Long Short-Term Memorycited by 37
3 more
- 2018Fisher Information and Natural Gradient Learning of Random Deep Networkscited by 28
- 2020Theory of gating in recurrent neural networkscited by 19
- 2016On Multiplicative Integration with Recurrent Neural Networkscited by 15
Built on by
Not cited yet.