English

Convex SGD: Generalization Without Early Stopping

Machine Learning 2024-04-16 v2 Statistics Theory Statistics Theory

Abstract

We consider the generalization error associated with stochastic gradient descent on a smooth convex function over a compact set. We show the first bound on the generalization error that vanishes when the number of iterations TT and the dataset size nn go to zero at arbitrary rates; our bound scales as O~(1/T+1/n)\tilde{O}(1/\sqrt{T} + 1/\sqrt{n}) with step-size αt=1/t\alpha_t = 1/\sqrt{t}. In particular, strong convexity is not needed for stochastic gradient descent to generalize well.

Keywords

Cite

@article{arxiv.2401.04067,
  title  = {Convex SGD: Generalization Without Early Stopping},
  author = {Julien Hendrickx and Alex Olshevsky},
  journal= {arXiv preprint arXiv:2401.04067},
  year   = {2024}
}
R2 v1 2026-06-28T14:11:30.384Z