English
Related papers

Related papers: Primal dual methods for Wasserstein gradient flows

200 papers

We develop a new computational framework to solve the partial differential equations (PDEs) governing the flow of the joint probability density functions (PDFs) in continuous-time stochastic nonlinear systems. The need for computing the…

Optimization and Control · Mathematics 2019-08-08 Kenneth F. Caluya , Abhishek Halder

We modify the JKO scheme, which is a time discretization of Wasserstein gradient flows, by replacing the Wasserstein distance with more general transport costs on manifolds. We show when the cost function has a mixed Hessian which defines a…

Analysis of PDEs · Mathematics 2024-02-28 Cale Rankin , Ting-Kam Leonard Wong

Semi-discrete optimal transport problems, which evaluate the Wasserstein distance between a discrete and a generic (possibly non-discrete) probability measure, are believed to be computationally hard. Even though such problems are…

Machine Learning · Computer Science 2022-05-02 Bahar Taskesen , Soroosh Shafieezadeh-Abadeh , Daniel Kuhn

We propose a Variable-Preconditioned Transformed Primal-Dual (VPTPD) method for solving generalized Wasserstein gradient flows based on the structure-preserving JKO scheme. This is a nontrivial extension of the TPD method [Chen et al.…

Numerical Analysis · Mathematics 2025-10-22 Jin Zeng , Dawei Zhan , Ruchi Guo , Chaozhen Wei

We present a discretization-free scalable framework for solving a large class of mass-conserving partial differential equations (PDEs), including the time-dependent Fokker-Planck equation and the Wasserstein gradient flow. The main…

Machine Learning · Computer Science 2023-11-15 Lingxiao Li , Samuel Hurault , Justin Solomon

Wasserstein gradient flow has emerged as a promising approach to solve optimization problems over the space of probability distributions. A recent trend is to use the well-known JKO scheme in combination with input convex neural networks to…

Machine Learning · Computer Science 2022-07-26 Jiaojiao Fan , Qinsheng Zhang , Amirhossein Taghvaei , Yongxin Chen

We develop structure preserving schemes for a class of nonlinear mobility continuity equation. When the mobility is a concave function, this equation admits a form of gradient flow with respect to a Wasserstein-like transport metric. Our…

Numerical Analysis · Mathematics 2025-10-21 Jose A. Carrillo , Li Wang , Chaozhen Wei

We study the distributionally robust optimization (DRO) in a dynamic context where the model uncertainty is captured by penalizing potential models in function of their adapted Wasserstein distance to a given reference model. We consider…

Probability · Mathematics 2025-09-30 Yifan Jiang

Solving Fredholm equations of the first kind is crucial in many areas of the applied sciences. In this work we adopt a probabilistic and variational point of view by considering a minimization problem in the space of probability measures…

Optimization and Control · Mathematics 2024-05-17 Francesca R. Crucinio , Valentin De Bortoli , Arnaud Doucet , Adam M. Johansen

The dynamical formulation of optimal transport, also known as Benamou-Brenier formulation or Computational Fluid Dynamics formulation, amounts to write the optimal transport problem as the optimization of a convex functional under a PDE…

Numerical Analysis · Mathematics 2020-05-25 Hugo Lavenant

Optimal transport is a foundational problem in optimization, that allows to compare probability distributions while taking into account geometric aspects. Its optimal objective value, the Wasserstein distance, provides an important loss…

Machine Learning · Computer Science 2020-02-21 Marin Ballu , Quentin Berthet , Francis Bach

Wasserstein gradient flows are continuous time dynamics that define curves of steepest descent to minimize an objective function over the space of probability measures (i.e., the Wasserstein space). This objective is typically a divergence…

Optimization and Control · Mathematics 2021-02-23 Adil Salim , Anna Korba , Giulia Luise

In this paper, we propose and analyze a fully discrete finite element projection method for the magnetohydrodynamic (MHD) equations. A modified Crank--Nicolson method and the Galerkin finite element method are used to discretize the model…

Numerical Analysis · Mathematics 2022-04-13 Cheng Wang , Jilu Wang , Zeyu Xia , Liwei Xu

We study the computational complexity of the optimal transport problem that evaluates the Wasserstein distance between the distributions of two K-dimensional discrete random vectors. The best known algorithms for this problem run in…

Optimization and Control · Mathematics 2022-10-17 Bahar Taşkesen , Soroosh Shafieezadeh-Abadeh , Daniel Kuhn , Karthik Natarajan

Gradient flows are a powerful tool for optimizing functionals in general metric spaces, including the space of probabilities endowed with the Wasserstein metric. A typical approach to solving this optimization problem relies on its…

Machine Learning · Statistics 2021-12-02 David Alvarez-Melis , Yair Schiff , Youssef Mroueh

We design and compute first-order implicit-in-time variational schemes with high-order spatial discretization for initial value gradient flows in generalized optimal transport metric spaces. We first review some examples of gradient flows…

Numerical Analysis · Mathematics 2023-08-16 Guosheng Fu , Stanley Osher , Wuchen Li

Over the past fifteen years, the theory of Wasserstein gradient flows of convex (or, more generally, semiconvex) energies has led to advances in several areas of partial differential equations and analysis. In this work, we extend the…

Analysis of PDEs · Mathematics 2017-05-04 Katy Craig

We study the decentralized distributed computation of discrete approximations for the regularized Wasserstein barycenter of a finite set of continuous probability measures distributedly stored over a network. We assume there is a network of…

Optimization and Control · Mathematics 2020-02-21 Pavel Dvurechensky , Darina Dvinskikh , Alexander Gasnikov , César A. Uribe , Angelia Nedić

We study the discretization of generalized Wasserstein distances with nonlinear mobilities on the real line via suitable discrete metrics on the cone of N ordered particles, a setting which naturally appears in the framework of…

Analysis of PDEs · Mathematics 2022-09-01 Simone Di Marino , Lorenzo Portinale , Emanuela Radici

This paper aims to investigate a full numerical approximation of non-autonomous semilnear parabolic partial differential equations (PDEs) with nonsmooth initial data. Our main interest is on such PDEs where the nonlinear part is stronger…

Numerical Analysis · Mathematics 2018-09-11 Antoine Tambue , Jean Daniel Mukam