English

Nesterov acceleration despite very noisy gradients

Machine Learning 2024-11-04 v3 Machine Learning Optimization and Control

Abstract

We present a generalization of Nesterov's accelerated gradient descent algorithm. Our algorithm (AGNES) provably achieves acceleration for smooth convex and strongly convex minimization tasks with noisy gradient estimates if the noise intensity is proportional to the magnitude of the gradient at every point. Nesterov's method converges at an accelerated rate if the constant of proportionality is below 1, while AGNES accommodates any signal-to-noise ratio. The noise model is motivated by applications in overparametrized machine learning. AGNES requires only two parameters in convex and three in strongly convex minimization tasks, improving on existing methods. We further provide clear geometric interpretations and heuristics for the choice of parameters.

Keywords

Cite

@article{arxiv.2302.05515,
  title  = {Nesterov acceleration despite very noisy gradients},
  author = {Kanan Gupta and Jonathan W. Siegel and Stephan Wojtowytsch},
  journal= {arXiv preprint arXiv:2302.05515},
  year   = {2024}
}

Comments

Accepted to NeurIPS 2024

R2 v1 2026-06-28T08:37:27.249Z