arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.05437math.OC
收藏

重尾噪声下Clipped-SGD的高概率收敛性

High Probability Convergence of Clipped-SGD Under Heavy-tailed Noise

Ta Duy Nguyen, Thien Hang Nguyen, Alina Ene, Huy Le Nguyen

AI总结:

本文提出一种新颖的分析框架,在重尾噪声和最少假设下,为凸与非凸光滑目标函数的带梯度裁剪SGD给出了时间最优的高概率收敛界。

摘要:

尽管随机梯度方法的期望收敛行为已被充分理解,但在高概率收敛(即收敛速率对期望成功概率参数具有对数依赖性)的理解上仍存在许多空白。在重尾噪声设定下,即随机梯度噪声仅对于某个 p∈(1,2] 具有有界的 p 阶矩,现有研究仅能针对带梯度裁剪的随机梯度下降(SGD)变体给出期望界,或者仅在特殊情形(如 p=2)或附加假设(如随机梯度具有有界的非中心矩)下证明高概率界。在本研究中,借助一种新颖的分析框架,我们仅利用最少的假设,针对凸与非凸光滑目标函数,在重尾噪声下为带裁剪的 SGD 给出了新的、时间最优(至多相差对数因子)的高概率收敛界。

英文摘要:

While the convergence behaviors of stochastic gradient methods are well understood \emph{in expectation}, there still exist many gaps in the understanding of their convergence with \emph{high probability}, where the convergence rate has a logarithmic dependency on the desired success probability parameter. In the \emph{heavy-tailed noise} setting, where the stochastic gradient noise only has bounded $p$-th moments for some $p\in(1,2]$, existing works could only show bounds \emph{in expectation} for a variant of stochastic gradient descent (SGD) with clipped gradients, or high probability bounds in special cases (such as $p=2$) or with extra assumptions (such as the stochastic gradients having bounded non-central moments). In this work, using a novel analysis framework, we present new and time-optimal (up to logarithmic factors) \emph{high probability} convergence bounds for SGD with clipping under heavy-tailed noise for both convex and non-convex smooth objectives using only minimal assumptions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑