Spectral Edge Dynamics of Training Trajectories: Signal--Noise Geometry Across Scales
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
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 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
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 370
- 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 292
- 2017Empirical Analysis of the Hessian of Over-Parametrized Neural Networkscited by 290
- 2018Gradient Descent Happens in a Tiny Subspacecited by 217
- 2018Implicit Self-Regularization in Deep Neural Networks: Evidence from Random Matrix Theory and Implications for Learningcited by 215
- 2023A Tale of Two Circuits: Grokking as Competition of Sparse and Dense Subnetworkscited by 83
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
- 2026Low-Dimensional and Transversely Curved Optimization Dynamics in Grokkingcited by 11
4 more
- 2026The Geometry of Multi-Task Grokking: Transverse Instability, Superposition, and Weight Decay Phase Structurecited by 9
- 2026Low-Dimensional Execution Manifolds in Transformer Learning Dynamics: Evidence from Modular Arithmetic Taskscited by 7
- 2026Early-Warning Signals of Grokking via Loss-Landscape Geometrycited by 7
- 2026Optimizer-Induced Low-Dimensional Drift and Transverse Dynamics in Transformer Trainingcited by 4
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 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
- 2026Algebraic Dead Directions in LayerNorm Transformers: A Forward-Pass-Only Diagnostic at LLM Scalecited by 3
- 2026Dead-Direction Signatures: A Cheap Spectral Reading of Singular Complexitycited by 2
- 2026When Do Attention Circuits Form? Developmental Trajectories of Capability and Attention-Sink Emergence Across Three 1B-ClassArchitecturescited by 1
- 2026The Spectral Lifecycle of Transformer Training: Transient Compression Waves, Persistent Spectral Gradients, and the Q/K--V Asymmetrycited by 0
- 2026Dead-Direction Signatures: A Cheap Spectral Reading of Singular Complexity19 Jun 2026
- 2026Algebraic Dead Directions in LayerNorm Transformers: A Forward-Pass-Only Diagnostic at LLM Scale17 Jun 2026
- 2026When Do Attention Circuits Form? Developmental Trajectories of Capability and Attention-Sink Emergence Across Three 1B-ClassArchitectures1 Jun 2026
- 2026Spectral Probe-Circuits: A Three-Step Recipe for Identifying Attention-Head Circuits in Pretrained Transformers22 May 2026
- 2026The Spectral Lifecycle of Transformer Training: Transient Compression Waves, Persistent Spectral Gradients, and the Q/K--V Asymmetry3 Apr 2026
- 2026Spectral Edge Dynamics: An Analytical-Empirical Study of Phase Transitions in Neural Network Training30 Mar 2026