English

Optimizer-Induced Low-Dimensional Drift and Transverse Dynamics in Transformer Training

Machine Learning 2026-03-20 v3 Artificial Intelligence

Abstract

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\beta_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.

Keywords

Cite

@article{arxiv.2602.23696,
  title  = {Optimizer-Induced Low-Dimensional Drift and Transverse Dynamics in Transformer Training},
  author = {Yongzhong Xu},
  journal= {arXiv preprint arXiv:2602.23696},
  year   = {2026}
}

Comments

23 pages, 4 figures

R2 v1 2026-07-01T10:54:57.433Z