arXiv · 1907.04232
Unified Optimal Analysis of the (Stochastic) Gradient Method
Abstract
In this note we give a simple proof for the convergence of stochastic gradient (SGD) methods on $\mu$-convex functions under a (milder than standard) $L$-smoothness assumption. We show that for carefully chosen stepsizes SGD converges after $T$ iterations as $O\left( LR^2 \exp \bigl[-\frac{\mu}{4L}T\bigr] + \frac{\sigma^2}{\mu T} \right)$ where $\sigma^2$ measures the variance in the stochastic noise. For deterministic gradient descent (GD) and SGD in the interpolation setting we have $\sigma^2 =0$ and we recover the exponential convergence rate. The bound matches with the best known iteration complexity of GD and SGD, up to constants.
Explore related subjects
Keep this discovery
Sebastian U. Stich. 2019-07-09. Unified Optimal Analysis of the (Stochastic) Gradient Method. https://arxiv.org/abs/1907.04232
Cite the original work for its findings. Save a collection to share your selection of sources.