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.

Discrete Gradient Methods and Linear Projection Methods for Preserving a Conserved Quantity of Stochastic Differential Equations

Convergence of Projected Subgradient Method with Sparse or Low-Rank Constraints

A Discontinuous Galerkin Method for Stochastic Conservation Laws.

Continuous-time stochastic gradient descent for optimizing over the stationary distribution of stochastic differential equations

Fully discrete energy-dissipative and conservative discrete gradient particle methods for a class of continuity equations

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

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

Accelerated Gradient Methods with Gradient Restart: Global Linear Convergence

Constrained Optimization with Compressed Gradients: A Dynamical Systems Perspective

Delayed Projection Techniques for Linearly Constrained Problems: Convergence Rates, Acceleration, and Applications

Gradient Convergence of Deep Learning-Based Numerical Methods for BSDEs

Numerical analysis of the stochastic Stefan problem

On Asymptotic Preserving schemes for a class of Stochastic Differential Equations in averaging and diffusion approximation regimes

Scalable Gradients for Stochastic Differential Equations

Linear Convergence of Stochastic Iterative Greedy Algorithms With Sparse Constraints

On structure preservation for fully discrete finite difference schemes of stochastic heat equation with Lévy space-time white noise

Almost sure convergence of stochastic Hamiltonian descent methods

A convergence analysis of the perturbed compositional gradient flow: averaging principle and normal deviations

Projection Method III: Spatial Discretization on the Staggered Grid

Higher order time discretization method for a class of semilinear stochastic partial differential equations with multiplicative noise

On the equivalence between SOR-type methods for linear systems and discrete gradient methods for gradient systems