中文

利用随机平均梯度最小化有限和

最优化与控制 2016-05-12 v2 机器学习 统计计算 机器学习

摘要

我们提出了随机平均梯度 (SAG) 方法,用于优化有限个光滑凸函数之和。与随机梯度 (SG) 方法类似,SAG 方法的迭代成本与和式中的项数无关。然而,通过利用先前梯度值的记忆,SAG 方法实现了比黑盒 SG 方法更快的收敛速率。一般而言,收敛速率从 O(1/k1/2)O(1/k^{1/2}) 提升至 O(1/k)O(1/k);当和式为强凸时,收敛速率从次线性的 O(1/k)O(1/k) 提升至 O(pk)O(p^k) 形式的线性收敛速率(其中 p<1p < 1)。此外,在许多情况下,就梯度评估次数而言,新方法的收敛速率也快于黑盒确定性梯度方法。数值实验表明,新算法通常显著优于现有的 SG 和确定性梯度方法,且通过使用非均匀采样策略可进一步提升性能。

关键词

引用

@article{arxiv.1309.2388,
  title  = {Minimizing Finite Sums with the Stochastic Average Gradient},
  author = {Mark Schmidt and Nicolas Le Roux and Francis Bach},
  journal= {arXiv preprint arXiv:1309.2388},
  year   = {2016}
}

备注

Revision from January 2015 submission. Major changes: updated literature follow and discussion of subsequent work, additional Lemma showing the validity of one of the formulas, somewhat simplified presentation of Lyapunov bound, included code needed for checking proofs rather than the polynomials generated by the code, added error regions to the numerical experiments