Abstract:A vast majority of machine learning algorithms train their models and perform inference by solving optimization problems. In order to capture the learning and prediction problems accurately, structural constraints such as sparsity or low rank are frequently imposed or else the objective itself is designed to be a non-convex function. This is especially true of algorithms that operate in high-dimensional spaces or that train non-linear models such as tensor models and deep networks. The freedom to express the learning problem as a non-convex optimization problem gives immense modeling power to the algorithm designer, but often such problems are NP-hard to solve. A popular workaround to this has been to relax non-convex problems to convex ones and use traditional methods to solve the (convex) relaxed optimization problems. However this approach may be lossy and nevertheless presents significant challenges for large scale optimization. On the other hand, direct approaches to non-convex optimization have met with resounding success in several domains and remain the methods of choice for the practitioner, as they frequently outperform relaxation-based techniques – popular heuristics include projected gradient descent and alternating minimization. However, these are often poorly understood in terms of their convergence and other properties. This monograph presents a selection of recent advances that bridge a long-standing gap in our understanding of these heuristics. We hope that an insight into the inner workings of these methods will allow the reader to appreciate the unique marriage of task structure and generative models that allow these heuristic techniques to (provably) succeed. The monograph will lead the reader through several widely used nonconvex optimization techniques, as well as applications thereof. The goal of this monograph is to both, introduce the rich literature in this area, as well as equip the reader with the tools and techniques needed to analyze these simple procedures for non-convex problems.

On nonconvex optimization for machine learning: Gradients, stochasticity, and saddle points

Beyond Convexity: Stochastic Quasi-Convex Optimization

Demystifying the Myths and Legends of Nonconvex Convergence of SGD

Stochastic Gradient Descent in the Viewpoint of Graduated Optimization

A gentle introduction to gradient-based optimization and variational inequalities for machine learning

On the saddle point problem for non-convex optimization

The Frontier of SGD and Its Variants in Machine Learning

A Deterministic Gradient-Based Approach to Avoid Saddle Points

An Alternative View: When Does SGD Escape Local Minima?

Bound Analysis of Natural Gradient Descent in Stochastic Optimization Setting

Understanding the unstable convergence of gradient descent.

Review Non-convex Optimization Method for Machine Learning

Non-convex Optimization for Machine Learning

Identifying and attacking the saddle point problem in high-dimensional non-convex optimization

Beyond the Edge of Stability via Two-step Gradient Updates

The Optimality of (Accelerated) SGD for High-Dimensional Quadratic Optimization

On Convergence of Incremental Gradient for Non-Convex Smooth Functions

Stochastic Gradient Descent with Biased but Consistent Gradient Estimators

Simple and Optimal Stochastic Gradient Methods for Nonsmooth Nonconvex Optimization

Riemannian stochastic optimization methods avoid strict saddle points

Gradient Descent, Stochastic Optimization, and Other Tales