Convergence of Practical Muon with Finite Newton-Schulz Iterations and Nesterov Momentum
Not on the map. Why?
Abstract (arXiv)
Practical Muon maintains momentum and performs a small, fixed number of Newton--Schulz iterations separately for each parameter matrix, often with a Nesterov correction. We analyze these layer-wise finite-step updates jointly on a coupled nonconvex objective, rather than replacing them by exact polar factors or one global orthogonalization. Under gradient-dependent (mathcal L0,mathcal L1,q)-smoothness and conditionally unbiased stochastic gradients with bounded layer-wise variance, we establish an mathcal O(T-1/4) bound on the expected average Frobenius gradient norm. The analysis retains the Nesterov recursion and requires neither bounded stochastic gradients, symmetric noise, nor a uniform positive lower bound on the nonzero output singular values. Its constants contain no explicit matrix-dimension or rank factors when the number of blocks and problem constants are fixed. The proof follows a descent inequality and a decomposition of the momentum tracking error into initialization, noise, and drift. For the original five-step quintic, we verify the required scalar-map bounds analytically; the result also allows step-dependent coefficients satisfying the same bounds. A complementary nuclear-norm result quantifies rank dependence under a stronger spectral condition. The vanishing rate uses coupled learning-rate and momentum schedules, including the standard single-coefficient Nesterov rule.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 14 of its 16 references that are in the map. First posted on arXiv 30 Sep 2026.
Builds on
14 of 16 references.
References · 14
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2020On Layer Normalization in the Transformer Architecturecited by 922
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
- 2025Training Deep Learning Models with Norm-Constrained LMOscited by 192
- 2025Practical Efficiency of Muon for Pretrainingcited by 80
- 2025Gluon: Making Muon & Scion Great Again! (Bridging Theory and Practice of LMO-based Optimizers for LLMs)cited by 72
- 2026Convergence of Muon with Newton-Schulzcited by 36
- 2022Maximal Initial Learning Rates in Deep ReLU Networkscited by 11
- 2023Depth Dependence of μP Learning Rates in ReLU MLPscited by 9
- 2026Muon with Nesterov Momentum: Heavy-Tailed Noise and (Randomized) Inexact Polar Decompositioncited by 8
Built on by
Citing papers per year: 2026: 1.
1 paper.