中文

加速方差缩减随机梯度方法

最优化与控制 2020-10-30 v3

摘要

方差缩减是改善随机梯度下降缓慢收敛的关键工具。然而,仅有少数方差缩减方法已被证明能直接受益于 Nesterov 加速技术,以匹配加速梯度方法的收敛速率。此类方法依赖于“负动量”,一种进一步方差缩减的技术,通常特定于 SVRG 梯度估计量。本工作中,我们表明负动量对于加速是不必要的,并开发了一个通用加速框架,使所有流行的方差缩减方法都能达到加速收敛速率。这些速率中出现的常数,包括其对函数个数 nn 的依赖性,随梯度估计量的均方误差和偏差而缩放。在一系列数值实验中,我们展示了使用我们框架的 SAGA、SVRG、SARAH 和 SARGE 的版本显著优于非加速版本,并与使用负动量的算法相比具有优势。

关键词

引用

@article{arxiv.1910.09494,
  title  = {Accelerating Variance-Reduced Stochastic Gradient Methods},
  author = {Derek Driggs and Matthias J. Ehrhardt and Carola-Bibiane Schönlieb},
  journal= {arXiv preprint arXiv:1910.09494},
  year   = {2020}
}

备注

33 pages