Reassessing Muon for Matrix Factorization
Not on the map. Why?
Abstract (arXiv)
Muon has recently emerged as a strong optimizer for large-scale deep learning, where it reshapes gradient updates through approximate orthogonalization and has been reported to outperform Adam and AdamW in large language model training. Its empirical success has motivated a growing body of theoretical work that interprets Muon as steepest descent under the spectral norm. Yet it remains unclear which of Muon's advantages stem from its update rule itself and which are artifacts of the scale, architecture, and data of modern deep networks. In this work, we isolate the optimizer from these confounding factors by studying Muon on a simple, well-understood, and spectrally structured problem: low-rank matrix factorization. Through a controlled comparison against carefully tuned adaptive baselines, we find that Muon does not consistently outperform AdamW in this setting and that several previously reported advantages are sensitive to hyperparameter choices. Our results provide a more nuanced picture of when spectrum-aware orthogonalization is beneficial and argue for evaluating modern optimizers on controlled problems in addition to end-to-end benchmarks.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 40 of its 50 references that are in the map. First posted on arXiv 14 Jul 2026.
Builds on
40 of 50 references.
References · 40
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 402
- 2025Muon is Scalable for LLM Trainingcited by 381
- 2017Implicit Regularization in Matrix Factorizationcited by 318
- 2018On the Optimization of Deep Networks: Implicit Acceleration by Overparameterizationcited by 280
- 2015Tensorizing Neural Networkscited by 246
- 2025Training Deep Learning Models with Norm-Constrained LMOscited by 177
- 2024Old Optimizer, New Norm: An Anthologycited by 173
- 2025The Polar Express: Optimal Matrix Sign Methods and Their Application to the Muon Algorithmcited by 104
- 2015On the Expressive Power of Deep Learning: A Tensor Analysiscited by 97
- 2025Understanding Gradient Orthogonalization for Deep Learning via Non-Euclidean Trust-Region Optimizationcited by 81
- 2025A Note on the Convergence of Muoncited by 78
28 more
- 2025Muon Optimizes Under Spectral Norm Constraintscited by 77
- 2025Practical Efficiency of Muon for Pretrainingcited by 74
- 2024Modular Duality in Deep Learningcited by 71
- 2020Accelerating Ill-Conditioned Low-Rank Matrix Estimation via Scaled Gradient Descentcited by 70
- 2025Gluon: Making Muon & Scion Great Again! (Bridging Theory and Practice of LMO-based Optimizers for LLMs)cited by 66
- 2025ASGO: Adaptive Structured Gradient Optimizationcited by 55
- 2025PolarGrad: A Class of Matrix-Gradient Optimizers from a Unifying Preconditioning Perspectivecited by 48
- 2025Muon Outperforms Adam in Tail-End Associative Memory Learningcited by 38
- 2025Implicit Bias of Spectral Descent and Muon on Multiclass Separable Datacited by 33
- 2023The Power of Preconditioning in Overparameterized Low-Rank Matrix Sensingcited by 32
- 2017Expressive power of recurrent neural networkscited by 30
- 2025Isotropic Curvature Model for Understanding Deep Learning Optimization: Is Gradient Orthogonalization Optimal?cited by 30
- 2026Convergence of Muon with Newton-Schulzcited by 30
- 2026Preconditioning Benefits of Spectral Orthogonalization in Muoncited by 28
- 2020Low-Rank Matrix Recovery with Scaled Subgradient Methods: Fast and Robust Convergence Without the Condition Numbercited by 27
- 2022Orthogonalising gradients to speed up neural network optimisationcited by 25
- 2025When do spectral gradient updates help in deep learning?cited by 25
- 2025Convergence Bound and Critical Batch Size of Muon Optimizercited by 24
- 2025Beyond the Ideal: Analyzing the Inexact Muon Updatecited by 23
- 2022Preconditioned Gradient Descent for Overparameterized Nonconvex Burer--Monteiro Factorization with Global Optimality Certificationcited by 21
- 2025How Muon's Spectral Design Benefits Generalization: A Study on Imbalanced Datacited by 16
- 2026Probabilistic Tiny Recursive Modelcited by 16
- 2025On The Concurrence of Layer-wise Preconditioning Methods and Provable Feature Learningcited by 12
- 2025Preconditioned Gradient Descent for Over-Parameterized Nonconvex Matrix Factorizationcited by 12
- 2025Muon Optimizer Accelerates Grokkingcited by 11
- 2025Error Feedback for Muon and Friendscited by 11
- 2026Improved Convergence Rates of Muon Optimizer for Nonconvex Optimizationcited by 9
- 2026REAM: Merging Improves Pruning of Experts in LLMscited by 9
Built on by
Not cited yet.