Title
On the Insufficiency of Existing Momentum Schemes for Stochastic Optimization
Abstract
Momentum based stochastic gradient methods such as heavy ball (HB) and Nesterov's accelerated gradient descent (NAG) method are widely used in practice for training deep networks and other supervised learning models, as they often provide significant improvements over stochastic gradient descent (SGD). In general, “fast gradient” methods have provable improvements over gradient descent only for the deterministic case, where the gradients are exact. In the stochastic case, the popular explanations for their wide applicability is that when these fast gradient methods are applied in the stochastic case, they partially mimic their exact gradient counterparts, resulting in some practical gain. This work provides a counterpoint to this belief by proving that there are simple problem instances where these methods cannot outperform SGD despite the best setting of its parameters. These negative problem instances are, in an informal sense, generic; they do not look like carefully constructed pathological instances. These results suggest (along with empirical evidence) that HB or NAG's practical performance gains are a by-product of minibatching. Furthermore, this work provides a viable (and provable) alternative, which, on the same set of problem instances, significantly improves over HB, NAG, and SGD's performance. This algorithm, denoted as ASGD, is a simple to implement stochastic algorithm, based on a relatively less popular version of Nesterov's AGD. Extensive empirical results in this paper show that ASGD has performance gains over HB, NAG, and SGD.
Year
DOI
Venue
2018
10.1109/ITA.2018.8503173
2018 Information Theory and Applications Workshop (ITA)
Keywords
DocType
ISBN
momentum schemes,stochastic optimization,stochastic gradient methods,heavy ball,deep networks,supervised learning models,stochastic gradient descent,provable improvements,deterministic case,stochastic case,fast gradient methods,exact gradient counterparts,practical gain,simple problem instances,negative problem instances,carefully constructed pathological instances,SGD's performance,stochastic algorithm,NAG practical performance gains,HB,Nesterov accelerated gradient descent method,ASGD
Conference
978-1-7281-1995-3
Citations 
PageRank 
References 
0
0.34
0
Authors
4
Name
Order
Citations
PageRank
Rahul Kidambi1326.75
Praneeth Netrapalli267434.41
Prateek Jain300.34
Sham Kakade44365282.77