We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Optimal Rates for Multi-pass Stochastic Gradient Methods

Abstract: We analyze the learning properties of the stochastic gradient method when multiple passes over the data and mini-batches are allowed. We study how regularization properties are controlled by the step-size, the number of passes and the mini-batch size. In particular, we consider the square loss and show that for a universal step-size choice, the number of passes acts as a regularization parameter, and optimal finite sample bounds can be achieved by early-stopping. Moreover, we show that larger step-sizes are allowed when considering mini-batches. Our analysis is based on a unifying approach, encompassing both batch and stochastic gradient methods as special cases. As a byproduct, we derive optimal convergence results for batch gradient methods (even in the non-attainable cases).
Comments: Fixed a typo in Eq (66)
Subjects: Machine Learning (cs.LG); Optimization and Control (math.OC); Machine Learning (stat.ML)
Journal reference: Journal of Machine Learning Research, 18:1-47, 2017
Cite as: arXiv:1605.08882 [cs.LG]
  (or arXiv:1605.08882v3 [cs.LG] for this version)

Submission history

From: Junhong Lin [view email]
[v1] Sat, 28 May 2016 12:11:22 GMT (76kb,D)
[v2] Sat, 21 Oct 2017 22:55:48 GMT (91kb,D)
[v3] Fri, 15 Mar 2019 02:07:20 GMT (91kb,D)

Link back to: arXiv, form interface, contact.