Intrinsic Wasserstein Distance
- Intrinsic Wasserstein Distance is a metric concept that generalizes classical optimal transport by integrating mass variation with geometric structure.
- It employs a generalized Benamou–Brenier formulation by adding source terms to handle mass creation and annihilation in transport problems.
- The duality through the flat metric links mass differences with Lipschitz constraints, enabling robust analysis in metric measure spaces and PDEs.
The intrinsic Wasserstein distance is a concept that encompasses a family of constructions and theoretical frameworks wherein the metric structure or geometry induced by Wasserstein (optimal transport) distances is made intrinsic to the space of probability measures or tailored to the underlying data geometry. This notion appears across several research domains, including extensions to measures with varying mass, variants that incorporate underlying geometric or manifold structure, dual formulations, and dynamical/variational representations. Intrinsic Wasserstein distances play a central role in analysis on metric measure spaces, high-dimensional statistics, partial differential equations, generative modeling, and in optimal transport–based comparison of structured objects.
1. Structural Limitations of Classical Wasserstein Distances and the Need for Intrinsic Constructions
Classical Wasserstein distances (for ), rooted in the Monge–Kantorovich optimal transport problem, are defined for probability measures of equal total mass. The primal formulation is
where denotes the set of couplings with and as marginals. This restriction to equal mass is an intrinsic limitation: in applications such as transport equations with sources, mass may not be conserved; hence, is undefined or infinite for measures of unequal mass (Piccoli et al., 2013). This motivates intrinsic generalizations that retain the geometric and variational flavor of optimal transport but allow for broader applicability.
2. Generalized Wasserstein Distances: Intrinsic Extension via Mass Variation
Piccoli and Rossi introduced the generalized Wasserstein distance to circumvent the equal-mass limitation (Piccoli et al., 2013). The construction “mixes” two types of costs:
- Transport cost (as in classical Wasserstein, weighted by )
- Mass variation cost (total variation or distance, weighted by )
Formally, for finite positive measures and ,
where the infimum is taken over all decompositions , , with being the parts to be transported. Then,
This distance interpolates between transport and total variation: for large , the cost of mass variation dominates, recovering an -type metric; for small (relative to ), transport (i.e., geometric movement) dominates. This extension renders the metric “intrinsic” in the sense that all finite measures, regardless of total mass, can be assigned a finite distance.
3. Generalized Dynamical Formulation: The Intrinsic Benamou–Brenier Framework
The classical Benamou–Brenier formula relates the squared distance between measures and to the infimum of an action functional over solutions to the continuity equation: In the generalized setting, to allow for mass sources/sinks, Piccoli and Rossi (Piccoli et al., 2013) extend this formula by introducing a source term: An associated action functional becomes
The generalized Benamou–Brenier formula states
which links the metric directly to a variational problem that simultaneously handles mass transport and net creation/annihilation. This provides an “intrinsic” dynamical metric that naturally extends beyond mass-preserving flows.
4. Duality and the Flat Metric: Intrinsic Kantorovich–Rubinstein Theorem
For and , the generalized Wasserstein distance admits a dual characterization: i.e., the supremum is over all functions that are both Lipschitz with constant at most one and uniformly bounded by one [(Piccoli et al., 2013), Theorem 13]. This is precisely the so-called “flat metric” or bounded Lipschitz distance. In contrast to the classical , which requires equal mass and only the Lipschitz constraint, the flat metric accounts also for pure mass differences, reflecting a fully intrinsic metric between finite signed measures: This dual formulation explicitly ties geometric (transport) and measure-theoretic (mass change) aspects, generalizing the Kantorovich–Rubinstein theorem to the non-conservative setting.
5. Mathematical Formalism and Parameter Effects
The core mathematical constructs for the intrinsic Wasserstein strategy are summarized as follows:
Generalized Cost | Definition |
---|---|
Generalized Distance | |
Benamou–Brenier (gen.) | |
Duality (p=1, a=b=1) |
The parameters and provide a mechanism for controlling the relative penalization. In the extreme , the metric is dominated by pure mass differences and approaches the distance. For small , approaches the classical transport metric. This parametric flexibility supports modeling scenarios with varying dominance of creation/annihilation versus rearrangement of mass.
6. Applications and Theoretical Implications
By extending Wasserstein distances intrinsically in this manner, several new analytical opportunities emerge:
- Partial Differential Equations with Sources: The generalized Benamou–Brenier framework enables the variational paper of PDEs where mass is not conserved, such as transport equations with source or sink terms.
- Measure Comparison: quantifies distance between arbitrary finite measures, not just probability measures of equal mass, which is pivotal for modeling, inference, and statistics in inhomogeneous or dissipative systems.
- Gradient Flow and Numerical Schemes: The inclusion of source terms in the continuity equation and dual formulations facilitates the design and analysis of new numerical schemes for gradient flows and variational approximations in optimal transport problems with mass variation.
- Theoretical Generality: The identification of the dual (flat metric) representation and extension of the Benamou–Brenier formula deepen the structural analogy between the analysis on optimal transport spaces and more classical Riemannian or metric geometry, but in a context that is fully intrinsic to the geometry of arbitrary measures.
7. Connections to Broader Intrinsic Optimal Transport Theory
Intrinsic modifications of the Wasserstein metric reflect a broader trend in adapting optimal transport to underlying geometric or application-specific structure. For example, in the context of measure-valued equations on manifolds, intrinsic constructions appear via the use of geodesic distances in the base space (manifold), replacement of the ground cost with one induced by spectral or localized structure, or adaptation to variable-mass settings (such as unbalanced transport or the inclusion of boundary sources). The generalized Wasserstein distances serve as foundational models for such extensions.
A plausible implication is that, by providing a robust variational framework and dual characterization, intrinsic Wasserstein distances can serve as the basis for future developments in comparative geometry of measure spaces, efficient computation on non-conservative or heterogeneous datasets, and further connections to information geometry and entropic transport formulations.
In summary, intrinsic Wasserstein distances generalize and adapt the classical optimal transport metric structure by incorporating mass variation, duality with bounded Lipschitz functions, and variational dynamical formulations. These developments enable the application of optimal transport methods to a broader array of problems in analysis, partial differential equations, and applied mathematics where the geometry of finite measures—not just probability measures of equal mass—is central (Piccoli et al., 2013).