Primal dual methods for Wasserstein gradient flows

DOI10.1007/S10208-021-09503-1arXiv1901.08081MaRDI QIDQ6312949FDOQ6312949

Authors: J. A. Carrillo, Katy Craig, Li Wang, Chaozhen Wei 

Publication date: 23 January 2019

Abstract: Combining the classical theory of optimal transport with modern operator splitting techniques, we develop a new numerical method for nonlinear, nonlocal partial differential equations, arising in models of porous media, materials science, and biological swarming. Our method proceeds as follows: First, we discretize in time, either via the classical JKO scheme or via a novel Crank-Nicolson type method we introduce. Next, we use the Benamou-Brenier dynamical characterization of the Wasserstein distance to reduce computing the solution of the discrete time equations to solving fully discrete minimization problems, with strictly convex objective functions and linear constraints. Third, we compute the minimizers by applying a recently introduced, provably convergent primal dual splitting scheme for three operators [Yan 2018]. By leveraging the PDEs' underlying variational structure, our method overcomes stability issues present in previous numerical work built on explicit time discretizations, which suffer due to the equations' strong nonlinearities and degeneracies. Our method is also naturally positivity and mass preserving and, in the case of the JKO scheme, energy decreasing. We prove that minimizers of the fully discrete problem converge to minimizers of the spatially continuous, discrete time problem as the spatial discretization is refined. We conclude with simulations of nonlinear PDEs and Wasserstein geodesics in one and two dimensions that illustrate the key properties of our approach, including higher order convergence our novel Crank-Nicolson type method, when compared to the classical JKO method.

Recommendations

• A primal-dual approach for a total variation Wasserstein flow
• Lagrangian schemes for Wasserstein gradient flows
• A smoothed dual approach for variational Wasserstein problems
• A smoothed dual approach for variational Wasserstein problems
• An augmented Lagrangian approach to Wasserstein gradient flows and applications
• The back-and-forth method for Wasserstein gradient flows
• Primal-dual subgradient methods for convex problems
• Duality on gradient estimates and Wasserstein controls
• Accelerated Bregman Primal-Dual Methods Applied to Optimal Transport and Wasserstein Barycenter Problems
• Primal-Dual Stochastic Gradient Method for Convex Programs with Many Functional Constraints

Mathematics Subject Classification ID

Numerical optimization and variational techniques (65K10) Variational methods applied to PDEs (35A15) Iterative procedures involving nonlinear operators (47J25) Nonlinear evolution equations (47J35) Numerical methods involving duality (49M29) Numerical analysis (65-XX) Continuum models (systems of particles, etc.) arising in equilibrium statistical mechanics (82B21)

This page was built for publication: Primal dual methods for Wasserstein gradient flows

Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q6312949)