arXiv:2605.06615cs.LGcs.AI2026-05被引 2

在稀疏噪声下,SignSGD比SGD快d倍,理论首次解释其优势。

When and Why SignSGD Outperforms SGD: A Theoretical Study Based on $\ell_1$-norm Lower Bounds

  • 用ℓ₁-范数和坐标独立噪声建模,突破传统理论限制。
  • 证明SignSGD复杂度降低d倍,当噪声稀疏时显著优于SGD。
  • 实测124M参数GPT-2预训练,理论预测与实际收敛速度一致。

基于符号的优化算法(如SignSGD和Muon)在训练大规模基础模型中表现出色,但其理论机制仍不明确。传统条件下,SGD在ℓ₂-范数意义下已是最优,难以解释符号方法的优势。本文通过引入ℓ₁-范数平稳性、ℓ∞-光滑性和可分离噪声模型,刻画了坐标式更新的本质特征。在此新几何设定下,我们为SignSGD推导出匹配的上下界,并明确其在稀疏噪声下能以因子d(问题维度)降低复杂度,从而严格超越SGD。进一步将框架扩展至矩阵域,给出Muon优化器的等效最优下界,证明矩阵符号操作保持维度最优缩放。最后,实证表明该理论预测准确反映了124M参数GPT-2模型预训练中的更快收敛行为。

原文摘要 · Abstract (English)

Sign-based optimization algorithms, such as SignSGD and Muon, have garnered significant attention for their remarkable performance in training large foundation models. Despite this empirical success, we still lack a theoretical understanding of when and why these sign-based methods outperform vanilla SGD. The core obstacle is that under standard smoothness and finite variance conditions, SGD is known to be minimax optimal for finding stationary points measured by $\ell_2$-norms, thereby fundamentally precluding any complexity gains for sign-based methods in standard settings. To overcome this barrier, we analyze sign-based optimizers leveraging $\ell_1$-norm stationarity, $\ell_\infty$-smoothness, and a separable noise model, which can better capture the coordinate-wise nature of signed updates. Under this distinct problem geometry, we derive matched upper and lower bounds for SignSGD and explicitly characterize the problem class in which SignSGD provably dominates SGD. Specifically, we compare the \emph{upper bound of SignSGD} with the \emph{lower bound of SGD}, illustrating that SignSGD effectively reduces the complexity by a factor of $d$ under \emph{sparse noise}, where $d$ is the problem dimension. Furthermore, we elevate this framework to the matrix domain, providing an equivalent optimal lower bound for the Muon optimizer, proving that extending the sign operator to matrices preserves this optimal scaling with dimensionality. Finally, we bridge our theoretical bounds to practice, demonstrating that the theoretical superiority of SignSGD accurately predicts its faster convergence during the pretraining of a 124M parameter GPT-2 model.

优化算法SignSGD理论分析稀疏噪声

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