About Speakers Schedule Contact Us INS
Young Researcher Workshop on Uncertainty Quantification and Machine Learning

Towards the theoretical understanding of large batch training in stochastic gradient descent

Speaker

Yuhua Zhu , Stanford University

Time

05 Jun, 16:00 - 16:30

Abstract

Stochastic gradient descent (SGD) is almost ubiquitously used for training non- convex optimization tasks. Recently, a hypothesis proposed by Keskar et al. that large batch methods tend to converge to sharp minimizers has received increasing attention by researchers. We theoretically justify this hypothesis by providing new properties of SGD in both finite-time and asymptotic regime. In particular, we give an explicit escaping time of SGD from a local minimum in the finite-time regime and prove that SGD tends to converge to flatter minima in the asymptotic regime (although may take exponential time to converge) regardless of the batch size. We also find that SGD with a larger learning rate to batch size ratio tends to converge to a flat minimum faster, however, its generalization performance could be worse than the SGD with a smaller learning rate to batch size ratio. We include experiments to corroborate these theoretical findings.