Power-Deformed Bures–Wasserstein Metric
- The metric is defined by deforming classical Bures–Wasserstein geometry with a power parameter that shifts the linearization base to interpolate between established fidelities.
- It unifies quantum-state fidelity measures—recovering Uhlmann, Holevo, and Matsumoto fidelities—and extends naturally to Rényi divergences and advanced SPD transport metrics.
- Its practical applications include robust representation learning and Riemannian batch normalization in SPD deep networks, improving stability in transport-based learning.
The power-deformed generalized Bures–Wasserstein metric denotes a family of constructions that deform classical Bures–Wasserstein geometry by a power parameter while preserving a Riemannian-geometric interpretation. In the quantum-state setting, the deformation is realized by changing the base point at which the Bures–Wasserstein manifold is linearized, producing generalized fidelities and generalized Bures distances that interpolate among Uhlmann, Holevo, and Matsumoto fidelities, and that extend naturally to Rényi-type divergences (Afham et al., 2024). In the symmetric positive definite (SPD) setting, the deformation is realized by combining a generalized Bures–Wasserstein metric parameterized by an SPD matrix with a matrix-power diffeomorphism, yielding a pullback metric used in representation learning and Riemannian batch normalization (Wang et al., 1 Apr 2025). These constructions sit within a broader line of work that relates Bures–Wasserstein geometry to Gaussian optimal transport, generalized Wasserstein costs, and matrix-valued transport metrics (Han et al., 2021, Li et al., 2020, Brenier et al., 2018).
1. Classical Bures–Wasserstein geometry
Let
be the manifold of density operators, and let
be the cone of positive definite matrices. The tangent space at is
while on the tangent space is without the trace constraint. The Bures, or symmetric logarithmic derivative, Riemannian metric is defined through the Lyapunov operator , where is the unique solution of , by
On 0, the corresponding inner product is
1
The associated Bures–Wasserstein distance is
2
It coincides with the 3-Wasserstein distance between zero-mean Gaussian measures when 4 and 5 are covariance matrices, and the Bures–Wasserstein geodesic is the displacement interpolation in that setting (Afham et al., 2024).
The Bures–Wasserstein geometry admits closed-form geodesic, exponential, and logarithm maps. If 6, with geometric mean
7
then
8
9
and
0
In the SPD-manifold literature, the same geometry is written with a real-symmetric Lyapunov operator 1 solving
2
and metric tensor
3
highlighting the same linear dependence on the eigenvalues of the base point that later motivates robustness to ill-conditioning (Wang et al., 1 Apr 2025).
2. Generalized fidelity and generalized Bures distance
A generalized fidelity is obtained by fixing a base point 4 and defining
5
Equivalent forms include
6
where 7 and 8 are unitary polar factors. The generalized squared Bures–Wasserstein distance at base 9 is then
0
with
1
For fixed 2, 3 is a bona fide metric on 4: it is symmetric, non-negative, vanishes only when 5, and satisfies the triangle inequality (Afham et al., 2024).
The central geometric fact is that 6 is the natural distance induced by linearizing the Bures–Wasserstein manifold at 7. The tangent-space characterization states
8
where
9
An equivalent Hilbert–Schmidt characterization is
0
These formulas make explicit that the generalized distance is not an ad hoc modification of fidelity, but the squared norm of a difference vector in the linearized Bures–Wasserstein tangent space (Afham et al., 2024).
Several standard fidelities are recovered by special choices of the base.
| Base choice | Recovered fidelity | Condition |
|---|---|---|
| 1 or 2 | Uhlmann fidelity 3 | Also along the BW geodesic segment between 4 and 5 |
| 6 | Holevo fidelity 7 | Exact reduction |
| 8 or 9 | Matsumoto fidelity 0 | Also along specified AI/Euc inverse paths |
This reduction pattern is one of the defining features of the generalized construction: the base point selects which classical comparator is recovered (Afham et al., 2024).
3. Power deformation by moving the linearization base
The power-deformed generalized Bures–Wasserstein metric arises by moving the base point along affine-invariant geodesics connecting a state to its inverse. For 1, one sets 2 or 3 and defines
4
together with
5
Explicitly,
6
where 7, and similarly
8
A symmetrized polar fidelity is
9
At 0, these formulas recover Uhlmann, Holevo, and Matsumoto fidelities, respectively (Afham et al., 2024).
A commonly used parameterization fixes a reference state 1 and sets
2
The 3-deformed generalized fidelity and distance are
4
5
Equivalently,
6
The special values are
7
Thus 8 yields Holevo fidelity directly, while the endpoints reproduce Uhlmann-type and Matsumoto-type behavior under the corresponding geodesic conditions (Afham et al., 2024).
The same source relates this deformation to monotone quantum Fisher-information metrics. The Bures metric corresponds to the symmetric logarithmic derivative, the right logarithmic derivative is linked to Matsumoto’s fidelity, and Holevo-related quantities lie between them. The path
9
therefore provides a tractable interpolation from SLD/Bures at 0, through Holevo at 1, to RLD/Matsumoto at 2. Numerically, along the AI-power paths 3 and 4, generalized fidelity appears monotone in 5 and bounded between Matsumoto and Uhlmann fidelities, in line with the inequalities 6 (Afham et al., 2024).
4. Path invariance, purification, and divergence extensions
The generalized fidelity is not arbitrary in its dependence on the base. When the base lies on specific geodesic-related paths, the generalized quantity collapses to familiar fidelities. If 7 lies on the Bures–Wasserstein geodesic 8 between 9 and 0, or on the inverse of the Bures–Wasserstein geodesic between 1 and 2, then
3
for all 4. If 5 lies on certain affine-invariant or Euclidean geodesic paths between inverses, or on the inverse of a Euclidean geodesic between 6 and 7, then
8
There are also covariance identities for mixed Bures–Wasserstein paths, such as
9
These results formalize the statement that changing the linearization point can preserve, rather than alter, the comparator when the motion of the base is synchronized with the geometry of 0 and 1 (Afham et al., 2024).
The generalized fidelity also admits an SDP-flavored block-matrix characterization. For the primal SDP with block-diagonal constraint matrix 2 and optimal Gram matrix
3
one obtains
4
and
5
An Uhlmann-like theorem complements this representation: if 6 and
7
then
8
Generalized fidelity is therefore a carefully chosen purification overlap, extending the role played by Uhlmann’s theorem (Afham et al., 2024).
The same framework extends to multivariate fidelities and Rényi divergences. For states 9 and base 00, the generalized multivariate fidelity is
01
If 02 maximizes 03, equivalently is the Bures–Wasserstein barycenter up to normalization, then
04
For Rényi-type quantities, the base-dependent trace functional
05
induces
06
Specific bases recover Petz, sandwiched, reverse-sandwiched, and geometric Rényi divergences:
07
respectively. In this sense, the base acts as the deformation variable that unifies several non-commutative Rényi quantizations (Afham et al., 2024).
5. Generalized Bures–Wasserstein metrics on the SPD manifold
A distinct but closely related formulation was introduced for SPD matrices. For 08, the generalized Bures–Wasserstein metric is parameterized by an SPD matrix 09 and defined by
10
where the generalized Lyapunov operator solves
11
When 12, this reduces to the classical Bures–Wasserstein metric. When 13, it coincides locally with the affine-invariant metric. The associated distance can be seen as the Bures–Wasserstein distance between congruence-transformed matrices, and the geometry is Riemannian-isometric to the Bures–Wasserstein geometry under the congruence map (Han et al., 2021, Wang et al., 1 Apr 2025).
The power-deformed GBWM introduced for learning on SPD manifolds combines this metric with the matrix-power diffeomorphism
14
and defines
15
The deformation therefore acts by transporting tangent vectors through the matrix-power map and rescaling the metric by 16. The limit 17 is a Log-Euclidean-type, or LEM-like, metric:
18
and locally
19
The deformation interpolates between GBWM at 20 and an LEM-like regime as 21 (Wang et al., 1 Apr 2025).
The isometric structure is central. If
22
then 23 is a Riemannian isometry from 24 to 25. Consequently, exponential maps, logarithms, parallel transport, distances, and weighted Fréchet means in the deformed geometry are computed by mapping to the Bures–Wasserstein manifold, performing the operation there, and mapping back. This construction supplies the geometric basis for the deformed Riemannian batch-normalization layer used in SPD deep networks (Wang et al., 1 Apr 2025).
6. Algorithms and learning applications
For the generalized fidelity 26 in the quantum-state formulation, a direct evaluation uses polar factors. One computes
27
for example through
28
and then evaluates
29
followed by
30
Computing 31 uses spectral decomposition or Schur methods, with complexity 32. An alternative uses Lyapunov solves to evaluate logarithm maps and tangent norms, with Bartels–Stewart or sign-function methods noted for numerical stability. In the power-deformed case, the base 33 or 34 is computed by spectral decomposition (Afham et al., 2024).
In the SPD-learning formulation, the deformed geometry is implemented isometrically. For deformed GBWM batch normalization, one maps the input batch to Bures–Wasserstein coordinates via
35
computes the Bures–Wasserstein batch mean and variance, performs centering, scaling, and biasing in the Bures–Wasserstein manifold, and maps back using
36
The batch variance is normalized by 37 because of the metric scaling. Differentiation relies on the Daleckii–Kreĭn formula for matrix powers and square roots and on explicit gradients for Lyapunov operators (Wang et al., 1 Apr 2025).
Empirical validation was reported on HDM05 action recognition, NTU RGB+D action recognition, and MAMEM-SSVEP-II EEG classification, with SPDNet and RResNet backbones. The reported results include 38 for SPDNet-GBWBN with 39 on MAMEM-SSVEP-II, 40 for RResNet-GBWBN with 41 on HDM05, and 42 on NTU RGB+D. The ablation on 43 states that deformed metrics with 44 generally improve over 45, and the experiments are presented as evidence that the learnable metric parameter 46 and the power deformation improve robustness on ill-conditioned SPD matrices (Wang et al., 1 Apr 2025).
7. Related transport frameworks and open problems
The power-deformed generalized Bures–Wasserstein metric should be distinguished from broader matrix-valued transport generalizations. The generalized BW geometry of SPD matrices parameterized by 47 was formulated earlier as a Mahalanobis-cost extension of Gaussian Bures–Wasserstein geometry, with explicit geodesics, exponential and logarithm maps, Levi-Civita connection, curvature bounds, and barycenter equations (Han et al., 2021). At a still broader level, weighted Wasserstein–Bures metrics on positive semidefinite matrix-valued Radon measures define complete geodesic spaces with conic structure through a convex Benamou–Brenier formulation. In that framework, a power deformation can be introduced by replacing the weight pair 48 with 49, preserving convexity, duality, geodesicity, and cone structure for each fixed 50 (Li et al., 2020). The Kantorovich–Bures metric on matrix-valued measures provides another transport–reaction geometry whose constant-in-space reduction recovers the classical Bures metric, and whose structure indicates where an 51-power deformation could be inserted, although the paper itself does not define such a deformation (Brenier et al., 2018).
Several open questions remain explicit in the generalized-fidelity literature. Data processing inequality and joint concavity for generalized fidelities and distances at arbitrary bases remain open; initial numerics are reported as showing no data-processing-inequality violations, but a proof is outstanding. A true SDP formulation of the generalized fidelity, more precisely of its real part, is not yet known: the current characterization is SDP-flavored rather than an SDP feasibility or optimization problem. Also open is the characterization of the image of the unitary factors 52 over all bases 53 as 54 (Afham et al., 2024).
Taken together, these developments define a coherent geometric program. In the quantum-state setting, power deformation means changing the point of Bures–Wasserstein linearization through 55 or 56, thereby interpolating among named fidelities, monotone metrics, and Rényi divergences. In the SPD-learning setting, power deformation means pulling back the generalized Bures–Wasserstein metric through 57, thereby producing a learnable and numerically robust geometry for normalization and optimization. A plausible implication is that the phrase “power-deformed generalized Bures–Wasserstein metric” now refers less to a single formula than to a common design principle: introduce a power parameter at the level of the geometric base or of the manifold coordinates, and use the induced pullback or linearized metric to tune the non-commutative geometry without abandoning the Bures–Wasserstein framework (Afham et al., 2024, Wang et al., 1 Apr 2025).