An Isotropy-Preserving Spectral Cap for Muon: Theory and Three Case Studies
Not on the map. Why?
Abstract (arXiv)
Muon and related matrix-sign optimizers are increasingly used to pre-train large language models, but their effect on the internal geometry of individual weight matrices is not well understood. This preliminary report proposes a unified framework built on a single idealizing assumption -- exact scale invariance of the loss under weight rescaling, which holds approximately in normalization-heavy networks. Under this assumption, plain SGD carries a built-in 1/||W|| brake on its update size, whereas Muon's matrix-sign step removes that brake, so both the Frobenius and spectral norms drift outward faster (t^{1/2} versus t^{1/4}). We further observe that the spectral-norm perturbation has a non-negative second-order term. This implies that a lightweight "spectral cap" -- which projects out only the first-order growth of the single top singular direction from each update -- can control the output covariance W K_X W^T without freezing training: the weight keeps learning through non-top directions, top-direction rotation, and top switching. We relate this cap to the min-entropy (H-infinity) of the singular-value spectrum. We then study three systems trained with Muon: a nanoGPT feed-forward projection, a 64-expert mixture-of-experts router, and the query/key projections of a bf16 FlashAttention block. In each case the cap increases isotropy and, at the margins -- a router collapsing to a single expert, and the near-divergence of one attention head -- prevents a concrete failure, while leaving validation loss essentially unchanged. We emphasize that the scale-invariance assumption is strong and that these small-scale results are preliminary; comments are welcome.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 2 of its 8 references that are in the map. First posted on arXiv 22 Jul 2026.
Builds on
2 of 8 references.
References · 2
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2025Why Low-Precision Transformer Training Fails: An Analysis on Flash Attentioncited by 12
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026SignMuon: Communication-Efficient Distributed Muon Optimization
- 2026Muon Meets Mamba: Spectral Optimization for State Space Models
- 2026Training-Loss Guarantees for Muon with Finite-Step Newton--Schulz Orthogonalization
- 2026Orthogonal Witness Control for Muon Optimization via Sigmoid Spectral Reshaping
- 2026Rethinking Muon Beyond Pretraining: Spectral Failures and High-Pass Remedies for VLA and RLVR
- 2025Muon: Training and Trade-offs with Latent Attention and MoE
- 2026Move on Muon : A Hamiltonian probability gradient flow perspective of Muon optimizer
- 2026Scaling Muon for Diffusion Transformers