The Monge-Kantorovich problem on Wasserstein space

Pedram Emami,Brendan Pass
2024-10-09
Abstract:We consider the Monge-Kantorovich problem between two random measuress. More precisely, given probability measures $\mathbb{P}_1,\mathbb{P}_2\in\mathcal{P}(\mathcal{P}(M))$ on the space $\mathcal{P}(M)$ of probability measures on a smooth compact manifold, we study the optimal transport problem between $\mathbb{P}_1$ and $\mathbb{P}_2 $ where the cost function is given by the squared Wasserstein distance $W_2^2(\mu,\nu)$ between $\mu,\nu \in \mathcal{P}(M)$. Under appropriate assumptions on $\mathbb{P}_1$, we prove that there exists a unique optimal plan and that it takes the form of an optimal map. An extension of this result to cost functions of the form $h(W_2(\mu,\nu))$, for strictly convex and strictly increasing functions $h$, is also established. The proofs rely heavily on a recent result of Schiavo \cite{schiavo2020rademacher}, which establishes a version of Rademacher's theorem on Wasserstein space.
Probability,Optimization and Control
What problem does this paper attempt to address?
This paper aims to solve the optimal transport problem in Wasserstein space. Specifically, given two probability measures \(P_1, P_2\in\mathcal{P}(\mathcal{P}(M))\), where \(M\) is a smooth compact manifold, the paper studies the optimal transport problem from \(P_1\) to \(P_2\), where the cost function is the squared Wasserstein distance \(W_2^2(\mu,\nu)\) between the two measures. ### Main problems 1. **Existence and uniqueness of the optimal transport plan**: - The paper proves that under appropriate assumptions, there exists a unique optimal transport plan, and this plan can be represented in the form of an optimal mapping. - Specifically, if \(P_1\) is absolutely continuous with respect to some reference measure \(P_0\), and \(P_1\) is concentrated almost everywhere on absolutely continuous probability measures, then the optimal transport problem has a unique solution, and this solution is concentrated on the graph of a mapping \(T:\mathcal{P}(M)\to\mathcal{P}(M)\). 2. **Extension to more general cost functions**: - The paper further extends the results to cost functions of the form \(h(W_2(\mu,\nu))\), where \(h\) is a strictly convex and strictly increasing \(C^2\) function. - For this type of cost functions, the paper proves that the optimal transport plan is also unique and is concentrated on the graph of a mapping \(T\). ### Key techniques and methods - **Generalization of Rademacher's theorem**: The paper makes use of Dello Schiavo's recent work, which established a version of Rademacher's theorem in Wasserstein space, which plays a crucial role in proving the existence and uniqueness of the optimal mapping. - **Kantorovich dual problem**: Through the Kantorovich dual problem, the paper constructs Kantorovich potentials and uses these potentials to analyze the properties of the optimal transport plan. - **Differential geometric tools**: The paper uses differential geometric tools on Riemannian manifolds, especially when dealing with the geometric structure of Wasserstein space. ### Application background - **Optimal transport under uncertainty**: In many practical applications, the marginal measures \(\mu\) and \(\nu\) may not be known exactly, but have a certain degree of uncertainty. This uncertainty can be expressed by the measure distribution in Wasserstein space, so the research results of this paper are of great significance in dealing with the optimal transport problem under uncertainty. ### Summary This paper solves the optimal transport problem in Wasserstein space through strict mathematical analysis, especially when the cost function is the squared Wasserstein distance or its more general form, and proves the existence and uniqueness of the optimal transport plan. These results are not only of great theoretical value, but also have wide application prospects in dealing with practical problems with uncertainty.