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

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

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

Statistics > Machine Learning

Title: Convergence diagnostics for stochastic gradient descent with constant step size

Abstract: Many iterative procedures in stochastic optimization exhibit a transient phase followed by a stationary phase. During the transient phase the procedure converges towards a region of interest, and during the stationary phase the procedure oscillates in that region, commonly around a single point. In this paper, we develop a statistical diagnostic test to detect such phase transition in the context of stochastic gradient descent with constant learning rate. We present theory and experiments suggesting that the region where the proposed diagnostic is activated coincides with the convergence region. For a class of loss functions, we derive a closed-form solution describing such region. Finally, we suggest an application to speed up convergence of stochastic gradient descent by halving the learning rate each time stationarity is detected. This leads to a new variant of stochastic gradient descent, which in many settings is comparable to state-of-art.
Comments: Accepted to Artificial Intelligence and Statistics, 2018
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG); Statistics Theory (math.ST); Computation (stat.CO)
Cite as: arXiv:1710.06382 [stat.ML]
  (or arXiv:1710.06382v2 [stat.ML] for this version)

Submission history

From: Jerry Chee [view email]
[v1] Tue, 17 Oct 2017 16:51:16 GMT (2087kb,D)
[v2] Fri, 23 Feb 2018 04:31:07 GMT (2087kb,D)

Link back to: arXiv, form interface, contact.