View PDF HTML (experimental)

Abstract:Suppose we are given two metric spaces and a family of continuous transformations from one to the other. Given a probability distribution on each of these two spaces -- namely the source and the target measures -- the Wasserstein alignment problem seeks the transformation that minimizes the optimal transport cost between the pushforward of the source distribution and the target distribution, ensuring the closest possible alignment in a probabilistic sense. Examples of interest include two distributions on two Euclidean spaces $\mathbb{R}^n$ and $\mathbb{R}^d$, and we want a spatial embedding of the $n$-dimensional source measure in $\mathbb{R}^d$ that is closest in some Wasserstein metric to the target distribution on $\mathbb{R}^d$. Similar data alignment problems also commonly arise in shape analysis and computer vision. In this paper, we show that this nonconvex optimal transport projection problem admits a convex Kantorovich-type dual that exploits statistical independence. This allows us to characterize the set of projections and devise a linear programming algorithm. For certain special examples, such as orthogonal transformations on Euclidean spaces of unequal dimensions and the $2$-Wasserstein cost, we characterize the covariance of the optimal projections. Our results also cover the generalization when we penalize each transformation by a function. An example is the inner product Gromov--Wasserstein distance minimization problem which has recently gained popularity.

Submission history

From: Ting-Kam Leonard Wong [view email]
[v1] Mon, 10 Mar 2025 01:56:45 UTC (613 KB)
[v2] Mon, 6 Jul 2026 02:29:52 UTC (646 KB)