arXiv:2411.07724cs.LGmath.OC2024-11被引 13

LION优化器收敛速度达到理论最优,且实证验证了其性能优势。

Convergence Rate Analysis of LION

  • 将LION视为带约束优化问题,证明其以√d·K⁻¹⁄⁴速率收敛至KKT点
  • 在无约束情况下仍保持相同收敛速率,且匹配非凸随机优化下界
  • 实验验证梯度范数比≈√d,证实理论分析的合理性,适合优化研究者

LION(evoLved sIgn mOmeNtum)是谷歌通过程序搜索发现的深度神经网络训练优化器,虽仅采用简单符号更新,却在大规模网络训练中表现出色。尽管已有研究探讨其收敛性,但对其收敛速率的全面分析仍不足。本文指出LION可被视作求解特定约束问题,并证明其以$ m O( extstyle rac{ ext{sqrt}{d}}{K^{1/4}})$的速率收敛至卡鲁什-库恩-塔克(KKT)点,其中$d$为问题维度,$K$为迭代步数。进一步地,在去除约束后,同样证明其以相同速率收敛至一般无约束问题的临界点。该速率不仅在问题维度$d$上达到当前最优,且与通常以梯度$ m orm{ abla f}_2$衡量的非凸随机优化算法理论下界紧密吻合。通过大量实验,验证了LION相较标准SGD能获得更低损失和更高性能;同时,实测梯度$ m orm{ abla f}_1 / orm{ abla f}_2$比值约为$Θ( extstyle rac{ ext{sqrt}{d}}{})$,从经验上证实了收敛速率在维度上的理论最优性。

原文摘要 · Abstract (English)

The LION (evoLved sIgn mOmeNtum) optimizer for deep neural network training was found by Google via program search, with the simple sign update yet showing impressive performance in training large scale networks. Although previous studies have investigated its convergence properties, a comprehensive analysis, especially the convergence rate, is still desirable. Recognizing that LION can be regarded as solving a specific constrained problem, this paper focuses on demonstrating its convergence to the Karush-Kuhn-Tucker (KKT) point at the rate of $\cal O(\sqrt{d}K^{-1/4})$ measured by gradient $\ell_1$ norm, where $d$ is the problem dimension and $K$ is the number of iteration steps. Step further, we remove the constraint and establish that LION converges to the critical point of the general unconstrained problem at the same rate. This rate not only delivers the currently optimal dependence on the problem dimension $d$ but also tightly matches the theoretical lower bound for nonconvex stochastic optimization algorithms, which is typically measured using the gradient $\ell_2$ norm, with respect to the number of iterations $K$. Through extensive experiments, we not only demonstrate that LION achieves lower loss and higher performance compared to standard SGD, but also empirically confirm that the gradient $\ell_1/\ell_2$ norm ratio aligns with $Θ(\sqrt{d})$, thus proving that our convergence rate matches the theoretical lower bound with respect to $d$ in the empirical sense.

优化器收敛分析深度学习

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