Spherical Hellinger–Kantorovich Distance
- Spherical Hellinger–Kantorovich distance is a geodesic metric on probability measures integrating optimal transport with creation and depletion of mass while conserving total mass.
- It unifies 2-Wasserstein and Hellinger/Fisher–Rao geometries through equivalent static (cone-lift, entropy–transport) and dynamical (Benamou–Brenier) formulations.
- Its robust geometric structure supports gradient flows and computational implementations, enabling scalable analysis in data science and nonlinear PDE modeling.
The spherical Hellinger–Kantorovich distance is a geodesic metric on the space of probability measures that integrates optimal transport with creation and depletion of mass, subject to conservation of total mass. It arises as the restriction of the unbalanced Hellinger–Kantorovich metric to the probability simplex, equipping the latter with a natural “spherical” geometry that captures both transport and reaction mechanisms. This construction unifies and interpolates between 2-Wasserstein (optimal transport) and Hellinger/Fisher–Rao geometries, yielding a distance that is especially well-behaved for probability measures on compact metric spaces, such as spheres, convex domains, or manifolds.
1. Conceptual Foundations and Definition
The spherical Hellinger–Kantorovich (SHK) distance, denoted , is defined on the set of Borel probability measures over a compact metric space . It is canonically derived from the Hellinger–Kantorovich (HK) distance on finite nonnegative measures by restricting consideration to probability measures and enforcing a total-mass constraint along geodesics.
Given , the static “soft-marginal” primal form of the HK distance is
with for , and otherwise. On probability measures, the spherical Hellinger–Kantorovich distance is then defined as
which measures the “angle” between elements in the positive-mass cone over 0, restricted to the unit sphere 1 (Sarrazin et al., 2023, Liero et al., 2015, Laschos et al., 2017).
The restriction to 2 imposes conservation of total mass, in contrast to the unbalanced HK metric, under which geodesics can leave the simplex and return as mass is temporarily “created” or “annihilated” (Sarrazin et al., 2023, Laschos et al., 2022).
2. Equivalent Characterizations
The SHK distance admits several equivalent static and dynamic formulations:
- Cone–lift (static) formulation: Probabilities are lifted to measures on the unit sphere of the metric cone
3
with cone distance
4
The SHK squared distance between 5 is
6
where 7 is the usual 2-Wasserstein distance on the cone (Liero et al., 2015, Liero et al., 2015, Laschos et al., 2017).
- Entropy–transport formulation: On 8,
9
with 0, and cost 1 for 2 (3 otherwise) (Liero et al., 2015).
- Benamou–Brenier (dynamical) formulation: For curves 4 in 5,
6
under the continuity equation
7
ensuring local mass change with global conservation (Sarrazin et al., 2023, Mustafi et al., 21 Mar 2026, Laschos et al., 2022, Liero et al., 2015).
The table below summarizes these main equivalences:
| Formulation | Domain | Mass constraint |
|---|---|---|
| Cone–lift | 8 | 9 |
| Entropy–transport | 0 | 1 |
| Benamou–Brenier | 2 | 3 |
3. Geodesic Structure and Riemannian Geometry
SHK endows 4 with a natural geodesic space structure. Geodesics are obtained as projections of constant-speed geodesics in the cone 5, reparametrized and renormalized to fix total mass at all times. Explicitly, if 6 solves the HK dynamical problem, then the corresponding SHK geodesic 7 is constructed via time-rescaling and normalization:
8
with 9 a sinusoidal reparametrization and corresponding transformations for 0 and 1 (Sarrazin et al., 2023).
The tangent space at 2 is
3
endowed with the inner product
4
This structure admits well-defined exponential and logarithmic maps, consistent discretizations, and a left-inverse identity (Sarrazin et al., 2023).
4. Special Cases, Limits, and Geometric Properties
- Dirac masses: For Dirac measures 5, 6, recovering the 2-Wasserstein geometry for atomic probabilities (Sarrazin et al., 2023, Laschos et al., 2022, Liero et al., 2015).
- Convexity and curvature: 7 is a positively curved metric space in the sense of Alexandrov if 8 has curvature 9 (Liero et al., 2015, Laschos et al., 2017). The squared distance is jointly convex and semiconcave on sets of measures with densities uniformly bounded above and below.
- Comparison with Wasserstein and Hellinger: 0 lies between Hellinger–Kakutani and 1 distances. As the geometry is “blown up” (2), SHK converges to the Hellinger–Kakutani; “shrinking” distances (3) yields 4 in the limit (Liero et al., 2015).
- Forbidden regions: For 5, pure transportation is penalized with infinite cost; mass must be split, created, or annihilated (Liero et al., 2015, Laschos et al., 2022).
5. Gradient Flows, Functional Inequalities, and Minimizing Movements
Gradient flows in SHK geometry correspond to solutions of nonlinear PDEs, often interpreted as degenerate Fokker–Planck equations. The SHK gradient of an entropy 6 (with first variation 7) is
8
with 9. This leads to evolution equations of the form
0
(Kondratyev et al., 2018, Mustafi et al., 21 Mar 2026).
Key functional inequalities include a generalized log-Sobolev (entropy–entropy-production) inequality
1
and Talagrand-type transportation inequalities:
2
These are crucial for establishing convergence rates and well-posedness of gradient flows in SHK geometry (Kondratyev et al., 2018, Laschos et al., 2022).
The “minimizing movement” (JKO) scheme applied to 3 produces discrete-time approximations to gradient flows, with each time step defined as a proximal minimization
4
and uniform density bounds and geometric convexity conditions ensuring convergence to Evolutionary Variational Inequality (EVI) solutions (Laschos et al., 2022).
6. Applications and Computational Aspects
Recent work exploits the SHK metric structure for associative memory models based on Sinkhorn divergences, with dynamics realized as gradient flows in SHK geometry. These algorithms update both support locations and weights of point clouds and show robust geometric convergence under suitable conditions (Mustafi et al., 21 Mar 2026). In data analysis contexts, the SHK distance, by linearizing the tangent space structure, allows for scalable implementations and for integration into methods requiring a linear metric geometry (Sarrazin et al., 2023).
On the technical side, discrete barycentric projections and log/exponential maps are developed for practical computation, with convergence guarantees (Sarrazin et al., 2023).
7. Relationship to Optimal Transport, Hellinger, and Cone Geometry
The SHK metric provides a canonical realization of the “spherical part” of the generalized Hellinger–Kantorovich cone over 5. The SHK metric can be viewed as the angle in the metric cone (for mass-varying HK), with the law of cosines explicitly relating the distances:
6
(Sarrazin et al., 2023, Liero et al., 2015, Liero et al., 2015, Laschos et al., 2017). The SHK thus interpolates between unbalanced transport (HK), classical balanced transport (7), and multiplicative-mass transport (Hellinger), and precisely quantifies the “angle” between probability distributions on a metric space.
Throughout, the SHK distance retains the geometric structure required for robust variational analysis, gradient flows, and dynamic interpolation, making it an indispensable tool in modern unbalanced transport theory.