mlsanityData to 14 Oct 2026

An Isotropy-Preserving Spectral Cap for Muon: Theory and Three Case Studies

arXiv 2607.19771 ↗First posted 22 Jul 2026cs.LG cs.AIcited by 0

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

  1. 2025Muon is Scalable for LLM Trainingcited by 428
  2. 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

  1. 2026SignMuon: Communication-Efficient Distributed Muon Optimization
  2. 2026Muon Meets Mamba: Spectral Optimization for State Space Models
  3. 2026Training-Loss Guarantees for Muon with Finite-Step Newton--Schulz Orthogonalization
  4. 2026Orthogonal Witness Control for Muon Optimization via Sigmoid Spectral Reshaping
  5. 2026Rethinking Muon Beyond Pretraining: Spectral Failures and High-Pass Remedies for VLA and RLVR
  6. 2025Muon: Training and Trade-offs with Latent Attention and MoE
  7. 2026Move on Muon : A Hamiltonian probability gradient flow perspective of Muon optimizer
  8. 2026Scaling Muon for Diffusion Transformers