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-time gradient flows in Gromov hyperbolic spaces

On a Complex Hessian Flow

Flow by Gauss Curvature to the Aleksandrov and Dual Minkowski Problems

Flow by Gauss curvature to the $L_p$-Gaussian Minkowski problem

Gradient flows in asymmetric metric spaces and applications

Local Conditions for Global Convergence of Gradient Flows and Proximal Point Sequences in Metric Spaces

Hyperbolic Gradient Flow: Evolution of Graphs in R^n+1

Proximal Point Algorithms on Hadamard Manifolds: Linear Convergence and Finite Termination

Pointwise Lipschitz Continuous Graph Algorithms via Proximal Gradient Analysis

Convergence analysis of inexact proximal point algorithms on Hadamard manifolds

Gradient Flows and Riemannian Structure in the Gromov-Wasserstein Geometry

Otal-Peigné's Theorem for Gromov-hyperbolic spaces

Computing the Gromov--Hausdorff distance using gradient methods

Gradient descent for unbounded convex functions on Hadamard manifolds and its applications to scaling problems

A gradient flow on control space with rough initial condition

Transformation of $p$-gradient flows to $p'$-gradient flows in metric spaces

On Distal Flows and Common Fixed Point Theorems in Banach Spaces

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

Dual flows in hyperbolic space and de Sitter space

Convergence and Trade-Offs in Riemannian Gradient Descent and Riemannian Proximal Point