The Spectral Lifecycle of Transformer Training: Transient Compression Waves, Persistent Spectral Gradients, and the Q/K--V Asymmetry
Not on the map. Why?
Abstract (arXiv)
We present the first systematic study of weight matrix singular value spectra during transformer pretraining, tracking full SVD decompositions of every weight matrix at 25-step intervals across three model scales (30M--285M parameters). We discover three phenomena: (1)~Transient Compression Waves: stable rank compression propagates as a traveling wave from early to late layers, creating a dramatic gradient that peaks early then reverses -- late layers eventually over-compress past early layers. (2)~Persistent Spectral Gradients: the power-law exponent~α develops a permanent depth gradient forming a non-monotonic inverted-U in deeper models, with peaks shifting toward earlier layers as depth increases. (3)~Q/K--V Functional Asymmetry: value/output projections compress uniformly while query/key projections carry the full depth-dependent dynamics. The dissociation between transient compression and persistent spectral shape reveals that rank and spectral shape encode fundamentally different information about training. We formalize this as a two-timescale dynamical model and derive scaling laws (Deltaα propto L0.26, R2=0.99). We validate on nine models across three families (custom, GPT-2, Pythia; 30M--1B parameters; 8--36 layers), demonstrate that α predicts layer importance (ρ=0.69--0.84, p<0.02), and show that spectral-guided pruning outperforms Last-N heuristics by 1.1×--3.6× across seven models in two families (GPT-2 124M--774M, Pythia 160M--1B), with worst-vs-best gaps up to 23.7× confirming the causal role of spectral structure.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 21 of its 28 references that are in the map. First posted on arXiv 3 Apr 2026.
Builds on
21 of 28 references.
References · 21
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,046
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 667
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 522
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
- 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 369
9 more
- 2018Implicit Self-Regularization in Deep Neural Networks: Evidence from Random Matrix Theory and Implications for Learningcited by 211
- 2020The Break-Even Point on Optimization Trajectories of Deep Neural Networkscited by 149
- 2020Predicting trends in the quality of state-of-the-art neural networks without access to training or testing datacited by 113
- 2024AlphaPruning: Using Heavy-Tailed Self Regularization Theory for Improved Layer-wise Pruning of Large Language Modelscited by 35
- 2024Approaching Deep Learning through the Spectral Dynamics of Weightscited by 29
- 2025From SGD to Spectra: A Theory of Neural Network Weight Dynamicscited by 8
- 2026The Norm-Separation Delay Law of Grokking: A First-Principles Theory of Delayed Generalizationcited by 6
- 2026Spectral Edge Dynamics of Training Trajectories: Signal--Noise Geometry Across Scalescited by 5
- 2025Early-stopping for Transformer model trainingcited by 1
Built on by
Not cited yet.