随机二阶方法改进了SGD在梯度主导函数上的已知最优样本复杂度
机器学习
2023-01-24 v2 最优化与控制
摘要
我们研究了随机立方正则牛顿法(SCRN)在一类满足梯度主导性质()的函数上的性能,该性质在机器学习和信号处理的广泛应用中成立。这一条件确保任何一阶驻点都是全局最优解。我们证明,SCRN在达到-全局最优解时的总样本复杂度为:当时为,当时为。SCRN改进了随机梯度下降(SGD)的已知最优样本复杂度。即使在适用于基于策略的强化学习(RL)的弱版梯度主导性质下,SCRN也相对于随机策略梯度方法取得了同样的改进。此外,我们表明,通过使用具有时变批量大小的方差缩减方法,SCRN在时的平均样本复杂度可降至。在各种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}
}