Regularity theory and geometry of unbalanced optimal transport

Thomas Gallouët,Roberta Ghezzi,François-Xavier Vialard

2024-07-01

Abstract:Using the dual formulation only, we show that the regularity of unbalanced optimal transport also called entropy-transport inherits from the regularity of standard optimal transport. We provide detailed examples of Riemannian manifolds and costs for which unbalanced optimal transport is regular.Among all entropy-transport formulations, Wasserstein-Fisher-Rao (WFR) metric, also called Hellinger-Kantorovich, stands out since it admits a dynamic formulation, which extends the Benamou-Brenier formulation of optimal transport. After demonstrating the equivalence between dynamic and static formulations on a closed Riemannian manifold, we prove a polar factorization theorem, similar to the one due to Brenier and Mc-Cann. As a byproduct, we formulate the Monge-Amp{è}re equation associated with WFR metric, which also holds for more general costs. Last, we study the link between c-convex functions for the cost induced by the WFR metric and the cost on the cone. The main result is that the weak Ma-Trudinger-Wang condition on the cone implies the same condition on the manifold for the cost induced by WFR.

Optimization and Control

What problem does this paper attempt to address?

The paper primarily explores the regularity theory and geometric properties of Unbalanced Optimal Transport (UOT). Specifically, the authors focus on how to derive the regularity of unbalanced optimal transport from the regularity of standard optimal transport.

### Problems Addressed in the Paper

1. **Regularity of Unbalanced Optimal Transport**:
- The authors demonstrate that the regularity of unbalanced optimal transport inherits from the regularity of standard optimal transport. By studying the dual formulation and its first-order optimality conditions, they prove that under certain conditions, the potentials in unbalanced optimal transport exhibit smooth properties similar to those in standard optimal transport.

2. **Regularity under Specific Costs**:
- For two important cost functions—the Wasserstein-Fisher-Rao distance (also known as the Hellinger-Kantorovich distance) and the Gaussian-Hellinger distance—the authors provide detailed examples to illustrate that unbalanced optimal transport is regular under these scenarios.

3. **Equivalence of Dynamic and Static Forms**:
- For the Wasserstein-Fisher-Rao distance on closed Riemannian manifolds, the authors prove the equivalence between the dynamic and static forms and provide a polar decomposition theorem, similar to the results of Brenier and McCann.

4. **Relation to Ma-Trudinger-Wang Condition**:
- The relationship between c-convex functions on cones and the cost induced by the Wasserstein-Fisher-Rao distance is studied. The main result is that if the weak Ma-Trudinger-Wang condition holds on the cone, then it also holds for the cost induced by the Wasserstein-Fisher-Rao distance.

### Summary of Methods

- **Transfer of Regularity**: The authors study the regularity of unbalanced optimal transport using the dual formulation and prove that when the potentials in unbalanced optimal transport satisfy certain smoothness conditions, their regularity can be inherited from standard optimal transport.

- **Existence of Lipschitz Potentials**: The authors further explore the conditions for the existence of Lipschitz potentials under unbounded cost functions, which is crucial for ensuring the transfer of regularity.

- **Applications under Specific Costs**: For the Wasserstein-Fisher-Rao distance and the Gaussian-Hellinger distance, the authors analyze their regularity performance in different spaces (such as Euclidean spaces and Riemannian manifolds) in detail.

Through this work, the authors provide a solid theoretical foundation for understanding the regularity of unbalanced optimal transport and offer important clues for further research and development.