mlsanityData to 7 Oct 2026

Delving into Muon and Beyond: Deep Analysis and Extensions

arXiv 2602.04669 ↗First posted 4 Feb 2026cs.LG cs.AI cs.CLcited by 9

Not on the map. Why?

Abstract (arXiv)

The Muon optimizer has recently attracted considerable attention for its strong empirical performance and use of orthogonalized updates on matrix-shaped parameters, yet its underlying mechanisms and relationship to adaptive optimizers such as Adam remain insufficiently understood. In this work, we aim to address these questions through a unified spectral perspective. Specifically, we view Muon as the p = 0 endpoint of a family of spectral transformations of the form U boldsymbolSigmap V' , and consider additional variants with p = 1/2 , p = 1/4 , and p = 1 . These transformations are applied to both first-moment updates, as in momentum SGD, and to root-mean-square (RMS) normalized gradient updates as in Adam. To enable efficient computation, we develop a coupled Newton iteration that avoids explicit singular value decomposition. Across controlled experiments, we find that RMS-normalized updates yield more stable optimization than first-moment updates. Moreover, while spectral compression provides strong stabilization benefits under first-moment updates, the Muon update (p = 0) does not consistently outperform Adam. These results suggest that Muon is best understood as an effective form of spectral normalization, but not a universally superior optimization method. Our source code will be released at https://github.com/Ocram7/BeyondMuon.

Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Builds on 22 of its 46 references that are in the map. First posted on arXiv 4 Feb 2026.

Builds on

22 of 46 references.

References · 22

  1. 2017Attention Is All You Needcited by 59,932
  2. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  3. 2024The Llama 3 Herd of Modelscited by 14,192
  4. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  5. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  6. 2016Layer Normalizationcited by 5,579
  7. 2020GLU Variants Improve Transformercited by 1,724
  8. 2016Optimization Methods for Large-Scale Machine Learningcited by 1,140
  9. 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 896
  10. 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
  11. 2019Large Batch Optimization for Deep Learning: Training BERT in 76 minutescited by 671
  12. 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 561
10 more
  1. 2025Kimi K2: Open Agentic Intelligencecited by 523
  2. 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 260
  3. 2022Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Modelscited by 172
  4. 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 134
  5. 2025Fantastic Pretraining Optimizers and Where to Find Themcited by 87
  6. 2024MARS: Unleashing the Power of Variance Reduction for Training Large Modelscited by 59
  7. 2025PolarGrad: A Class of Matrix-Gradient Optimizers from a Unifying Preconditioning Perspectivecited by 48
  8. 2024Cautious Optimizers: Improving Training with One Line of Codecited by 42
  9. 2025Isotropic Curvature Model for Understanding Deep Learning Optimization: Is Gradient Orthogonalization Optimal?cited by 30
  10. 2025What Really Matters in Matrix-Whitening Optimizers?cited by 15

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 9.

9 papers.