Variational Analysis in the Wasserstein Hierarchy

TL;DR

This paper develops a variational framework for higher-order Wasserstein spaces on Riemannian manifolds using category theory, characterizing geodesics and defining gradients for functional analysis.

Contribution

It introduces a novel variational structure on iterated Wasserstein spaces, extending optimal transport theory with categorical tools for geometric and functional analysis.

Findings

Characterization of constant speed geodesics in $P^{(n)}_2(M)$

Introduction of a gradient notion for functionals on $P^{(n)}_2(M)$

Framework for analyzing differentiability and convexity of functionals

Abstract

Let $M$ be a complete connected Riemannian manifold. For $n \geq 0$, we endow the Wasserstein space $P^{(n)}_2(M) = P_2(\ldots P_2(M)\ldots)$, equipped with the Wasserstein distance $W_2$, with a variational structure that generalizes the standard variational structure on $P_2(M)$ provided by optimal transport theory. Our approach makes use of tools from category theory to lift the geometric structure of the manifold $M$ to the spaces $P^{(n)}_2(M)$, in order to establish in a principled way a rigorous theoretical framework for variational analysis on the space $P^{(n)}_2(M)$. In particular, we obtain a precise characterization of the constant speed geodesics of the space $P^{(n)}_2(M)$ in terms of optimal velocity plans. Moreover, we introduce a notion of gradient for functionals defined on $P^{(n)}_2(M)$, which allows us to study the differentiability and the convexity of various types of such functionals.