arXiv · 2203.01744
Accelerated SGD for Non-Strongly-Convex Least Squares
Abstract
We consider stochastic approximation for the least squares regression problem in the non-strongly convex setting. We present the first practical algorithm that achieves the optimal prediction error rates in terms of dependence on the noise of the problem, as $O(d/t)$ while accelerating the forgetting of the initial conditions to $O(d/t^2)$. Our new algorithm is based on a simple modification of the accelerated gradient descent. We provide convergence results for both the averaged and the last iterate of the algorithm. In order to describe the tightness of these new bounds, we present a matching lower bound in the noiseless setting and thus show the optimality of our algorithm.
Explore related subjects
Keep this discovery
Aditya Varre, Nicolas Flammarion. 2022-03-03. Accelerated SGD for Non-Strongly-Convex Least Squares. https://arxiv.org/abs/2203.01744
Cite the original work for its findings. Save a collection to share your selection of sources.