English

Spectral Edge Dynamics: An Analytical-Empirical Study of Phase Transitions in Neural Network Training

Machine Learning 2026-05-08 v3 Artificial Intelligence

Abstract

We develop the spectral edge analysis: phase transitions in neural network training -- grokking, capability gains, loss plateaus -- are controlled by the spectral gap of the rolling-window Gram matrix of parameter updates. In the extreme aspect ratio regime (parameters P108P \sim 10^8, window W10W \sim 10), the classical BBP detection threshold is vacuous; the operative structure is the intra-signal gap separating dominant from subdominant modes at position k=argmaxσj/σj+1k^* = \mathrm{argmax}\, \sigma_j/\sigma_{j+1}. From three assumptions we derive: (i) gap dynamics governed by a Dyson-type ODE with curvature asymmetry, damping, and gradient driving; (ii) a spectral loss decomposition linking each mode's learning contribution to its Davis--Kahan stability coefficient; (iii) the Gap Maximality Principle, showing that kk^* is the unique dynamically privileged position -- its collapse is the only one that disrupts learning, and it sustains itself through an α\alpha-feedback loop requiring no assumption on the optimizer. The adiabatic parameter A=ΔGF/(ηg2)\mathcal{A} = \|\Delta G\|_F / (\eta\, g^2) controls circuit stability: A1\mathcal{A} \ll 1 (plateau), A1\mathcal{A} \sim 1 (phase transition), A1\mathcal{A} \gg 1 (forgetting). Tested across six model families (150K--124M parameters): gap dynamics precede every grokking event (24/24 with weight decay, 1/24 without), the gap position is optimizer-dependent (Muon: k=1k^*=1, AdamW: k=2k^*=2 on the same model), and 19/20 quantitative predictions are confirmed. The framework is consistent with the edge of stability, Tensor Programs, Dyson Brownian motion, the Lottery Ticket Hypothesis, and neural scaling laws.

Keywords

Cite

@article{arxiv.2603.28964,
  title  = {Spectral Edge Dynamics: An Analytical-Empirical Study of Phase Transitions in Neural Network Training},
  author = {Yongzhong Xu},
  journal= {arXiv preprint arXiv:2603.28964},
  year   = {2026}
}

Comments

63 pages, 5 figures