mlsanityData to 14 Oct 2026

Spectral Edge Dynamics of Training Trajectories: Signal--Noise Geometry Across Scales

arXiv 2603.15678 ↗First posted 14 Mar 2026cs.LG cs.AIcited by 6

Not on the map. Why?

Abstract (arXiv)

Despite hundreds of millions of parameters, transformer training trajectories evolve within only a few coherent directions. We introduce Spectral Edge Dynamics (SED) to quantify this structure: a rolling-window SVD of parameter updates reveals a sharp boundary -- the spectral edge -- between coherent optimization directions and stochastic noise, identified via the maximum consecutive singular value ratio σk / σk+1. Across a 51M-parameter TinyStories model (4 seeds) and GPT-2 124M under distribution shift, the spectral edge exhibits a universal three-phase pattern (rise, plateau, collapse). The effective signal rank adapts to task complexity (k^* = 2 at 51M, k^* = 3 at 124M), and the directional coupling between spectral geometry and validation loss reverses with window size -- a lag flip reflecting the timescale of trajectory integration. Johnson--Lindenstrauss projection to d = 10W dimensions (e.g., d = 100 for W = 10) preserves the spectral gap within 5.7%, making the framework applicable to models of arbitrary scale. In companion work, the same spectral geometry provides early-warning signals of grokking -- predicting generalization 600--1,700 steps before it occurs across modular arithmetic, Dyck languages, and the SCAN benchmark.

Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 16 of its 26 references that are in the map. First posted on arXiv 14 Mar 2026.

Builds on

16 of 26 references.

References · 16

  1. 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
  2. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
  3. 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 449
  4. 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
  5. 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 370
  6. 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 292
  7. 2017Empirical Analysis of the Hessian of Over-Parametrized Neural Networkscited by 290
  8. 2018Gradient Descent Happens in a Tiny Subspacecited by 217
  9. 2018Implicit Self-Regularization in Deep Neural Networks: Evidence from Random Matrix Theory and Implications for Learningcited by 215
  10. 2023A Tale of Two Circuits: Grokking as Competition of Sparse and Dense Subnetworkscited by 83
  11. 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
  12. 2026Low-Dimensional and Transversely Curved Optimization Dynamics in Grokkingcited by 11
4 more
  1. 2026The Geometry of Multi-Task Grokking: Transverse Instability, Superposition, and Weight Decay Phase Structurecited by 9
  2. 2026Low-Dimensional Execution Manifolds in Transformer Learning Dynamics: Evidence from Modular Arithmetic Taskscited by 7
  3. 2026Early-Warning Signals of Grokking via Loss-Landscape Geometrycited by 7
  4. 2026Optimizer-Induced Low-Dimensional Drift and Transverse Dynamics in Transformer Trainingcited by 4

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 6.

6 papers.