Energy-Gated Attention: Spectral Salience as an Inductive Bias for Transformer Attention
Not on the map. Why?
Abstract (arXiv)
Standard transformer attention computes pairwise similarity between queries and keys, treating all tokens as equally salient regardless of their intrinsic informational content. In turbulent fluid dynamics, coherent structures -- the energetically dominant, spatially organized patterns that persist amid background chaos -- carry a disproportionate fraction of total energy and govern all transport. We propose that tokens play an analogous role in transformer attention: informationally dense positions (morphological boundaries, syntactic heads, discourse markers) concentrate spectral energy and should attract proportionally more attention than background tokens (function words, repeated patterns, low-information filler). We propose Energy-Gated Attention (EGA): a simple modification that gates value aggregation by the spectral energy of key token embeddings, computed by a single learned linear projection that discovers the dominant spectral mode of the embedding field. On TinyShakespeare, EGA achieves +0.103 validation loss improvement with only 12,480 additional parameters (<0.26% overhead) and no measurable computational cost. The result is consistent on Penn Treebank (+0.101), demonstrating dataset independence. A systematic ablation across three wavelet families (fixed Morlet, Daubechies db2/db4, and a parametric Morlet) establishes that fixed structured bases are suboptimal -- the optimal energy direction is data-adaptive and non-sinusoidal -- while identifying learned wavelet packets as a promising open direction. The learned energy threshold converges to tau ~= 0.35 independently of initialization, corresponding to the fraction (~36%) of tokens carrying above-average spectral energy in English text, a stable linguistic property consistent with the fraction of content words in running English text.
Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 11 of its 22 references that are in the map. First posted on arXiv 21 May 2026.
Builds on
11 of 22 references.
References · 11
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2020Longformer: The Long-Document Transformercited by 3,413
- 2020Big Bird: Transformers for Longer Sequencescited by 1,684
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,103
- 2021FNet: Mixing Tokens with Fourier Transformscited by 347
- 2018Multi-level Wavelet-CNN for Image Restorationcited by 184
- 2021LEAF: A Learnable Frontend for Audio Classificationcited by 69
- 2020Language Through a Prism: A Spectral Approach for Multiscale Language Representationscited by 39
- 2024Towards Signal Processing In Large Language Modelscited by 7
Built on by
Citing papers per year: 2026: 5.
5 papers.
- 2026Energy-Gated Attention and Wavelet Positional Encoding: Complementary Inductive Biases for Transformer Attentioncited by 4
- 2026Beyond Sinusoids: A Morlet Wavelet Framework for Transformer Positional Encodingcited by 3
- 2026Multiscale POD of Transformer Attention Fields: Scale-Selective Analysis via Morlet Scalogramcited by 2
- 2026FourierQK: Spectral Preprocessing of Query-Key Projections Improves Transformer Attentioncited by 1
- 2026FourierQK: Filter Shape, Admissibility and the Leakage-Coverage Lawcited by 0
- 2026FourierQK: Filter Shape, Admissibility and the Leakage-Coverage Law9 Jul 2026
- 2026FourierQK: Spectral Preprocessing of Query-Key Projections Improves Transformer Attention8 Jul 2026
- 2026Multiscale POD of Transformer Attention Fields: Scale-Selective Analysis via Morlet Scalogram4 Jun 2026
- 2026Beyond Sinusoids: A Morlet Wavelet Framework for Transformer Positional Encoding31 May 2026
- 2026Energy-Gated Attention and Wavelet Positional Encoding: Complementary Inductive Biases for Transformer Attention25 May 2026