arXiv:2602.05657cs.LGmath.OC2026-02

揭示了非凸优化中SGD长期尾部衰减速率,比现有结果快一个数量级。

Tight Long-Term Tail Decay of (Clipped) SGD in Non-Convex Optimization

  • 基于大偏差理论分析梯度模平方尾部,给出长期衰减上界。
  • 在重尾噪声下,剪裁SGD尾部衰减率达$e^{-t^{β_p}/\log(t)}$,优于旧方法。
  • 证明下界为$e^{-t}$,表明上界紧致,适合关注长期稳定性的研究者。

关于随机梯度下降(SGD)诱导过程的尾部行为研究近年来备受关注,因其能为算法单次运行提供强保证。尽管已有大量工作给出高概率误差界,但对固定失败概率的尾部衰减速率缺乏直接分析。此外,现有成果多为有限时间结果,难以捕捉现代学习模型(通常训练数百万轮)的真实长期尾部衰减特性。本文通过大偏差理论,填补这些空白,首次系统研究了基于SGD方法的长期尾部衰减。首先,在非凸目标函数与有界噪声条件下,给出原始SGD最优迭代梯度模平方尾部的上界,长期衰减速率为$e^{-t/\log(t)}$。其次,在更弱的重尾噪声假设下(阶数为$ p \in (1,2] $的有界矩),考虑剪裁SGD(c-SGD),得到上界衰减速率为$e^{-t^{β_p}/\log(t)}$,其中$β_p = \frac{4(p-1)}{3p-2}$($p \in (1,2)$)且当$ p=2 $时为$e^{-t/\log^2(t)}$。最后,我们建立下界$e^{-t}$,证明上述上界仅差多项式对数因子,即紧致。值得注意的是,我们的结果在长期下比基于有限时间界的方法(如$e^{-\sqrt{t}}$和$e^{-t^{β_p/2}}$)快一个数量级,揭示出此前未被认识的更快衰减区域,为单次运行提供了更强的长期保证。

原文摘要 · Abstract (English)

The study of tail behaviour of SGD-induced processes has been attracting a lot of interest, due to offering strong guarantees with respect to individual runs of an algorithm. While many works provide high-probability guarantees, quantifying the error rate for a fixed probability threshold, there is a lack of work directly studying the probability of failure, i.e., quantifying the tail decay rate for a fixed error threshold. Moreover, existing results are of finite-time nature, limiting their ability to capture the true long-term tail decay which is more informative for modern learning models, typically trained for millions of iterations. Our work closes these gaps, by studying the long-term tail decay of SGD-based methods through the lens of large deviations theory, establishing several strong results in the process. First, we provide an upper bound on the tails of the gradient norm-squared of the best iterate produced by (vanilla) SGD, for non-convex costs and bounded noise, with long-term decay at rate $e^{-t/\log(t)}$. Next, we relax the noise assumption by considering clipped SGD (c-SGD) under heavy-tailed noise with bounded moment of order $p \in (1,2]$, showing an upper bound with long-term decay at rate $e^{-t^{β_p}/\log(t)}$, where $β_p = \frac{4(p-1)}{3p-2}$ for $p \in (1,2)$ and $e^{-t/\log^2(t)}$ for $p = 2$. Finally, we provide lower bounds on the tail decay, at rate $e^{-t}$, showing that our rates for both SGD and c-SGD are tight, up to poly-logarithmic factors. Notably, our results demonstrate an order of magnitude faster long-term tail decay compared to existing work based on finite-time bounds, which show rates $e^{-\sqrt{t}}$ and $e^{-t^{β_p/2}}$, $p \in (1,2]$, for SGD and c-SGD, respectively. As such, we uncover regimes where the tails decay much faster than previously known, providing stronger long-term guarantees for individual runs.

优化理论随机梯度尾部衰减大偏差

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