When do spectral gradient updates help in deep learning?
Not on the map. Why?
Abstract (arXiv)
Spectral gradient methods, such as the recently popularized Muon optimizer, are a promising alternative to standard Euclidean gradient descent for training deep neural networks and transformers, but it is still unclear in which regimes they are expected to perform better. We propose a simple layerwise condition that predicts when a spectral update yields a larger decrease in the loss than a Euclidean gradient step. This condition compares, for each parameter block, the squared nuclear-to-Frobenius ratio of the gradient to the stable rank of the incoming activations. To understand when this condition may be satisfied, we first prove that post-activation matrices have low stable rank at Gaussian initialization in random feature regression, feedforward networks, and transformer blocks. In spiked random feature models we then show that, after a short burn-in, the Euclidean gradient's nuclear-to-Frobenius ratio grows with the data dimension while the stable rank of the activations remains bounded, so the predicted advantage of spectral updates scales with dimension. We validate these predictions in synthetic regression experiments and in NanoGPT-scale language model training, where we find that intermediate activations have low-stable-rank throughout training and the corresponding gradients maintain large nuclear-to-Frobenius ratios. Together, these results identify conditions for spectral gradient methods, such as Muon, to be effective in training deep networks and transformers.
Key facts. Cited by 25 papers in the map (data through 7 Oct 2026). First posted on arXiv 3 Dec 2025.
Builds on
0 of 0 references.
No references here yet.
Built on by
Citing papers per year: 2025: 1; 2026: 24.
25 papers.
Sort:
By area:
- 2026Preconditioning Benefits of Spectral Orthogonalization in Muoncited by 28
- 2026The Newton-Muon Optimizercited by 14
- 2026Adam Improves Muon: Adaptive Moment Estimation with Orthogonalized Momentumcited by 13
- 2026Insights on Muon from Simple Quadraticscited by 11
- 2026On the Width Scaling of Neural Optimizers Under Matrix Operator Norms I: Row/Column Normalization and Hyperparameter Transfercited by 11
- 2026HTMuon: Improving Muon via Heavy-Tailed Spectral Correctioncited by 11
- 2026Muon is Not That Special: Random or Inverted Spectra Work Just as Wellcited by 8
- 2025Understanding the Mechanisms of Fast Hyperparameter Transfercited by 6
- 2026Spectral Gradient Descent Mitigates Anisotropy-Driven Misalignment: A Case Study in Phase Retrievalcited by 5
- 2026Clarifying Shampoo: Adapting Spectral Descent to Stochasticity and the Parameter Trajectorycited by 5
- 2026Enhancing LLM Training via Spectral Clippingcited by 5
- 2026Muon-OGD: Muon-based Spectral Orthogonal Gradient Projection for LLM Continual Learningcited by 5
13 more
- 2026Symmetry-Compatible Principle for Optimizer Design: Embeddings, LM Heads, SwiGLU MLPs, and MoE Routerscited by 5
- 2026PRISM: Distribution-free Adaptive Computation of Matrix Functions for Accelerating Neural Network Trainingcited by 4
- 2026Spectral Flattening Is All Muon Needs: How Orthogonalization Controls Learning Rate and Convergencecited by 4
- 2026Muon with Finite Newton-Schulz: The Smoothing Benefit in Nonsmooth Nonconvex Optimizationcited by 4
- 2026Decoupling Variance and Scale-Invariant Updates in Adaptive Gradient Descent for Unified Vector and Matrix Optimizationcited by 3
- 2026Anytime Training with Schedule-Free Spectral Optimizationcited by 3
- 2026NS-RGS: Newton-Schulz based Riemannian gradient method for orthogonal group synchronizationcited by 2
- 2026Hierarchical Muon: Tiled Newton-Schulz Updates for Efficient Muon Optimizationcited by 2
- 2026Transforming Rank: How Architecture Navigates the Spectral Pathologies of Depthcited by 2
- 2026Schattor: Schatten-family methods for deep learning optimizationcited by 0
- 2026Muon learns balanced solutions in matrix factorization without slow saddle-to-saddle dynamicscited by 0
- 2026Reassessing Muon for Matrix Factorizationcited by 0
- 2026Refusal geometry reflects refusal training: diverse refusal prefixes can raise stable rank and weaken refusal vector ablation attackscited by 0
- 2026Muon with Finite Newton-Schulz: The Smoothing Benefit in Nonsmooth Nonconvex Optimization26 Aug 2026
- 2026Refusal geometry reflects refusal training: diverse refusal prefixes can raise stable rank and weaken refusal vector ablation attacks26 Aug 2026
- 2026Transforming Rank: How Architecture Navigates the Spectral Pathologies of Depth15 Jul 2026
- 2026Reassessing Muon for Matrix Factorization14 Jul 2026
- 2026Muon learns balanced solutions in matrix factorization without slow saddle-to-saddle dynamics29 Jun 2026
- 2026Hierarchical Muon: Tiled Newton-Schulz Updates for Efficient Muon Optimization25 Jun 2026
- 2026Schattor: Schatten-family methods for deep learning optimization14 Jun 2026
- 2026Anytime Training with Schedule-Free Spectral Optimization21 May 2026
- 2026Symmetry-Compatible Principle for Optimizer Design: Embeddings, LM Heads, SwiGLU MLPs, and MoE Routers18 May 2026
- 2026Spectral Flattening Is All Muon Needs: How Orthogonalization Controls Learning Rate and Convergence13 May 2026
- 2026Muon is Not That Special: Random or Inverted Spectra Work Just as Well11 May 2026
- 2026Muon-OGD: Muon-based Spectral Orthogonal Gradient Projection for LLM Continual Learning9 May 2026
13 more
- 2026NS-RGS: Newton-Schulz based Riemannian gradient method for orthogonal group synchronization7 Apr 2026
- 2026The Newton-Muon Optimizer1 Apr 2026
- 2026Enhancing LLM Training via Spectral Clipping15 Mar 2026
- 2026HTMuon: Improving Muon via Heavy-Tailed Spectral Correction10 Mar 2026
- 2026On the Width Scaling of Neural Optimizers Under Matrix Operator Norms I: Row/Column Normalization and Hyperparameter Transfer10 Mar 2026
- 2026Adam Improves Muon: Adaptive Moment Estimation with Orthogonalized Momentum19 Feb 2026
- 2026Insights on Muon from Simple Quadratics12 Feb 2026
- 2026Clarifying Shampoo: Adapting Spectral Descent to Stochasticity and the Parameter Trajectory10 Feb 2026
- 2026Decoupling Variance and Scale-Invariant Updates in Adaptive Gradient Descent for Unified Vector and Matrix Optimization6 Feb 2026
- 2026Spectral Gradient Descent Mitigates Anisotropy-Driven Misalignment: A Case Study in Phase Retrieval30 Jan 2026
- 2026PRISM: Distribution-free Adaptive Computation of Matrix Functions for Accelerating Neural Network Training29 Jan 2026
- 2026Preconditioning Benefits of Spectral Orthogonalization in Muon20 Jan 2026
- 2025Understanding the Mechanisms of Fast Hyperparameter Transfer28 Dec 2025
- 2026Preconditioning Benefits of Spectral Orthogonalization in Muoncited by 28
- 2026The Newton-Muon Optimizercited by 14
- 2026Adam Improves Muon: Adaptive Moment Estimation with Orthogonalized Momentumcited by 13
- 2026Insights on Muon from Simple Quadraticscited by 11
- 2026On the Width Scaling of Neural Optimizers Under Matrix Operator Norms I: Row/Column Normalization and Hyperparameter Transfercited by 11
- 2026HTMuon: Improving Muon via Heavy-Tailed Spectral Correctioncited by 11
- 2026Muon is Not That Special: Random or Inverted Spectra Work Just as Wellcited by 8
- 2025Understanding the Mechanisms of Fast Hyperparameter Transfercited by 6
- 2026Spectral Gradient Descent Mitigates Anisotropy-Driven Misalignment: A Case Study in Phase Retrievalcited by 5
- 2026Clarifying Shampoo: Adapting Spectral Descent to Stochasticity and the Parameter Trajectorycited by 5
- 2026Enhancing LLM Training via Spectral Clippingcited by 5
- 2026Muon-OGD: Muon-based Spectral Orthogonal Gradient Projection for LLM Continual Learningcited by 5
12 most cited of 23 All 23 (JSON)
- 2026Muon with Finite Newton-Schulz: The Smoothing Benefit in Nonsmooth Nonconvex Optimization26 Aug 2026
- 2026Transforming Rank: How Architecture Navigates the Spectral Pathologies of Depth15 Jul 2026
- 2026Reassessing Muon for Matrix Factorization14 Jul 2026
- 2026Muon learns balanced solutions in matrix factorization without slow saddle-to-saddle dynamics29 Jun 2026
- 2026Hierarchical Muon: Tiled Newton-Schulz Updates for Efficient Muon Optimization25 Jun 2026
- 2026Schattor: Schatten-family methods for deep learning optimization14 Jun 2026
- 2026Anytime Training with Schedule-Free Spectral Optimization21 May 2026
- 2026Spectral Flattening Is All Muon Needs: How Orthogonalization Controls Learning Rate and Convergence13 May 2026
- 2026Muon is Not That Special: Random or Inverted Spectra Work Just as Well11 May 2026
- 2026Muon-OGD: Muon-based Spectral Orthogonal Gradient Projection for LLM Continual Learning9 May 2026
- 2026NS-RGS: Newton-Schulz based Riemannian gradient method for orthogonal group synchronization7 Apr 2026
- 2026The Newton-Muon Optimizer1 Apr 2026
12 newest of 23 All 23 (JSON)