arXiv · 2604.07380
The Lifecycle of the Spectral Edge: From Gradient Learning to Weight-Decay Compression
Abstract
We decompose the spectral edge -- the dominant direction of the Gram matrix of parameter updates -- into its gradient and weight-decay components during grokking in two sequence tasks (Dyck-1 and SCAN). We find a sharp two-phase lifecycle: before grokking the edge is gradient-driven and functionally active; at grokking, gradient and weight decay align, and the edge becomes a compression axis that is perturbation-flat yet ablation-critical (>4000x more impactful than random directions). Three universality classes emerge (functional, mixed, compression), predicted by the gap flow equation. Nonlinear probes show information is re-encoded, not lost (MLP $R^2=0.99$ where linear $R^2=0.86$), and removing weight decay post-grok reverses compression while preserving the algorithm.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Yongzhong Xu. 2026-04-08. The Lifecycle of the Spectral Edge: From Gradient Learning to Weight-Decay Compression. https://arxiv.org/abs/2604.07380
Cite the original work for its findings. Save a collection to share your selection of sources.