GROMOV–WASSERSTEIN DISTANCES: ENTROPIC REGULARIZATION, DUALITY AND SAMPLE COMPLEXITY

Zhengxin Zhang, Ziv Goldfeld, Youssef Mroueh, Bharath K. Sriperumbudur

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

Abstract

The Gromov–Wasserstein (GW) distance, rooted in optimal transport (OT) theory, quantifies dissimilarity between metric measure spaces and provides a framework for aligning heterogeneous datasets. While computational aspects of the GW problem have been widely studied, a duality theory and fundamental statistical questions concerning empirical convergence rates remained obscure. This work closes these gaps for the quadratic GW distance over Euclidean spaces of different dimensions dx and dy. We treat both the standard and the entropically regularized GW distance, and derive dual forms that represent them in terms of the well-understood OT and entropic OT (EOT) problems, respectively. This enables employing proof techniques from statistical OT based on regularity analysis of dual potentials and empirical process theory, using which we establish the first GW empirical convergence rates. The derived two-sample rates are n−2/max{min{dx,dy},4} (up to a log factor when min{dx,dy} = 4) for standard GW and n−1/2 for entropic GW (EGW), which matches the corresponding rates for standard and entropic OT. The parametric rate for EGW is evidently optimal, while for standard GW we provide matching lower bounds, which establish sharpness of the derived rates. We also study stability of EGW in the entropic regularization parameter and prove approximation and continuity results for the cost and optimal couplings. Lastly, the duality is leveraged to shed new light on the open problem of the one-dimensional GW distance between uniform distributions on n points, illuminating why the identity and anti-identity permutations may not be optimal. Our results serve as a first step towards a comprehensive statistical theory as well as computational advancements for GW distances, based on the discovered dual formulations.

Original languageEnglish (US)
Pages (from-to)1616-1645
Number of pages30
JournalAnnals of Statistics
Volume52
Issue number4
DOIs
StatePublished - Aug 2024

All Science Journal Classification (ASJC) codes

  • Statistics and Probability
  • Statistics, Probability and Uncertainty

Fingerprint

Dive into the research topics of 'GROMOV–WASSERSTEIN DISTANCES: ENTROPIC REGULARIZATION, DUALITY AND SAMPLE COMPLEXITY'. Together they form a unique fingerprint.

Cite this