A short proof of near-linear convergence of adaptive gradient descent under fourth-order growth and convexity
Organizations: Department of Statistics and Data Science, The Wharton School, University of Pennsylvania, Philadelphia, PA 19104
Abstract
Davis, Drusvyatskiy, and Jiang showed that gradient descent with an adaptive stepsize converges locally at a nearly-linear rate for smooth functions that grow at least quartically away from their minimizers. The argument is intricate, relying on monitoring the performance of the algorithm relative to a certain manifold of slow growth -- called the ravine. In this work, we provide a direct Lyapunov-based argument that bypasses these difficulties when the objective is in addition convex and a has a unique minimizer. As a byproduct of the argument, we obtain a more adaptive variant than the original algorithm with encouraging numerical performance.
Figures & tables
Gradient Descent
Gradient descent iteratively updates model parameters to reduce a loss using its slope, with variants for speed, scale, and constraints.
See all 118 papers