中文

带 Richardson-Romberg 外推的随机梯度下降非渐近分析

最优化与控制 2025-08-08 v3 机器学习 机器学习

摘要

我们解决使用常数步长的随机梯度下降(SGD)算法来求解强凸光滑最小化问题的难题。以前的工作建议将 Polyak-Ruppert 平均程序与 Richardson-Romberg 外推相结合,以在轻微增加方差的同时减少 SGD 的渐近偏差。我们显著扩展了前面的结果,通过对 resulting estimator 相对于迭代次数 n 的均方误差提供了展开。我们表明,根均方误差可分解为两个术语:一个阶数为 O(n1/2)\mathcal{O}(n^{-1/2}) 的主导项,其对明确的 minimax 最优渐近协方差矩阵有显式依赖,以及阶数为 O(n3/4)\mathcal{O}(n^{-3/4}) 的次项,其中指数 3/43/4 是目前已知的最佳值。我们还将此结果推广到更高阶矩的界限。我们的分析依赖于将 SGD 迭代视为时齐马尔可夫链的属性。特别是,我们在 suitably 定义的加权 Wasserstein semimetric 下证明了该链具有几何 Ergodic。

关键词

引用

@article{arxiv.2410.05106,
  title  = {Nonasymptotic Analysis of Stochastic Gradient Descent with the Richardson-Romberg Extrapolation},
  author = {Marina Sheshukova and Denis Belomestny and Alain Durmus and Eric Moulines and Alexey Naumov and Sergey Samsonov},
  journal= {arXiv preprint arXiv:2410.05106},
  year   = {2025}
}

备注

ICLR-2025, camera-ready version. Some typos and definitions of constants have been fixed in the appendix