mlsanityData to 7 Oct 2026

Gradient Boosting within a Single Attention Layer

arXiv 2604.03190 ↗First posted 3 Apr 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Transformer attention computes a single softmax-weighted average over values -- a one-pass estimate that cannot correct its own errors. We introduce gradient-boosted attention, which applies the principle of gradient boosting within a single attention layer: a second attention pass, with its own learned projections, attends to the prediction error of the first and applies a gated correction. Under a squared reconstruction objective, the construction maps onto Friedman's gradient boosting machine, with each attention pass as a base learner and the per-dimension gate as the shrinkage parameter. We show that a single Hopfield-style update erases all query information orthogonal to the stored-pattern subspace, and that further iteration under local contraction can collapse distinct queries in the same region to the same fixed point. We also show that separate projections for the correction pass can recover residual information inaccessible to the shared-projection approach of Tukey's twicing. On 10M-token subsets of WikiText-103 and OpenWebText, gradient-boosted attention improves test perplexity by 6.0% and 5.6% over standard attention, outperforming both Twicing Attention and a parameter-matched wider baseline on both benchmarks, with two rounds capturing most of the benefit. We further show, both theoretically and empirically, that the mechanism requires the additive residual structure of Pre-LN transformers: under Post-LN, the same architecture degrades perplexity by 9.6%.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 17 of its 23 references that are in the map. First posted on arXiv 3 Apr 2026.

Builds on

17 of 23 references.

References · 17

  1. 2017Attention Is All You Needcited by 59,932
  2. 2016Pointer Sentinel Mixture Modelscited by 2,992
  3. 2016Identity Mappings in Deep Residual Networkscited by 1,960
  4. 2020On Layer Normalization in the Transformer Architecturecited by 906
  5. 2018Universal Transformerscited by 613
  6. 2019Deep Equilibrium Modelscited by 603
  7. 2020Hopfield Networks is All You Needcited by 461
  8. 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 266
  9. 2021PonderNet: Learning to Pondercited by 156
  10. 2023Transformers Implement Functional Gradient Descent to Learn Non-Linear Functions In Contextcited by 63
  11. 2026Attention Residualscited by 57
  12. 2020Gradient Boosting Neural Networks: GrowNetcited by 46
5 more
  1. 2017Learning Deep ResNet Blocks Sequentially using Boosting Theorycited by 32
  2. 2025In-context denoising with one-layer transformers: connections between attention and associative memory retrievalcited by 7
  3. 2025DeepCrossAttention: Supercharging Transformer Residual Connectionscited by 7
  4. 2019Residual Networks Behave Like Boosting Algorithmscited by 3
  5. 2025Transformer Meets Twicing: Harnessing Unattended Residual Informationcited by 3

Built on by

Not cited yet.