中文

随机二阶方法改进了SGD在梯度主导函数上的已知最优样本复杂度

机器学习 2023-01-24 v2 最优化与控制

摘要

我们研究了随机立方正则牛顿法(SCRN)在一类满足梯度主导性质(1α21\le\alpha\le2)的函数上的性能,该性质在机器学习和信号处理的广泛应用中成立。这一条件确保任何一阶驻点都是全局最优解。我们证明,SCRN在达到ϵ\epsilon-全局最优解时的总样本复杂度为:当1α<3/21\le\alpha< 3/2时为O(ϵ7/(2α)+1)\mathcal{O}(\epsilon^{-7/(2\alpha)+1}),当3/2α23/2\le\alpha\le 2时为O~(ϵ2/(α))\mathcal{\tilde{O}}(\epsilon^{-2/(\alpha)})。SCRN改进了随机梯度下降(SGD)的已知最优样本复杂度。即使在适用于基于策略的强化学习(RL)的弱版梯度主导性质下,SCRN也相对于随机策略梯度方法取得了同样的改进。此外,我们表明,通过使用具有时变批量大小的方差缩减方法,SCRN在α=1\alpha=1时的平均样本复杂度可降至O(ϵ2){\mathcal{O}}(\epsilon^{-2})。在各种RL设置下的实验结果显示,与一阶方法相比,SCRN具有卓越的性能。

关键词

引用

@article{arxiv.2205.12856,
  title  = {Stochastic Second-Order Methods Improve Best-Known Sample Complexity of SGD for Gradient-Dominated Function},
  author = {Saeed Masiha and Saber Salehkaleybar and Niao He and Negar Kiyavash and Patrick Thiran},
  journal= {arXiv preprint arXiv:2205.12856},
  year   = {2023}
}