学习率衰减可指数级加速SGD实现非凸函数全局优化
Learning Rate Decay Can Exponentially Accelerate SGD for Global Optimization of Nonconvex Functions
Guneykan Ozgul, Dylan Herman, Junhyung Lyle Kim, Shree Hari Sureshbabu, Shouvanik Chakrabarti
AI 速读 基于摘要
作者证明在一类自然非凸函数上,带学习率衰减的SGD在小步长情形下实现全局优化所需的梯度查询次数比任何固定学习率SGD指数级更少。
math.OC · cs.DS · cs.LG
查看原文摘要
Stochastic Gradient Descent (SGD) is a workhorse algorithm for continuous optimization. It is a folk belief among deep learning practitioners that learning rate decay in SGD substantially improves performance. However, it has remained unclear whether learning rate decay offers an asymptotic advantage for global nonconvex optimization. We answer this question in the affirmative by demonstrating a natural class of nonconvex functions for which, we prove that SGD with learning rate decay requires exponentially fewer gradient queries for global optimization than SGD with any fixed learning rate in the small-step size regime. To the best of our knowledge, this is the first exponential advantage that has been demonstrated for learning rate decay. The class of functions we consider includes many popular benchmark nonconvex functions. Our results provide a robust explanatory theory for the empirically observed benefits of learning rate decay for global optimization. We show that annealing performs the non-logconcave sampling task associated to the optimization problem in polynomial time, via a novel application of weak Poincaré inequalities.