arXiv · 2107.08686
Improved Learning Rates for Stochastic Optimization
Abstract
Stochastic optimization is a cornerstone of modern machine learning. This paper studies the generalization performance of two classical stochastic optimization algorithms: stochastic gradient descent (SGD) and Nesterov's accelerated gradient (NAG). We establish new learning rates for both algorithms, with improved guarantees in some settings or comparable rates under weaker assumptions in others. We also provide numerical experiments to support the theory.
Explore related subjects
Keep this discovery
Shaojie Li, Pengwei Tang, Yong Liu. 2021-07-19. Improved Learning Rates for Stochastic Optimization. https://arxiv.org/abs/2107.08686
Cite the original work for its findings. Save a collection to share your selection of sources.