English

MuCon: Clipped Muon Updates for LLM Training

Machine Learning 2026-05-27 v1

Abstract

Muon-style optimizers take a matrix-valued momentum or preconditioned update B=Udiag(σ1,,σr)VB = U \operatorname{diag}(\sigma_1,\ldots,\sigma_r) V^\top and replace it with its canonical partial polar factor Pol(B)=UV\operatorname{Pol}(B) = U V^\top. This maps every nonzero singular value to one. MuCon is the clipped-Muon variant studied here: it applies singular-value clipping to the same Muon matrix, DMuCon_τ(B)=MClip_τ(B)=Udiag(min{σ_i,τ})V,τ>0D^{\mathrm{MuCon}}\_\tau(B) = \operatorname{MClip}\_\tau(B) = U \operatorname{diag}\bigl(\min\{\sigma\_i,\tau\}\bigr) V^\top, \qquad \tau > 0. Thus, MClip_τ\operatorname{MClip}\_\tau denotes the mathematical clipping operator, while MuCon denotes the optimizer primitive that substitutes this clipped direction for Muon's polar direction. The Muon/MuCon scaling parameterization used in this work is called SpectralP\text{SpectralP}: it is the hidden-matrix scaling recipe under which polar Muon or clipped MuCon directions are applied. The map MClip_τ\operatorname{MClip}\_\tau is the Frobenius projection onto the spectral-norm ball {X:X2τ}\{X : \|X\|_2 \le \tau\}: it leaves singular values at or below τ\tau unchanged and modifies only the violating singular directions. This paper asks when the MuCon clipping step can be approximated without a full dense SVD. We record two exact identities, a polar/absolute-value formula and a scalar-root formulation leading to a rational Newton filter for the clipped positive-semidefinite factor, and identify the numerical obstruction common to both: singular values near the threshold make sign decisions and rational solves ill-conditioned. Matrix-function methods are therefore useful only when paired with stable polar/square-root primitives or explicit regularization near the clipping boundary.

Keywords

Cite

@article{arxiv.2605.26459,
  title  = {MuCon: Clipped Muon Updates for LLM Training},
  author = {Albert Yi},
  journal= {arXiv preprint arXiv:2605.26459},
  year   = {2026}
}
R2 v1 2026-07-22T07:33:38.367Z