arXiv:2410.13954cs.LGmath.OC2024-10被引 3

提出统一框架,让非线性随机梯度下降在重尾噪声下仍能高概率收敛。

Nonlinear Stochastic Gradient Descent and Heavy-tailed Noise: A Unified Framework and High-probability Guarantees

  • 将多种非线性方法视为黑箱,建立统一收敛保证。
  • 非凸时梯度范数平方以t^{-1/4}率收敛,强凸时最后迭代点趋近最优解。
  • 无需假设噪声矩,适用于更广泛场景,适合研究在线学习鲁棒性者。

我们研究了存在重尾噪声时在线学习的高概率收敛性。为应对重尾问题,考虑了一类广义的非线性SGD方法,涵盖符号、量化、分量与联合截断等多种常见非线性形式。本文将非线性处理为黑箱,从而对一大类方法建立了统一的收敛保证。对于对称噪声和非凸损失函数,我们证明梯度范数平方以$ ilde{ ext{O}}(t^{-1/4})$速率收敛;对于强凸损失函数,最后迭代点以$ ext{O}(t^{-ζ})$速率收敛至总体最优解,其中$ζ o (0,1)$依赖于噪声与问题参数。若噪声为对称与非对称成分的(有偏)混合,则收敛至平稳点邻域,其大小取决于混合系数、非线性形式与噪声特性。相比现有工作仅考虑截断且要求无偏噪声具有有限$p$阶矩($p o (1,2]$),本工作无需任何关于噪声矩的假设,覆盖更广。当$p < 6/5$(非凸)或$p < 8/7$(强凸)时,我们的速率指数恒定且严格优于现有结果。实验验证了理论,表明截断并非总是最优选择,凸显通用框架的价值。

原文摘要 · Abstract (English)

We study high-probability convergence in online learning, in the presence of heavy-tailed noise. To combat the heavy tails, a general framework of nonlinear SGD methods is considered, subsuming several popular nonlinearities like sign, quantization, component-wise and joint clipping. In our work the nonlinearity is treated in a black-box manner, allowing us to establish unified guarantees for a broad range of nonlinear methods. For symmetric noise and non-convex costs we establish convergence of gradient norm-squared, at a rate $\widetilde{\mathcal{O}}(t^{-1/4})$, while for the last iterate of strongly convex costs we establish convergence to the population optima, at a rate $\mathcal{O}(t^{-ζ})$, where $ζ\in (0,1)$ depends on noise and problem parameters. Further, if the noise is a (biased) mixture of symmetric and non-symmetric components, we show convergence to a neighbourhood of stationarity, whose size depends on the mixture coefficient, nonlinearity and noise. Compared to state-of-the-art, who only consider clipping and require unbiased noise with bounded $p$-th moments, $p \in (1,2]$, we provide guarantees for a broad class of nonlinearities, without any assumptions on noise moments. While the rate exponents in state-of-the-art depend on noise moments and vanish as $p \rightarrow 1$, our exponents are constant and strictly better whenever $p < 6/5$ for non-convex and $p < 8/7$ for strongly convex costs. Experiments validate our theory, showing that clipping is not always the optimal nonlinearity, further underlining the value of a general framework.

优化算法在线学习重尾噪声非线性SGD

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