arXiv:2410.15637cs.LGmath.OC2024-10被引 6

提出非线性SGD在重尾噪声下的统一误差分析框架,突破传统方法限制。

Large Deviation Upper Bounds and Improved MSE Rates of Nonlinear SGD: Heavy-tailed Noise and Power of Symmetry

  • 采用黑箱方式处理非线性映射,统一涵盖截断、量化等常见操作。
  • 在非凸目标下实现最优均方误差率 $\widetilde{\mathcal{O}}(t^{-1/2})$,重尾噪声下收敛更快。
  • 适用于对噪声分布对称性有要求的场景,适合研究鲁棒优化与分布式学习者。

本文研究在线设置下一类广义非线性随机梯度方法在重尾噪声下的大偏差上界与均方误差(MSE)保证。不同于依赖非线性闭式表达(如截断)的现有工作,本框架将非线性视为黑箱,可统一覆盖多种有界非线性函数,包括符号函数、量化、归一化以及分量或联合截断。针对具有对称概率密度函数、在零邻域为正且矩可能无界的重尾噪声,我们给出一系列强结果。对于非凸损失函数,建立了梯度范数平方最小值的大偏差上界,其尾部衰减呈指数级,速率可达 $\sqrt{t}/\log(t)$,并明确给出了依赖步长、非线性、噪声及问题参数的速率函数。进一步地,推导出非凸情况下梯度范数平方最小值的最优均方误差率 $\widetilde{\mathcal{O}}(t^{-1/2})$。对于强凸损失函数与最终迭代点,所获均方误差率可任意逼近最优率 $\mathcal{O}(t^{-1})$,优于现有重尾噪声下的最优结果。最后,证明了梯度范数平方最小值几乎必然收敛,且可使收敛速率任意接近 $o(t^{-1/4})$。

原文摘要 · Abstract (English)

We study large deviation upper bounds and mean-squared error (MSE) guarantees of a general framework of nonlinear stochastic gradient methods in the online setting, in the presence of heavy-tailed noise. Unlike existing works that rely on the closed form of a nonlinearity (typically clipping), our framework treats the nonlinearity in a black-box manner, allowing us to provide unified guarantees for a broad class of bounded nonlinearities, including many popular ones, like sign, quantization, normalization, as well as component-wise and joint clipping. We provide several strong results for a broad range of step-sizes in the presence of heavy-tailed noise with symmetric probability density function, positive in a neighbourhood of zero and potentially unbounded moments. In particular, for non-convex costs we provide a large deviation upper bound for the minimum norm-squared of gradients, showing an asymptotic tail decay on an exponential scale, at a rate $\sqrt{t} / \log(t)$. We establish the accompanying rate function, showing an explicit dependence on the choice of step-size, nonlinearity, noise and problem parameters. Next, for non-convex costs and the minimum norm-squared of gradients, we derive the optimal MSE rate $\widetilde{\mathcal{O}}(t^{-1/2})$. Moreover, for strongly convex costs and the last iterate, we provide an MSE rate that can be made arbitrarily close to the optimal rate $\mathcal{O}(t^{-1})$, improving on the state-of-the-art results in the presence of heavy-tailed noise. Finally, we establish almost sure convergence of the minimum norm-squared of gradients, providing an explicit rate, which can be made arbitrarily close to $o(t^{-1/4})$.

非线性SGD重尾噪声均方误差收敛分析

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