Spectral Flattening Is All Muon Needs: How Orthogonalization Controls Learning Rate and Convergence
Not on the map. Why?
Abstract (arXiv)
Muon orthogonalizes the momentum buffer before each update, replacing its singular values with ones via Newton-Schulz iterations. This simple change lets Muon tolerate far larger learning rates and converge faster than other optimizers, but why? We show that the mechanism is spectral flattening, and develop two results around it. First, we prove that Muon's maximal stable step size scales with the average singular value of the gradient rather than the largest, which bottlenecks standard gradient descent. Second, we recast Muon as a preconditioned gradient method and show, under a Kronecker-factored curvature model, that it improves the effective convergence factor, with the improvement controlled by the spectrum of the gradient covariance. Extensive experiments validate both results: Muon remains stable at learning rates that cause SGD to diverge within the first few iterations, and reaches accuracy milestones several epochs earlier even at identical step sizes. Taken together, our results offer a principled, geometric explanation for Muon's empirical success.
Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 18 of its 25 references that are in the map. First posted on arXiv 13 May 2026.
Builds on
18 of 25 references.
References · 18
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2017The Marginal Value of Adaptive Gradient Methods in Machine Learningcited by 250
- 2020Loss landscapes and optimization in over-parameterized non-linear systems and neural networkscited by 233
- 2024Old Optimizer, New Norm: An Anthologycited by 193
- 2025Fantastic Pretraining Optimizers and Where to Find Themcited by 100
- 2025Understanding Gradient Orthogonalization for Deep Learning via Non-Euclidean Trust-Region Optimizationcited by 92
- 2025Muon Optimizes Under Spectral Norm Constraintscited by 86
- 2025A Note on the Convergence of Muoncited by 83
- 2025Practical Efficiency of Muon for Pretrainingcited by 80
- 2025AdaMuon: Adaptive Muon Optimizercited by 67
6 more
- 2025When do spectral gradient updates help in deep learning?cited by 31
- 2025An Exploration of Non-Euclidean Gradient Descent: Muon and its Many Variantscited by 28
- 2026Adam Improves Muon: Adaptive Moment Estimation with Orthogonalized Momentumcited by 14
- 2025Muon Optimizer Accelerates Grokkingcited by 12
- 2025MuonAll: Muon Variant for Efficient Finetuning of Large Language Modelscited by 7
- 2025Muon: Training and Trade-offs with Latent Attention and MoEcited by 3
Built on by
Citing papers per year: 2026: 5.
5 papers.
- 2026The Spectral Dynamics and Noise Geometry of Muoncited by 3
- 2026A Note on Stability for Orthogonalized Matrix Momentum with Client Samplingcited by 1
- 2026Muse: Representation Geometry of Muon Beyond Normalized Momentumcited by 1
- 2026MALT: Lightweight Curvature-Aware Muon via Diagonal Preconditioningcited by 1
- 2026DGA-Muon: Decoupled Geometry-Aligned Adaptive Scaling for Muoncited by 0
- 2026DGA-Muon: Decoupled Geometry-Aligned Adaptive Scaling for Muon5 Oct 2026
- 2026MALT: Lightweight Curvature-Aware Muon via Diagonal Preconditioning5 Aug 2026
- 2026Muse: Representation Geometry of Muon Beyond Normalized Momentum16 Jul 2026
- 2026The Spectral Dynamics and Noise Geometry of Muon7 Jun 2026
- 2026A Note on Stability for Orthogonalized Matrix Momentum with Client Sampling1 Jun 2026