Timezone: »

Katyusha X: Simple Momentum Method for Stochastic Sum-of-Nonconvex Optimization
Zeyuan Allen-Zhu

Wed Jul 11 09:15 AM -- 12:00 PM (PDT) @ Hall B #71

The problem of minimizing sum-of-nonconvex functions (i.e., convex functions that are average of non-convex ones) is becoming increasing important in machine learning, and is the core machinery for PCA, SVD, regularized Newton's method, accelerated non-convex optimization, and more. We show how to provably obtain an accelerated stochastic algorithm for minimizing sum-of-nonconvex functions, by adding one additional line to the well-known SVRG method. This line corresponds to momentum, and shows how to directly apply momentum to the finite-sum stochastic minimization of sum-of-nonconvex functions. As a side result, our method enjoys linear parallel speed-up using mini-batch.

Author Information

Zeyuan Allen-Zhu (Microsoft Research AI)

Related Events (a corresponding poster, oral, or spotlight)

More from the Same Authors