Hierarchy of discriminative power and complexity in learning quantum ensembles

Jian Yao, Pengtao Li, Xiaohui Chen, Quntao Zhuang

arXiv:2601.22005·quant-ph·Published 2026-01-29

Distance metrics are central to machine learning, yet distances between ensembles of quantum states remain poorly understood due to fundamental quantum measurement constraints. We introduce a hierarchy of integral probability metrics, termed MMD-$k$, which generalizes the maximum mean discrepancy to quantum ensembles and exhibit a strict trade-off between discriminative power and statistical efficiency as the moment order $k$ increases. For pure-state ensembles of size $N$, estimating MMD-$k$ using experimentally feasible SWAP-test-based estimators requires $Θ(N^{2-2/k})$ samples for constant $k$, and $Θ(N^3)$ samples to achieve full discriminative power at $k = N$. In contrast, the quantum Wasserstein distance attains full discriminative power with $Θ(N^2 \log N)$ samples. These results provide principled guidance for the design of loss functions in quantum machine learning, which we illustrate in the training quantum denoising diffusion probabilistic models.

TopicsQuantum Physics & Information

Tagsquantum-machine-learning

arXiv categoriesquant-ph, math.ST, stat.ML

arXiv abstract pagePDF