2017

Accelerated Gradient Descent Escapes Saddle Points Faster than Gradient Descent

Jin, Chi, Netrapalli, Praneeth, Jordan, Michael I.

Understand

Nesterov's accelerated gradient descent (AGD), an instance of the general family of "momentum methods", provably achieves faster convergence rate than gradient descent (GD) in the convex setting.

  • However, whether these methods are superior to GD in the nonconvex setting remains open.
  • This paper studies a simple variant of AGD, and shows that it escapes saddle points and finds a second-order stationary point in $\tilde{O}(1/\epsilon^{7/4})$ iterations, faster than the $\tilde{O}(1/\epsilon^{2})$ iterations required by GD.
  • To the best of our knowledge, this is the first Hessian-free algorithm to find a second-order stationary point faster than GD, and also the first single-loop algorithm with a faster rate than GD even in the setting of finding a first-order stationary point.

Reading the bibliography…