English

Simple and optimal high-probability bounds for strongly-convex stochastic gradient descent

Machine Learning 2019-09-04 v1 Optimization and Control Machine Learning

Abstract

We consider stochastic gradient descent algorithms for minimizing a non-smooth, strongly-convex function. Several forms of this algorithm, including suffix averaging, are known to achieve the optimal O(1/T)O(1/T) convergence rate in expectation. We consider a simple, non-uniform averaging strategy of Lacoste-Julien et al. (2011) and prove that it achieves the optimal O(1/T)O(1/T) convergence rate with high probability. Our proof uses a recently developed generalization of Freedman's inequality. Finally, we compare several of these algorithms experimentally and show that this non-uniform averaging strategy outperforms many standard techniques, and with smaller variance.

Keywords

Cite

@article{arxiv.1909.00843,
  title  = {Simple and optimal high-probability bounds for strongly-convex stochastic gradient descent},
  author = {Nicholas J. A. Harvey and Christopher Liaw and Sikander Randhawa},
  journal= {arXiv preprint arXiv:1909.00843},
  year   = {2019}
}
R2 v1 2026-06-23T11:03:25.850Z