Optimizer-Induced Low-Dimensional Drift and Transverse Dynamics in Transformer Training
Not on the map. Why?
Abstract (arXiv)
We analyze cumulative parameter trajectories of transformer training under AdamW and identify a dominant low-dimensional drift direction ("backbone") that captures 60--80% of long-horizon displacement from initialization. This direction is highly stable over rolling training windows yet reorients gradually across phases, particularly following objective reweighting. Per-batch gradients exhibit near-noise-floor alignment with the backbone, whereas optimizer-integrated updates align strongly with it, indicating that the structure emerges from accumulated optimizer dynamics rather than instantaneous gradient geometry. Replacing AdamW with SGD-family optimizers eliminates this structure, and reducing β2 smoothly degrades backbone dominance and reheating recoverability. Reheating experiments show that transverse probe modes can be transiently re-excited without substantially altering accumulated backbone drift. These results provide a trajectory-level characterization of optimizer-induced geometric structure in transformer training and shift attention from instantaneous gradient properties to cumulative update dynamics.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 8 of its 20 references that are in the map. First posted on arXiv 27 Feb 2026.
Builds on
8 of 20 references.
References · 8
- 2017Attention Is All You Needcited by 60,361
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 637
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 449
- 2014New insights and perspectives on the natural gradient methodcited by 443
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2026Low-Dimensional and Transversely Curved Optimization Dynamics in Grokkingcited by 11
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Spectral Edge Dynamics of Training Trajectories: Signal--Noise Geometry Across Scalescited by 6
- 2026Spectral Edge Dynamics: An Analytical-Empirical Study of Phase Transitions in Neural Network Trainingcited by 4
- 2026Spectral Probe-Circuits: A Three-Step Recipe for Identifying Attention-Head Circuits in Pretrained Transformerscited by 3
- 2026ANCORA: Learning to Question via Manifold-Anchored Self-Play for Verifiable Reasoningcited by 0
- 2026Spectral Probe-Circuits: A Three-Step Recipe for Identifying Attention-Head Circuits in Pretrained Transformers22 May 2026
- 2026ANCORA: Learning to Question via Manifold-Anchored Self-Play for Verifiable Reasoning30 Apr 2026
- 2026Spectral Edge Dynamics: An Analytical-Empirical Study of Phase Transitions in Neural Network Training30 Mar 2026
- 2026Spectral Edge Dynamics of Training Trajectories: Signal--Noise Geometry Across Scales14 Mar 2026