Muon$^p$: Muon with Fractional Spectral Powers
About
Muon is an increasingly widely used optimizer that replaces a gradient $G=USV^\top$ with its polar factor $UV^\top$, thereby flattening the singular spectrum. However, full flattening discards singular-value information that may matter for adaptation. We introduce Muon$^p$, a Muon-style optimizer that instead uses fractional spectral-power updates $US^pV^\top$ for rational $p\in(0,1)$, interpolating between Muon and gradient descent. To make it practical, we prove that fractional spectral powers cannot be computed by any fixed univariate polynomial iteration, and furthermore derive low-degree odd bivariate recurrences that approximate $US^pV^\top$ using only matrix multiplications, preserving Muon's matrix-multiplication-only structure and compute complexity. We show that Muon$^p$ maximizes the linear improvement in loss under the Schatten $q$-norm for $q=1+\frac{1}{p}$. Empirically, Muon$^p$ is especially effective for finetuning: on billion-scale models, Muon$^p$ improves validation perplexity and downstream task performance. We further analyze when Muon$^p$ is less suitable, through the lens of spectral geometry. Our results reveal important insights on when preserving the singular spectrum can bring significant gains, and introduce a principled way to achieve them.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Commonsense Reasoning | HellaSwag | HellaSwag Accuracy48.18 | 897 | |
| Commonsense Reasoning | PIQA | Accuracy75.41 | 400 | |
| Language Modeling | FineWeb (val) | -- | 259 | |
| Question Answering | OpenBookQA | Accuracy28.2 | 157 | |
| Mathematical Reasoning | GSM8K (val) | -- | 115 | |
| Language Modeling | LAMBADA | Accuracy61.81 | 114 | |
| Coding | OpenCodeInstruct (val) | Loss0.2382 | 6 | |
| Mathematical Reasoning | Numina (val) | Pass@842.41 | 6 | |
| Mathematical Reasoning | Numina (test) | Pass@853.1 | 6 | |
| Language Modeling | Pile of Law (val) | Perplexity2.491 | 3 |