English
Related papers

Related papers: JKO schemes with general transport costs

200 papers

This paper considers the problem of regression over distributions, which is becoming increasingly important in machine learning. Existing approaches often ignore the geometry of the probability space or are computationally expensive. To…

Machine Learning · Computer Science 2025-10-31 Maksim Maslov , Alexander Kugaevskikh , Matthew Ivanov

Sampling from nonsmooth target probability distributions is essential in various applications, including the Bayesian Lasso. We propose a splitting-based sampling algorithm for the time-implicit discretization of the probability flow for…

Computation · Statistics 2025-07-14 Fuqun Han , Stanley Osher , Wuchen Li

In this work, we study the Wasserstein gradient flow of the Riesz energy defined on the space of probability measures. The Riesz kernels define a quadratic functional on the space of measure which is not in general geodesically convex in…

Analysis of PDEs · Mathematics 2024-01-30 Siwan Boufadène , François-Xavier Vialard

We will propose a new algorithm for finding critical points of cost functions defined on a differential manifold. We will lift the initial cost function to a manifold that can be embedded in a Riemannian manifold (Euclidean space) and will…

Optimization and Control · Mathematics 2018-09-05 Petre Birtea , Dan Comănescu , Călin-Adrian Popa

We propose a general method to identify nonlinear Fokker--Planck--Kolmogorov equations (FPK equations) as gradient flows on the space of probability measures on $\mathbb{R}^d$ with a natural differential geometry. Our notion of gradient…

Analysis of PDEs · Mathematics 2024-11-11 Marco Rehmeier , Michael Röckner

We study the probability measures $\rho\in \mathcal M(\mathbb R^2)$ minimizing the functional \[ J[\rho]=\iint \log\frac1{|x-y|}d\rho(x)d\rho(y)+d^2(\rho, \rho_0), \] where $\rho_0$ is a given probability measure and $d(\rho, \rho_0)$ is…

Analysis of PDEs · Mathematics 2019-05-22 Aram Karakhanyan

In his classical argument, Rao derives the Riemannian distance corresponding to the Fisher metric using a mapping between the space of positive measures and Euclidean space. He obtains the Hellinger distance on the full space of measures…

Statistics Theory · Mathematics 2022-09-27 Jesse van Oostrum

The geometry of dynamical systems estimated from trajectory data is a major challenge for machine learning applications. Koopman and transfer operators provide a linear representation of nonlinear dynamics through their spectral…

Machine Learning · Statistics 2025-09-30 Thibaut Germain , Rémi Flamary , Vladimir R. Kostic , Karim Lounici

We propose a fundamental metric for measuring the distance between two distributions. This metric, referred to as the decision-focused (DF) divergence, is tailored to stochastic linear optimization problems in which the objective…

Statistics Theory · Mathematics 2026-02-04 Suhan Liu , Mo Liu

In this paper we investigate the numerical approximation of an analogue of the Wasserstein distance for optimal transport on graphs that is defined via a discrete modification of the Benamou--Brenier formula. This approach involves the…

Numerical Analysis · Mathematics 2017-07-24 Matthias Erbar , Martin Rumpf , Bernhard Schmitzer , Stefan Simon

We propose a spatial discretization of the fourth-order nonlinear DLSS equation on the circle. Our choice of discretization is motivated by a novel gradient flow formulation with respect to a metric that generalizes martingale transport.…

Analysis of PDEs · Mathematics 2025-02-14 Daniel Matthes , Eva-Maria Rott , Giuseppe Savaré , André Schlichting

We formulate and solve a regression problem with time-stamped distributional data. Distributions are considered as points in the Wasserstein space of probability measures, metrized by the 2-Wasserstein metric, and may represent images,…

Systems and Control · Electrical Eng. & Systems 2021-06-30 Amirhossein Karimi , Tryphon T. Georgiou

Wasserstein gradient and Hamiltonian flows have emerged as essential tools for modeling complex dynamics in the natural sciences, with applications ranging from partial differential equations (PDEs) and optimal transport to quantum…

Numerical Analysis · Mathematics 2025-11-11 Jianyu Hu , Juan-Pablo Ortega , Daiying Yin

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

We consider the problem of recovering the Riemannian metric on a compact closed manifold from the optimal transport maps when the underlying cost function is the squared Riemann distance. We show that the metric can be uniquely determined…

Analysis of PDEs · Mathematics 2025-11-20 Jian Zhai , Kelvin Shuangjian Zhang

We define new differential structures on the Wasserstein spaces $\mathcal{W}_p(M)$ for $p > 2$ and a general Riemannian manifold $(M,g)$. We consider a very general and possibly degenerate second order partial differential flow equation…

Analysis of PDEs · Mathematics 2026-03-17 Arthur Schichl

In recent work arXiv:2109.07820 we have shown the equivalence of the widely used nonconvex (generalized) branched transport problem with a shape optimization problem of a street or railroad network, known as (generalized) urban planning…

Optimization and Control · Mathematics 2022-10-19 Julius Lohmann , Bernhard Schmitzer , Benedikt Wirth

Motivated by a constrained minimization problem, it is studied the gradient flows with respect to Hessian Riemannian metrics induced by convex functions of Legendre type. The first result characterizes Hessian Riemannian structures on…

Optimization and Control · Mathematics 2018-11-27 Felipe Alvarez , Jérôme Bolte , Olivier Brahic

We establish a version of the Li--Yau--Hamilton inequality for the Granular-Medium equation on the torus, both at the PDE level and for its time-discrete approximation given by the JKO scheme. We then apply this estimate to derive further…

Analysis of PDEs · Mathematics 2025-10-13 Fanch Coudreuse

We investigate here the optimal transportation problem on configuration space for the quadratic cost. It is shown that, as usual, provided that the corresponding Wasserstein is finite, there exists one unique optimal measure and that this…

Probability · Mathematics 2007-05-23 L. Decreusefond