Abstract:Optimization is at the heart of machine learning, statistics, and many applied scientific disciplines. It also has a long history in physics, ranging from the minimal action principle to finding ground states of disordered systems such as spin glasses. Proximal algorithms form a class of methods that are broadly applicable and are particularly well-suited to nonsmooth, constrained, large-scale, and distributed optimization problems. There are essentially five proximal algorithms currently known, each proposed in seminal work: Forward-backward splitting, Tseng splitting, Douglas-Rachford, alternating direction method of multipliers, and the more recent Davis-Yin. These methods sit on a higher level of abstraction compared to gradient-based ones, with deep roots in nonlinear functional analysis. In this paper we show that all of these methods are actually different discretizations of a single differential equation, namely, the simple gradient flow which dates back to Cauchy (1847). An important aspect behind many of the success stories in machine learning relies on "accelerating" the convergence of first-order methods. However, accelerated methods are notoriously difficult to analyze, counterintuitive, and without an underlying guiding principle. We show that similar discretization schemes applied to Newton's equation with an additional dissipative force, which we refer to as accelerated gradient flow, allow us to obtain accelerated variants of all these proximal algorithms-the majority of which are new although some recover known cases in the literature. Furthermore, we extend these methods to stochastic settings, allowing us to make connections with Langevin and Fokker-Planck equations. Similar ideas apply to gradient descent, heavy ball, and Nesterov's method which are simpler. Our results therefore provide a unified framework from which several important optimization methods are nothing but simulations of classical dissipative systems.

Stochastic Modified Flows, Mean-Field Limits and Dynamics of Stochastic Gradient Descent

Stochastic Modified Flows for Riemannian Stochastic Gradient Descent

Asymptotic Analysis via Stochastic Differential Equations of Gradient Descent Algorithms in Statistical and Computational Paradigms

Stochastic modified equations and dynamics of stochastic gradient algorithms i: Mathematical foundations

Functional Central Limit Theorem and Strong Law of Large Numbers for Stochastic Gradient Langevin Dynamics

Hitting the High-Dimensional Notes: An ODE for SGD learning dynamics on GLMs and multi-index models

A new flow dynamic approach for Wasserstein gradient flows

Rigorous dynamical mean field theory for stochastic gradient descent methods

Analysis of Stochastic Gradient Descent in Continuous Time

Gradient flows on metric graphs with reservoirs: Microscopic derivation and multiscale limits

Dynamical mean-field theory for stochastic gradient descent in Gaussian mixture classification

An SDE Perspective on Stochastic Inertial Gradient Dynamics with Time-Dependent Viscosity and Geometric Damping

Gradient flows and proximal splitting methods: A unified view on accelerated and stochastic optimization

Stochastic Gradient Flow Dynamics of Test Risk and its Exact Solution for Weak Features

The Large Deviation Principle for Stochastic Flow of Stochastic Slow-Fast Motions

The Limiting Dynamics of SGD: Modified Loss, Phase Space Oscillations, and Anomalous Diffusion

Stochastic Differential Equations models for Least-Squares Stochastic Gradient Descent

Convergence of Constant Step Stochastic Gradient Descent for Non-Smooth Non-Convex Functions

Large-Scale Wasserstein Gradient Flows

Large Deviations for Stochastic Flows and Their Applications