Muon Does Not Converge on Convex Lipschitz Functions
Not on the map. Why?
Abstract (arXiv)
Muon and its variants have shown strong empirical performance in a variety of deep learning tasks. Existing convergence analyses of Muon rely on smoothness assumptions, though arguably the most successful function class for developing deep learning methods (such as AdaGrad, Shampoo, Schedule-Free and more) has been the class of convex and Lipschitz functions. In this paper we question whether the classical convex Lipschitz model is a useful one for understanding Muon. Our answer is no. We show that Muon does not converge on the class of convex and Lipschitz functions, regardless of the choice of learning rate schedule. We also show that error feedback restores convergence of Muon and all the non-Euclidean subgradient methods with momentum. However, this theoretical fix using error feedback degrades the performance of Muon in two representative settings for image classification (CIFAR-10) and language modeling (nanoGPT on FineWeb-Edu 10B). Our conclusion is that convex Lipschitz theory, despite having a prominent role in the design of practical methods for deep learning, is not the most suited one for Muon. This suggests that Muon's success must come from structure absent from this model, most plausibly related to smoothness.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 21 of its 57 references that are in the map. First posted on arXiv 9 May 2026.
Builds on
21 of 57 references.
References · 21
- 2016Layer Normalizationcited by 5,669
- 2025Kimi K2: Open Agentic Intelligencecited by 567
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2024Old Optimizer, New Norm: An Anthologycited by 193
- 2023Handbook of Convergence Theorems for (Stochastic) Gradient Methodscited by 163
- 2023A Spectral Condition for Feature Learningcited by 109
- 2025Understanding Gradient Orthogonalization for Deep Learning via Non-Euclidean Trust-Region Optimizationcited by 92
- 2025A Note on the Convergence of Muoncited by 83
- 2023Benchmarking Neural Network Training Algorithmscited by 52
- 2025PolarGrad: A Class of Matrix-Gradient Optimizers from a Unifying Preconditioning Perspectivecited by 52
- 2025Lions and Muons: Optimization via Stochastic Frank-Wolfe under Heavy-Tailed Noisecited by 41
- 2025Isotropic Curvature Model for Understanding Deep Learning Optimization: Is Gradient Orthogonalization Optimal?cited by 37
9 more
- 2025On the Convergence of Muon and Beyondcited by 31
- 2025When do spectral gradient updates help in deep learning?cited by 31
- 2025Convergence Bound and Critical Batch Size of Muon Optimizercited by 28
- 2025An Exploration of Non-Euclidean Gradient Descent: Muon and its Many Variantscited by 28
- 2026Adaptive Matrix Online Learning through Smoothing with Guarantees for Nonsmooth Nonconvex Optimizationcited by 14
- 2025Optimization Benchmark for Diffusion Models on Dynamical Systemscited by 7
- 2025On the Convergence Analysis of Muoncited by 5
- 2026Where Does Warm-Up Come From? Adaptive Scheduling for Norm-Constrained Optimizerscited by 5
- 2025Training neural networks faster with minimal tuning using pre-computed lists of hyperparameters for NAdamWcited by 1
Built on by
Citing papers per year: 2026: 9.
9 papers.
- 2026Muon with Finite Newton-Schulz: The Smoothing Benefit in Nonsmooth Nonconvex Optimizationcited by 5
- 2026The Spectral Dynamics and Noise Geometry of Muoncited by 3
- 2026Tight Complexity of Muon under Heavy-Tailed Noisecited by 2
- 2026PoLoRA: A Preconditioned Orthogonalized LoRA Optimizercited by 2
- 2026On the Convergence of Adam, Revisitedcited by 1
- 2026Sign compression for Muon: SignMuon, MuonSign, and the Limits of Error Feedbackcited by 0
- 2026Nonsmooth Optimization via Orthogonalized Momentumcited by 0
- 2026The Row Normalization Puzzle in Muoncited by 0
- 2026Muon meets Tamed Langevin: Momentum Preconditioning beyond Convex and gradient-Lipschitz Potentialscited by 0
- 2026Muon meets Tamed Langevin: Momentum Preconditioning beyond Convex and gradient-Lipschitz Potentials1 Oct 2026
- 2026The Row Normalization Puzzle in Muon30 Sep 2026
- 2026Nonsmooth Optimization via Orthogonalized Momentum12 Sep 2026
- 2026Muon with Finite Newton-Schulz: The Smoothing Benefit in Nonsmooth Nonconvex Optimization26 Aug 2026
- 2026Sign compression for Muon: SignMuon, MuonSign, and the Limits of Error Feedback31 Jul 2026
- 2026PoLoRA: A Preconditioned Orthogonalized LoRA Optimizer20 Jul 2026
- 2026On the Convergence of Adam, Revisited3 Jul 2026
- 2026The Spectral Dynamics and Noise Geometry of Muon7 Jun 2026
- 2026Tight Complexity of Muon under Heavy-Tailed Noise18 May 2026