mlsanityData to 7 Oct 2026

Feed-Forward Steering in Transformer Residual Dynamics

arXiv 2608.02071 ↗First posted 3 Aug 2026cs.LG math.DScited by 1

Not on the map. Why?

Abstract (arXiv)

Attention-only dynamical theories model Transformer residual directions as particles aggregating on a sphere. We extend this framework by incorporating the feed-forward network (FFN) term as a local steering field acting on each token state. The resulting theory predicts that the tangential component of the FFN field is necessary for motion in residual-direction space, that critical residual directions correspond to nonlinear projective equilibria, and that a commutator defect determines when a finite attention--FFN block can be accurately approximated by a parallel, additive flow. Across GPT-2, Pythia, Mistral, and Llama models, the extended theory improves one-step angular prediction relative to an attention-only baseline, with the contribution of the FFN increasing from GPT-2 to Llama-3-8B. Intervention experiments show that retaining only the tangential FFN component preserves most model quality, whereas retaining only the radial component causes performance to collapse. The tangential component also preserves output diversity under aggregation pressure. As a practical application, layers with small commutator defects can be approximately parallelized with only a modest increase in loss, whereas layers with large defects degrade rapidly. These findings support the interpretation of FFN layers as directional steering fields that shape Transformer residual geometry and govern the feasibility of block-level interventions.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 14 of its 18 references that are in the map. First posted on arXiv 3 Aug 2026.

Builds on

14 of 18 references.

References · 14

  1. 2024The Llama 3 Herd of Modelscited by 14,192
  2. 2023Mistral 7Bcited by 3,065
  3. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  4. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
  5. 2020GLU Variants Improve Transformercited by 1,724
  6. 2019Root Mean Square Layer Normalizationcited by 1,425
  7. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
  8. 2022Mass-Editing Memory in a Transformercited by 976
  9. 2020On Layer Normalization in the Transformer Architecturecited by 906
  10. 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 522
  11. 2023A mathematical perspective on Transformerscited by 160
  12. 2023The emergence of clusters in self-attention dynamicscited by 133
2 more
  1. 2024Clustering in Causal Attention Maskingcited by 31
  2. 2025Normalization in Attention Dynamicscited by 9

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.