Current browse context:
stat.ML
Change to browse by:
References & Citations
Statistics > Machine Learning
Title: Convergence diagnostics for stochastic gradient descent with constant step size
(Submitted on 17 Oct 2017 (v1), last revised 23 Feb 2018 (this version, v2))
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.
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.