Primal dual methods for Wasserstein gradient flows

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.

References in corpus (14)

• A Finite-Volume Method for Nonlinear Nonlocal Equations with a Gradient Flow Structure
• Optimal Transport with Proximal Splitting
• The Hele-Shaw asymptotics for mechanical models of tumor growth
• Proximal methods for stationary Mean Field Games with local couplings
• A discontinuous Galerkin method for nonlinear parabolic equations and gradient flow problems with interaction potentials
• BV Estimates in Optimal Transportation and Applications
• Fisher information regularization schemes for Wasserstein gradient flows
• Numerical Study of a Particle Method for Gradient Flows
• Numerical simulation of nonlinear continuity equations by evolving diffeomorphisms
• A Lagrangian scheme for the solution of nonlinear diffusion equations using moving simplex meshes
• An accurate front capturing scheme for tumor growth models with a free boundary limit
• Nonconvex gradient flow in the Wasserstein metric and applications to constrained nonlocal interactions
• The Landau equation as a Gradient Flow
• Convergence framework for the second boundary value problem for the Monge-Ampère equation

Cited by in corpus (7)

• Fisher information regularization schemes for Wasserstein gradient flows
• Neural Parametric Fokker-Planck Equations
• Positive and free energy satisfying schemes for diffusion with interaction potentials
• Frank-Wolfe Methods in Probability Space
• A Proximal-Gradient Algorithm for Crystal Surface Evolution
• Variational Asymptotic Preserving Scheme for the Vlasov-Poisson-Fokker-Planck System
• Tightness of radially-symmetric solutions to 2D aggregation-diffusion equations with weak interaction forces