中文

梯度下降与梯度流的路径长度界

机器学习 2021-07-20 v4 人工智能 最优化与控制 机器学习

摘要

我们针对各类光滑凸和非凸函数,推导了梯度下降(GD)和梯度流(GF)曲线的路径长度 ζ\zeta 的界。除其他结果外,我们证明:(a) 若迭代以因子 (1c)(1-c) 线性收敛,则 ζ\zeta 至多为 O(1/c)\mathcal{O}(1/c);(b) 在 Polyak-Kurdyka-Lojasiewicz(PKL)条件下,ζ\zeta 至多为 O(κ)\mathcal{O}(\sqrt{\kappa}),其中 κ\kappa 为条件数,且至少为 Ω~(dκ1/4)\widetilde\Omega(\sqrt{d} \wedge \kappa^{1/4});(c) 对于二次函数,ζ\zetaΘ(min{d,logκ})\Theta(\min\{\sqrt{d},\sqrt{\log \kappa}\}),且在某些情况下可与 κ\kappa 无关;(d) 仅假设凸性时,ζ\zeta 至多为 24dlogd2^{4d\log d};(e) 对于可分离拟凸函数,ζ\zetaΘ(d){\Theta}(\sqrt{d})。因此,我们推进了当前对 GD 和 GF 曲线性质超越收敛速率的理解。我们期望我们的技术能促进未来对其他算法的研究。

关键词

引用

@article{arxiv.1908.01089,
  title  = {Path Length Bounds for Gradient Descent and Flow},
  author = {Chirag Gupta and Sivaraman Balakrishnan and Aaditya Ramdas},
  journal= {arXiv preprint arXiv:1908.01089},
  year   = {2021}
}

备注

55 pages. Accepted for publication at the Journal of Machine Learning Research (JMLR, 2021)