Spectral Scaling Laws of Muon
Not on the map. Why?
Abstract (arXiv)
Orthonormalized update rules have rapidly become a leading choice of optimizer for training large language models, with recent open-source state-of-the-art models adopting Muon. To keep these updates tractable, Muon performs the orthonormalization with the Newton--Schulz (NS) iteration. Since NS is only approximate, directions with small singular values fail to be orthonormalized. In Muon, NS is applied to the momentum matrix at every step, yet little is known about how the singular value spectrum of these momentum matrices behaves during training, or how that behavior changes with model size. We present the first systematic study of this question. Tracking singular value quantiles of the momentum buffer across layers in models ranging from 77M to 2.8B parameters, we observe a consistent picture: after a short burn-in, the quantiles stabilize at a value determined by the layer type and model size. These stabilization values follow remarkably clean power laws in model size, with layer-dependent exponents. Layers up to mid-late depth scale very mildly with model size M (around M-0.25), so the standard 5-step NS configuration used at academic scale will continue to orthonormalize them at much larger scales. Some of the late layers, however, scale much more aggressively (up to M-0.96) and will fall into the NS failure regime at frontier scale unless one uses more NS iterations or better-tuned coefficients. NS iterations are computationally expensive at scale; our laws give practitioners a principled, layer-aware recipe for choosing the minimum NS configuration that still orthonormalizes the directions that matter -- avoiding unnecessary computation without sacrificing update quality.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 20 of its 33 references that are in the map. First posted on arXiv 2 Jun 2026.
Builds on
20 of 33 references.
References · 20
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
- 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 1,040
- 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 587
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
- 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
- 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 192
- 2025Training Deep Learning Models with Norm-Constrained LMOscited by 192
- 2025Fantastic Pretraining Optimizers and Where to Find Themcited by 100
8 more
- 2025NorMuon: Making Muon more efficient and scalablecited by 84
- 2023A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scalecited by 77
- 2024Modular Duality in Deep Learningcited by 77
- 2025Dion: Distributed Orthonormalized Updatescited by 68
- 2020Disentangling Adaptive Gradient Methods from Learning Ratescited by 42
- 2025MuonBP: Faster Muon via Block-Periodic Orthogonalizationcited by 35
- 2025Purifying Shampoo: Investigating Shampoo's Heuristics by Decomposing its Preconditionercited by 23
- 2025Dion2: A Simple Method to Shrink Matrix in Muoncited by 13
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026SOAP, Muon, and Beyond: Pushing LLM Pretraining Scalescited by 6
- 2026Muon meets Tamed Langevin: Momentum Preconditioning beyond Convex and gradient-Lipschitz Potentialscited by 0
- 2026A Solvable Model of Adaptive Learning Rate Rescaling: Acceleration, Stability & Scalingcited by 0
- 2026Does Muon Need Fine-Grained Spectral Shaping?cited by 0
- 2026Does Muon Need Fine-Grained Spectral Shaping?5 Oct 2026
- 2026A Solvable Model of Adaptive Learning Rate Rescaling: Acceleration, Stability & Scaling5 Oct 2026
- 2026Muon meets Tamed Langevin: Momentum Preconditioning beyond Convex and gradient-Lipschitz Potentials1 Oct 2026
- 2026SOAP, Muon, and Beyond: Pushing LLM Pretraining Scales13 Jul 2026