Wasserstein-over-Wasserstein (WoW) Metric

• The WoW metric is a hierarchical extension of optimal transport that quantifies the minimal quadratic cost between meta-measures.
• It leverages variational formulations and geometric properties to ensure stability, uniqueness, and tractable barycenter computations for meta-level data.
• Algorithmic techniques like the double-sliced Wasserstein approximation make large-scale, high-dimensional optimal transport computations feasible.

The Wasserstein-over-Wasserstein (WoW) metric is a rigorous extension of the optimal transport paradigm to spaces of probability measures over probability measures. Specifically, the WoW metric quantifies the minimal expected Wasserstein-2 (quadratic cost) transport cost between meta-measures—probability distributions whose atoms themselves are probability measures on a base space, commonly a Riemannian manifold or $\mathbb{R}^d$. This two-level metric structure is central to modern applications involving hierarchical, class-conditional, or distributional datasets. The WoW metric has a well-established variational, geometric, and algorithmic theory, including barycenters, convexity, regularity, and tractable approximations, with practical relevance in data science, invariant learning, and synthetic dataset modeling.

1. Mathematical Definition and Geometric Properties

Let $(M,d, \mathrm{vol})$ be a connected, complete (often compact) Riemannian $n$-manifold, and denote by $P(M)$ the space of Borel probability measures on $M$. The quadratic Wasserstein-2 distance on $P(M)$ is defined as

$$W_2^2(\mu,\nu) = \inf_{\gamma \in \Gamma(\mu,\nu)} \int_{M \times M} d^2(x,y) \, d\gamma(x,y)$$

where $\Gamma(\mu,\nu)$ is the set of couplings of $\mu$ and $\nu$.

Given two meta-measures $(M,d, \mathrm{vol})$0, the WoW metric is

$(M,d, \mathrm{vol})$1

where $(M,d, \mathrm{vol})$2 are couplings of the meta-measures. Under suitable conditions on $(M,d, \mathrm{vol})$3, such as compactness or sufficient moment growth control, $(M,d, \mathrm{vol})$4 and $(M,d, \mathrm{vol})$5 are Polish (complete, separable) spaces, and $(M,d, \mathrm{vol})$6 is a true metric, inheriting geodesic and convexity properties from the base Wasserstein geometry (Kim et al., 2014).

2. Barycenters and Variational Structure

The notion of barycenter in Wasserstein space extends naturally to the meta-level. Given $(M,d, \mathrm{vol})$7, a WoW-barycenter is any minimizer $(M,d, \mathrm{vol})$8 of

$(M,d, \mathrm{vol})$9

Existence and uniqueness of the barycenter are guaranteed if the support of $n$0 assigns positive mass to absolutely continuous measures, leveraging strict convexity arising from optimal transport uniqueness in this class.

For finitely supported $n$1 with at least one $n$2 absolutely continuous, the barycenter inherits absolute continuity with explicit $n$3 bounds controlled by the densities of the constituents. For general $n$4, sharp regularity results ensure absolute continuity and stability under weak* convergence of $n$5, with continuity of the barycenter map (Kim et al., 2014).

3. Functional Inequalities and Geodesic Convexity

Key functional inequalities, such as displacement convexity and generalized Jensen’s inequalities, are preserved in the WoW framework. For functionals on $n$6 of entropy, potential energy, or interaction type, $n$7-displacement convexity along Wasserstein geodesics yields for the WoW-barycenter $n$8: $n$9 This structural convexity underpins applications such as generalized Brunn–Minkowski inequalities and regularity theorems, even for infinite supports and in curved geometric settings (Kim et al., 2014).

4. Algorithmic Approaches and Double-Sliced Wasserstein (DSW) Approximation

Direct computation of the WoW metric is prohibitive for large datasets: for meta-measures on $P(M)$0 empirical measures of $P(M)$1 points each, the cost is $P(M)$2 for the pairwise cost matrix and $P(M)$3 for the second-level transport.

To address scalability, the double-sliced Wasserstein (DSW) metric leverages one-dimensional quantile isometries and two layers of slicing:

• Outer slicing via projections $P(M)$4 onto $P(M)$5 (across $P(M)$6),
• Inner slicing via functionals in $P(M)$7 (quantile embeddings and functional projections parametrized by Gaussian processes).

The DSW distance between meta-measures $P(M)$8 in $P(M)$9 is computed by Monte Carlo sampling directions and functional slices, with each sample reducing to 1D Wasserstein calculations among projected atoms. For discretized meta-measures, DSW is theoretically equivalent to the true WoW under suitable projections and quantile grids. Complexity is reduced to $M$0 for $M$1 samples, achieving favorable trade-offs in accuracy and runtime in experiments (Piening et al., 26 Sep 2025).

Metric	Complexity (per batch)	Empirical Fidelity	Suitability
WoW	$M$2	Highest	Small/medium $M$3
DSW	$M$4	Equiv. on discrete	Large-scale, scalable

5. Differential Geometry, Gradient Flows, and Statistical Modeling

WoW endows $M$5 with a Riemannian-type geometry analogous to that of $M$6 under $M$7. The tangent space at $M$8 is the closure of cylinder functions on $M$9, and the metric and exponential map enable the construction of constant-speed geodesics and the formal calculus of variations.

Functionals $P(M)$0 admit WoW (sub/super)-differentials, and corresponding WoW-gradient flows are defined by

$P(M)$1

For Maximum Mean Discrepancy (MMD) and Sliced Wasserstein kernels on $P(M)$2, efficient WoW-gradient-based optimization is feasible. Discrete implementations via particles-of-measures lead to dynamics that update support points through the exponential map in the underlying manifold (Bonet et al., 9 Jun 2025).

6. Applications and Empirical Evaluation

The WoW metric and its variants have been applied to a range of problems involving datasets as structured collections of distributions:

• Shape and point-cloud classification: Shapes or clouds are encoded as meta-measures of local or class-conditioned distributions; DSW-based 3-NN matches or surpasses other metrics in accuracy and efficiency (Piening et al., 26 Sep 2025).
• Domain adaptation and transfer learning: WoW-driven gradient flows enable structuring of source and target class distributions for high-fidelity adaptation (Bonet et al., 9 Jun 2025).
• Dataset distillation and generative modeling: Meta-level optimization enables matching or distilling large datasets into small, structured representatives, with WoW as the optimization backbone (Bonet et al., 9 Jun 2025).
• Image comparison by patch distributions: Batched image datasets, modeled as meta-measures over patches, benefit from DSW-accelerated computations for robust, sample-efficient divergence minimization (Piening et al., 26 Sep 2025).

Empirical results indicate that DSW approximates WoW accurately while achieving substantial computational speed-up, with strong correlations to exact optimal transport dataset distances, and robustness to high-dimensional or ill-conditioned moment structure (Piening et al., 26 Sep 2025).

7. Regularity, Stability, and Theoretical Guarantees

Analytical results establish absolute continuity of WoW barycenters under mild regularity of the underlying measures, with explicit bounds on density in terms of curvature (Ricci lower bounds) and measure concentration. The functional cost map $P(M)$3 is geodesically convex in $P(M)$4 and strictly convex if the support of $P(M)$5 contains absolutely continuous measures, ensuring uniqueness and strong stability of barycenters, to which the barycenter map is continuous in the weak* topology (Kim et al., 2014). For DSW, equivalence to WoW on discretized mixtures is supported by convergence theorems and Carleman/Cramér–Wold arguments (Piening et al., 26 Sep 2025).

---

The WoW metric thus enables a compositional and hierarchical extension of optimal transport, with a robust theory encompassing analysis, optimization, and practical implementations for structured and distributional data. Its interplay with sliced and kernel-based approximations offers scalable routes for large and high-dimensional problems, while ensuring rigorous statistical and geometric fidelity.