arXiv:2410.05106math.OCcs.LG2024-10ICLR被引 6

改进随机梯度下降的误差分析,实现更精确的收敛性评估

Nonasymptotic Analysis of Stochastic Gradient Descent with the Richardson-Romberg Extrapolation

  • 将SGD与理查森-罗姆伯格外推结合,提升估计精度
  • 首次给出均方误差的非渐近展开,含精确阶数项
  • 适用于需要高精度优化的机器学习任务

针对强凸光滑优化问题,研究使用固定步长的随机梯度下降(SGD)算法。已有工作提出将Polyak-Ruppert平均与理查森-罗姆伯格外推结合,以降低SGD的渐近偏差,仅小幅增加方差。本文显著扩展了前述结果,给出了估计器均方误差关于迭代次数 $n$ 的展开式。表明根均方误差可分解为两部分:主导项为 $ ext{O}(n^{-1/2})$,依赖于极小极大最优渐近协方差矩阵;二阶项为 $ ext{O}(n^{-3/4})$,其中 $3/4$ 为目前最优幂次。同时将结果推广至更高阶矩界。分析基于将SGD迭代视为时齐马尔可夫链,并证明其在特定加权沃尔什半度量下几何遍历。

原文摘要 · Abstract (English)

We address the problem of solving strongly convex and smooth minimization problems using stochastic gradient descent (SGD) algorithm with a constant step size. Previous works suggested to combine the Polyak-Ruppert averaging procedure with the Richardson-Romberg extrapolation to reduce the asymptotic bias of SGD at the expense of a mild increase of the variance. We significantly extend previous results by providing an expansion of the mean-squared error of the resulting estimator with respect to the number of iterations $n$. We show that the root mean-squared error can be decomposed into the sum of two terms: a leading one of order $\mathcal{O}(n^{-1/2})$ with explicit dependence on a minimax-optimal asymptotic covariance matrix, and a second-order term of order $\mathcal{O}(n^{-3/4})$, where the power $3/4$ is best known. We also extend this result to the higher-order moment bounds. Our analysis relies on the properties of the SGD iterates viewed as a time-homogeneous Markov chain. In particular, we establish that this chain is geometrically ergodic with respect to a suitably defined weighted Wasserstein semimetric.

优化算法随机梯度误差分析

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