mlsanityData to 7 Oct 2026

Spectral Lens: Activation and Gradient Spectra as Diagnostics of LLM Optimization

arXiv 2605.05683 ↗First posted 7 May 2026stat.ML cs.LGcited by 3

Not on the map. Why?

Abstract (arXiv)

Training loss and throughput can hide distinct internal representation in language-model training. To examine these hidden mechanics, we use spectral measurements as practical and operational diagnostics. Using a controlled family of decoder-only models adapted from the modded NanoGPT codebase, we introduce an empirical protocol based on activation covariance and per-sample gradient SVD spectra. This dual-view reveals three empirical findings and one mechanistic explanation. First, batch size acts as a latent determinant of representation geometry: runs that reach equal loss settle into systematically distinct activation spectra. Second, the activation covariance tail measured early in training reliably forecasts downstream token efficiency. Third, movement of the activation spectrum head (leading modes), together with gradient spectra, characterizes underlying learning-dynamics changes, separating learning-side architectural improvements from primarily execution-side gains. These predictive and diagnostic signals persist across the 12-, 36-, and 48-layer model tiers. Finally, a mechanistic model proves the main observations and explains how activation covariance spectra correlate with task-aligned feature learning.

Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 37 of its 49 references that are in the map. First posted on arXiv 7 May 2026.

Builds on

37 of 49 references.

References · 37

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020Language Models are Few-Shot Learnerscited by 32,199
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  4. 2020Scaling Laws for Neural Language Modelscited by 6,411
  5. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  6. 2022Training Compute-Optimal Large Language Modelscited by 2,927
  7. 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,818
  8. 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
  9. 2021Scaling Language Models: Methods, Analysis & Insights from Training Gophercited by 1,199
  10. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
  11. 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,013
  12. 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
25 more
  1. 2019How Contextual are Contextualized Word Representations? Comparing the Geometry of BERT, ELMo, and GPT-2 Embeddingscited by 770
  2. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
  3. 2021Explaining Neural Scaling Lawscited by 353
  4. 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 289
  5. 2018An Empirical Model of Large-Batch Trainingcited by 268
  6. 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 229
  7. 2018Implicit Self-Regularization in Deep Neural Networks: Evidence from Random Matrix Theory and Implications for Learningcited by 211
  8. 2022Predictability and Surprise in Large Generative Modelscited by 209
  9. 2018Gradient Descent Happens in a Tiny Subspacecited by 208
  10. 2018A System for Massively Parallel Hyperparameter Tuningcited by 201
  11. 2020Spectrum Dependent Learning Curves in Kernel Regression and Wide Neural Networkscited by 200
  12. 2020Spectral Bias and Task-Model Alignment Explain Generalization in Kernel Regression and Infinitely Wide Neural Networkscited by 166
  13. 2021All Bark and No Bite: Rogue Dimensions in Transformer Language Models Obscure Representational Qualitycited by 150
  14. 2022RankMe: Assessing the downstream performance of pretrained self-supervised representations by their rankcited by 145
  15. 2019Traditional and Heavy-Tailed Self Regularization in Neural Network Modelscited by 104
  16. 2022A Solvable Model of Neural Scaling Lawscited by 101
  17. 2022The Geometry of Multilingual Language Model Representationscited by 94
  18. 2023A Tale of Two Circuits: Grokking as Competition of Sparse and Dense Subnetworkscited by 78
  19. 2019Measurements of Three-Level Hierarchical Structure in the Outliers in the Spectrum of Deepnet Hessianscited by 68
  20. 2025Tracing the Representation Geometry of Language Models from Pretraining to Post-trainingcited by 33
  21. 2023Predicting Grokking Long Before it Happens: A look into the loss landscape of models which grokcited by 31
  22. 2025Superposition Yields Robust Neural Scalingcited by 30
  23. 2024Nonlinear spiked covariance matrices and signal propagation in deep neural networkscited by 25
  24. 2022On the Overlooked Structure of Stochastic Gradientscited by 13
  25. 2026On the Mechanism and Dynamics of Modular Addition: Fourier Features, Lottery Ticket, and Grokkingcited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.