English

Relative Translation Invariant Wasserstein Distance

Machine Learning 2026-05-26 v2 Machine Learning

Abstract

Motivated by the Bures distance, we introduce a new family of distances, \emph{relative translation invariant Wasserstein distances}, denoted by RWpRW_p, as an extension of the classical Wasserstein distances WpW_p for p[1,+)p \in [1, +\infty). We establish that RWpRW_p defines a valid metric and demonstrate that this type of metric is more intrinsic than the classical Wasserstein distance. A bi-level algorithm is designed to compute the general RWpRW_p distance between arbitrary discrete distributions. Moreover, when p=2p = 2, we show that the optimal coupling matrix is invariant under distributional translation in the discrete setting, and we further propose two algorithms, the RW2\mathrm{RW}_2-LP algorithm and the RW2\mathrm{RW}_2-Sinkhorn algorithm, to improve the numerical stability of computing W2W_2 distance and the optimal coupling matrix solutions. Finally, we conduct three experiments to validate our theoretical results and algorithms. The first two experiments report that the RW2\mathrm{RW}_2-LP algorithm and the RW2\mathrm{RW}_2-Sinkhorn algorithm, both with and without normalization, can significantly reduce the numerical errors compared to standard algorithms. The third experiment shows that RWpRW_p algorithms are computationally scalable and applicable to the retrieval of similar thunderstorm patterns in practical applications.

Keywords

Cite

@article{arxiv.2409.02416,
  title  = {Relative Translation Invariant Wasserstein Distance},
  author = {Binshuai Wang and Qiwei Di and Ming Yin and Mengdi Wang and Quanquan Gu and Peng Wei},
  journal= {arXiv preprint arXiv:2409.02416},
  year   = {2026}
}

Comments

Accepted by Transactions on Machine Learning Research (TMLR). Final accepted version. The implementation is publicly available at \url{https://github.com/DRKWang/rw_metric}

R2 v1 2026-06-28T18:33:30.919Z