To Use or not to Use Muon: How Simplicity Bias in Optimizers Matters
Not on the map. Why?
Abstract (arXiv)
While Adam has long been the ubiquitous default optimizer for deep neural networks, Muon has recently seen rapid adoption due to its superior training speed. Although much of the literature focuses on validating the benefits of Muon, our work investigates the potential downsides of the mechanism driving this speedup. On the theoretical front, we analyze the learning dynamics of simplified Muon on deep linear networks and linear attention. Our analysis reveals that Muon gains speed by avoiding saddle points, but does so at the expense of the simplicity bias characteristic of Gradient Descent (GD), where the complexity of the functional solution learned grows sequentially. Experiments demonstrate the consequences of losing the simplicity bias, showing that Muon struggles to uncover common underlying structure across tasks and may be prone to fitting spurious features. More broadly, this paper serves as a reminder that faster optimization is rarely a free lunch; improvements in optimization can come at the cost of changes in the inductive biases that shape generalization.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 46 of its 62 references that are in the map. First posted on arXiv 28 Feb 2026.
Builds on
46 of 62 references.
References · 46
- 2018On the Spectral Bias of Neural Networkscited by 1,321
- 2017A Closer Look at Memorization in Deep Networkscited by 936
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 748
- 2022Transformers learn in-context by gradient descentcited by 714
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
- 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 502
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 402
- 2020The Pitfalls of Simplicity Bias in Neural Networkscited by 349
- 2023Trained Transformers Learn Linear Models In-Contextcited by 331
- 2017High-dimensional dynamics of generalization error in neural networkscited by 238
- 2019SGD on Neural Networks Learns Functions of Increasing Complexitycited by 203
- 2018Gradient descent aligns the layers of deep linear networkscited by 182
34 more
- 2018A mathematical theory of semantic development in deep neural networkscited by 182
- 2018Deep learning generalizes because the parameter-function map is biased towards simple functionscited by 180
- 2020Towards Resolving the Implicit Bias of Gradient Descent for Matrix Factorization: Greedy Low-Rank Learningcited by 148
- 2023SGD learning on neural networks: leap complexity and saddle-to-saddle dynamicscited by 142
- 2019Implicit Regularization of Discrete Gradient Dynamics in Linear Neural Networkscited by 122
- 2024The Evolution of Statistical Induction Heads: In-Context Learning Markov Chainscited by 115
- 2021Neural Networks as Kernel Learners: The Silent Alignment Effectcited by 106
- 2025Fantastic Pretraining Optimizers and Where to Find Themcited by 87
- 2018An analytic theory of generalization dynamics and transfer learning in deep linear networkscited by 84
- 2021Evading the Simplicity Bias: Training a Diverse Set of Models Discovers Solutions with Superior OOD Generalizationcited by 83
- 2021Saddle-to-Saddle Dynamics in Deep Linear Networks: Small Initialization Training, Symmetry, and Sparsitycited by 83
- 2020The Surprising Simplicity of the Early-Time Learning Dynamics of Neural Networkscited by 58
- 2018Gradient Descent Quantizes ReLU Network Featurescited by 57
- 2024Asymptotic theory of in-context learning by linear attentioncited by 54
- 2023Transformers learn through gradual rank increasecited by 52
- 2022Neural networks trained with SGD learn distributions of increasing complexitycited by 50
- 2025Training Dynamics of In-Context Learning in Linear Attentioncited by 44
- 2024Dynamic metastability in the self-attention modelcited by 42
- 2021Out-of-distribution Generalization in the Presence of Nuisance-Induced Spurious Correlationscited by 41
- 2025Muon Outperforms Adam in Tail-End Associative Memory Learningcited by 38
- 2022The Neural Race Reduction: Dynamics of Abstraction in Gated Networkscited by 37
- 2023Don't blame Dataset Shift! Shortcut Learning due to Gradients and Cross Entropycited by 30
- 2025Isotropic Curvature Model for Understanding Deep Learning Optimization: Is Gradient Orthogonalization Optimal?cited by 30
- 2022Incremental Learning in Diagonal Linear Networkscited by 27
- 2022Exact Solutions of a Deep Linear Networkcited by 24
- 2024A distributional simplicity bias in the learning dynamics of transformerscited by 23
- 2022Nuisances via Negativa: Adjusting for Spurious Correlations via Data Augmentationcited by 18
- 2025How Muon's Spectral Design Benefits Generalization: A Study on Imbalanced Datacited by 16
- 2025Learning In-context n-grams with Transformers: Sub-n-grams Are Near-stationary Pointscited by 15
- 2025Saddle-to-Saddle Dynamics Explains A Simplicity Bias Across Neural Network Architecturescited by 14
- 2025The Rich and the Simple: On the Implicit Bias of Adam and SGDcited by 12
- 2026The Impact of Anisotropic Covariance Structure on the Training Dynamics and Generalization Error of Linear Networkscited by 2
- 2025Understanding the Staged Dynamics of Transformers in Learning Latent Structurecited by 1
- 2026Incremental Learning of Sparse Attention Patterns in Transformerscited by 1
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026The Newton-Muon Optimizercited by 14
- 2026Scale-Invariant Neural Network Optimization: Norm Geometry and Heavy-Tailed Noisecited by 2
- 2026MiMuon: Mixed Muon Optimizer with Improved Generalization for Large Modelscited by 2
- 2026Muon in Vision Transformers: Optimizer-Recipe Interactions and Gradient Spectracited by 1
- 2026From SGD to Muon: Adaptive Optimization via Schatten-p Normscited by 0
- 2026The Loss Does Not See the Basis, but Adam Doescited by 0
- 2026The Row Normalization Puzzle in Muoncited by 0
- 2026The Row Normalization Puzzle in Muon30 Sep 2026
- 2026The Loss Does Not See the Basis, but Adam Does5 Aug 2026
- 2026Muon in Vision Transformers: Optimizer-Recipe Interactions and Gradient Spectra23 May 2026
- 2026From SGD to Muon: Adaptive Optimization via Schatten-p Norms19 May 2026
- 2026MiMuon: Mixed Muon Optimizer with Improved Generalization for Large Models19 May 2026
- 2026Scale-Invariant Neural Network Optimization: Norm Geometry and Heavy-Tailed Noise18 May 2026
- 2026The Newton-Muon Optimizer1 Apr 2026