arXiv:2605.09552math.OCcs.LG2026-05

揭示了Muon在不同数据条件下表现差异的三阶段规律

Phases of Muon: When Muon Eclipses SignSGD

论文配图:Phases of Muon: When Muon Eclipses SignSGD
图 1 · 摘自论文原文
  • 通过解析推导,建立可解释的确定性动态模型
  • 发现大批次下SignSVD有平方根预条件,小批次则退化为SGD
  • 提出三相图解释不同数据分布下的方法优劣选择

近期,Muon等谱优化器在大规模随机优化中展现出优异的实证性能,常优于Adam。然而其行为机制仍不清晰。本文针对高维矩阵最小二乘问题,分析包括Muon在内的随机谱优化器,推导出显式的确定性动力学,聚焦于(随机)SignSVD(Muon近似)和(随机)SignSGD(作为Adam的代理)。分析表明:大批次时,SignSVD对数据协方差谱执行平方根预条件;小批次时,低频模式行为类似SGD,导致收敛变慢。而SignSGD在一般协方差下无预条件,且无转变,表现出不同的最优学习率与收敛特性。两者在各向同性数据下仅相差常数因子,但在各向异性数据下表现迥异。对幂律协方差模型(数据指数α,目标指数β)的分析揭示了(α, β)平面上存在三个相区:一区中SignSGD始终占优,一区中SignSVD始终占优,第三区中二者性能呈现权衡。

原文摘要 · Abstract (English)

Recently, Muon and related spectral optimizers have demonstrated strong empirical performance as scalable stochastic methods, often outperforming Adam. Yet their behaviour remains poorly understood. We analyze stochastic spectral optimizers, including Muon, on a high-dimensional matrix-valued least squares problem. We derive explicit deterministic dynamics that provide a tractable framework for studying learning behaviour with a focus on (stochastic) SignSVD, which Muon approximates, and (stochastic) SignSGD, the latter serving as a proxy for Adam. Our analysis shows that for large batch size, SignSVD performs a square-root preconditioning with respect to the data covariance spectrum, while for small batch size smaller eigenmodes behave like SGD, slowing down convergence. We contrast with SignSGD which for generic covariance performs no preconditioning and has no transition, leading to different optimal learning rates and convergence characteristics. The two methods match up to a constant factor with isotropic data, but behave differently with anisotropic data. An analysis of a power law covariance model with data exponent $α$ and target exponent $β$ shows there are three phases in the $(α,β)$ plane: one where SignSGD is uniformly favored, one where SignSVD is uniformly favored, and a third where the two methods exhibit a trade-off in performance.

优化算法谱方法收敛分析机器学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。