arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.05359cs.LGmath.OCstat.ML

Adaptivity of Stochastic Gradient Methods for Nonconvex Optimization

KAUST(阿卜杜拉国王科技大学);Stanford University(斯坦福大学);University of California, Berkeley(加州大学伯克利分校)

Samuel Horváth, Lihua Lei, Peter Richtárik, Michael I. Jordan

英文摘要:

Adaptivity is an important yet under-studied property in modern optimization theory. The gap between the state-of-the-art theory and the current practice is striking in that algorithms with desirable theoretical guarantees typically involve drastically different settings of hyperparameters, such as step-size schemes and batch sizes, in different regimes. Despite the appealing theoretical results, such divisive strategies provide little, if any, insight to practitioners to select algorithms that work broadly without tweaking the hyperparameters. In this work, blending the "geometrization" technique introduced by Lei & Jordan 2016 and the \texttt{SARAH} algorithm of Nguyen et al., 2017, we propose the Geometrized \texttt{SARAH} algorithm for non-convex finite-sum and stochastic optimization. Our algorithm is proved to achieve adaptivity to both the magnitude of the target accuracy and the Polyak-Łojasiewicz (PL) constant if present. In addition, it achieves the best-available convergence rate for non-PL objectives simultaneously while outperforming existing algorithms for PL objectives.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑