---
title: Quantum Wasserstein Distance in Generative Models
url: https://www.emergentmind.com/topics/quantum-wasserstein-distance
type: topic
---

# Quantum Wasserstein Distance in Generative Models

Quantum Wasserstein Distance is a statistical metric adapted to measure the distance between probability distributions over quantum state spaces. It generalizes the classical Wasserstein distance—central to optimal transport theory—to quantum probability distributions and has become integral to the development and evaluation of quantum generative models, particularly quantum denoising diffusion probabilistic models (QuDDPM) [2310.05866].

## 1. Mathematical Formulation of Quantum Wasserstein Distance

In quantum generative modeling, quantum states are represented by density matrices $\rho, \sigma$ on a Hilbert space of dimension $d=2^n$. The quantum Wasserstein-$p$ distance provides a notion of "cost to transport" between two ensembles of such states. 

Unlike the classical setting, where probability distributions are scalar functions, quantum states are matrices with constraints: Hermitian, positive semidefinite, and trace-one. To apply optimal transport, distance matrices (cost matrices) are constructed, with entries reflecting physical distances or dissimilarities between quantum states. Two widely utilized forms include:

- For pure states: $C_{i,j} = 1 - |\langle\psi_i|\phi_j\rangle|^2$.
- For mixed states: $C_{i,j} = 1 - G(\rho_i, \sigma_j)$, where $G(\rho, \sigma)$ is the superfidelity $G(\rho, \sigma) = \operatorname{Tr}(\rho\sigma) + \sqrt{\left[1-\operatorname{Tr}(\rho^2)\right]\left[1-\operatorname{Tr}(\sigma^2)\right]}$ [2411.17608].

Given quantum ensembles $\{\rho_i\}$ and $\{\sigma_j\}$, one solves the linear program:

$$
\min_P \langle P, C \rangle \qquad
\text{s.t. } P\mathbf{1}_w = r, \;\; P^\top\mathbf{1}_v = s, \;\; P \geq 0
$$

where $P$ is the transport matrix, and $r, s$ are uniform probability weights over samples.

## 2. Role in Quantum Diffusion Models

The quantum Wasserstein distance is explicitly used as a loss functional in multiple quantum generative frameworks for matching the forward-diffused (noisy) ensemble to the learned (denoised) ensemble [2310.05866][2404.06336][2411.17608]:

- In QuDDPM, training objectives may minimize the $p$-Wasserstein distance between empirical ensembles at each diffusion step. This enables robust learning even where fidelity-based kernel methods such as MMD (Maximum Mean Discrepancy) become ill-posed or degenerate—e.g., in quantum data with nontrivial topologies (such as states distributed on a Bloch sphere ring) [2310.05866].
- Structure-preserving diffusion models (SPDM) further elevate Wasserstein distances (including sliced and maximum sliced variants) to quantify the generative quality of quantum mixed-state ensembles and the entanglement structure of their output [2404.06336].

These schemes require efficient algorithms for evaluating pairwise quantum cost matrices and for solving the optimal assignment problem in high dimensions.

## 3. Properties and Computational Aspects

Quantum Wasserstein distances retain key properties from the classical case:

- Metric properties: nonnegativity, identity of indiscernibles, and (when suitably defined) the triangle inequality.
- Connection with physical observables: For certain choices of the cost function, the Wasserstein distance reflects operationally meaningful measures, e.g., differences in magnetization or entanglement negativity distributions in many-body quantum states [2310.05866][2404.06336].

Sample complexity for estimating Wasserstein distance scales polynomially in the number of qubits $n$ when using empirical ensembles, provided each ensemble is of polynomial size [2310.05866]. In mixed-state models, structure-preserving architectures ensure all sampled states satisfy quantum constraints (Hermiticity, positivity, trace-one), enabling robust cost matrix construction without post hoc correction [2404.06336][2411.17608].

Efficient quantum Wasserstein computation has been demonstrated via linear programming (for discrete samples) and is tractable for $d\sim2^n$ when $n\leq4$. For larger systems, sliced variants or kernel approximations reduce complexity [2404.06336]. Quantum optimal transport algorithms for physical hardware remain a topic of ongoing research.

## 4. Applications in Quantum Generative Modeling

Quantum Wasserstein distance has been essential in the empirical and theoretical validation of quantum generative models:

- Topological quantum data: Wasserstein metrics can distinguish nonlinear structures, such as Bloch-circle ensembles, and avoid degeneracy inherent in MMD kernels on highly symmetric spaces [2310.05866].
- Mixed-state ensemble synthesis: SPDM uses the Wasserstein distance as its primary generative metric, analyzing recovery of known quantum negativity and eigenvalue distributions in 4-qubit mixed states [2404.06336].
- Many-body phase reconstruction: Wasserstein distance between true and generated magnetization distributions quantifies physical faithfulness of the generated quantum many-body ground states [2310.05866][2411.17608].

Empirical values in benchmark studies are typically on the order of $10^{-4}$ to $10^{-2}$ for high-fidelity models, and increase to $>0.1$ for random state references or under poorly trained architectures [2404.06336].

## 5. Limitations, Open Questions, and Future Directions

Quantum Wasserstein distances, while versatile, have several limitations arising from the high-dimensional structure and peculiarities of quantum data:

- Scalability: Exact computation is restricted to small $n$ due to the $2^n\times 2^n$ nature of cost matrices; approximations or sliced methods are necessary for larger quantum systems [2404.06336].
- Degeneracy: Certain data topologies or noise schedules may cause Wasserstein metrics to become ill-defined or degenerate; selection of appropriate cost functions remains an open challenge [2310.05866].
- Hardware implementation: Current approaches rely mostly on classical postprocessing of quantum-generated samples. Quantum-native computation of Wasserstein distance on hardware is not yet practical for $n>10$.

Open directions include the extension of Wasserstein-like metrics to continuous-variable quantum systems, the development of hardware-tailored sliced Wasserstein estimation, and the analytic characterization of how quantum Wasserstein distance reflects underlying physical properties (e.g., entanglement spectrum, phase transitions) in generative quantum models [2404.06336][2310.05866].

## 6. Comparative Context Within Quantum Generative Metrics

In quantum generative model training and evaluation, Wasserstein distance is frequently compared with other metrics:

| Metric | Fundamental Object | Typical Use Case / Context |
|--------|-------------------|---------------------------|
| Quantum Wasserstein | Cost matrix over state pairs | Topological quantum data, many-body ensemble generation |
| Quantum MMD (fidelity kernel) | Pairwise overlaps $\operatorname{Tr}(\rho\sigma)$ | Rapid estimation for pure states, but susceptible to degeneracy |
| Superfidelity | $G(\rho,\sigma)$ upper bounds Uhlmann fidelity | Mixed-state metric, efficient computation [2411.17608] |

Wasserstein distance is preferred where the geometry of quantum data is nontrivial, or where kernel-based metrics fail to capture essential generative quality [2310.05866][2404.06336][2411.17608]. This suggests that the quantum Wasserstein distance will remain a central tool in the evaluation, optimization, and analysis of quantum generative learning models.

Source: https://www.emergentmind.com/topics/quantum-wasserstein-distance