mlsanityData to 14 Oct 2026

Spectral Scaling Laws of Muon

arXiv 2606.04058 ↗First posted 2 Jun 2026cs.LG cs.AIcited by 4

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

  1. 2024The Llama 3 Herd of Modelscited by 14,715
  2. 2020Scaling Laws for Neural Language Modelscited by 6,586
  3. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  4. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  5. 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 1,040
  6. 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 587
  7. 2025Muon is Scalable for LLM Trainingcited by 428
  8. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
  9. 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
  10. 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 192
  11. 2025Training Deep Learning Models with Norm-Constrained LMOscited by 192
  12. 2025Fantastic Pretraining Optimizers and Where to Find Themcited by 100
8 more
  1. 2025NorMuon: Making Muon more efficient and scalablecited by 84
  2. 2023A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scalecited by 77
  3. 2024Modular Duality in Deep Learningcited by 77
  4. 2025Dion: Distributed Orthonormalized Updatescited by 68
  5. 2020Disentangling Adaptive Gradient Methods from Learning Ratescited by 42
  6. 2025MuonBP: Faster Muon via Block-Periodic Orthogonalizationcited by 35
  7. 2025Purifying Shampoo: Investigating Shampoo's Heuristics by Decomposing its Preconditionercited by 23
  8. 2025Dion2: A Simple Method to Shrink Matrix in Muoncited by 13

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 4.

4 papers.