如何捕获梯度流
最优化与控制
2021-01-01 v3 机器学习
摘要
我们考虑在 的紧致域上寻找光滑函数的 -近似驻点的问题。与梯度下降等无关维数的方法不同,我们此处关注 有限且可能较小的情形。该视角由 Vavasis 于 1993 年探讨,他提出了一种算法,对任意固定有限维 ,其优于梯度下降的 预言机复杂度。例如 时,Vavasis 方法得到复杂度 。此外对于 他还证明了确定性算法的下界 (我们将其推广至随机算法)。我们的主要贡献是一种称为梯度流捕获(GFT)的算法,并分析了其预言机复杂度。在维数 时,GFT 填补了与 Vavasis 下界的差距(至多相差对数因子),因为我们证明其复杂度为 。在维数 时,我们展示出复杂度 ,优于 Vavasis 的 。在更高维时,GFT 具有对数并行深度策略的显著性质,与梯度下降或 Vavasis 算法的多项式深度形成鲜明对比。在此高维情形下,GFT 的总工作量相对于该问题另一种已知的仅有多对数深度策略(即朴素网格搜索)有二次方的改进。我们以另一种名为 \emph{cut and flow}(CF)的算法补充该结果,它在任意固定维数下均优于 Vavasis 算法。
引用
@article{arxiv.2001.02968,
title = {How to trap a gradient flow},
author = {Sébastien Bubeck and Dan Mikulincer},
journal= {arXiv preprint arXiv:2001.02968},
year = {2021}
}
备注
25 pages, 5 figures. Added an improved algorithm for dimensions > 3