Unsupervised Ground Metric Learning Using Wasserstein Singular Vectors - Archive ouverte HAL Accéder directement au contenu
Communication Dans Un Congrès Année : 2022

Unsupervised Ground Metric Learning Using Wasserstein Singular Vectors

Geert-Jan Huizing
Laura Cantini

Résumé

Defining meaningful distances between samples in a dataset is a fundamental problem in machine learning. Optimal Transport (OT) lifts a distance between features (the "ground metric") to a geometrically meaningful distance between samples. However, there is usually no straightforward choice of ground metric. Supervised ground metric learning approaches exist but require labeled data. In absence of labels, only ad-hoc ground metrics remain. Unsupervised ground metric learning is thus a fundamental problem to enable data-driven applications of OT. In this paper, we propose for the first time a canonical answer by simultaneously computing an OT distance between samples and between features of a dataset. These distance matrices emerge naturally as positive singular vectors of the function mapping ground metrics to OT distances. We provide criteria to ensure the existence and uniqueness of these singular vectors. We then introduce scalable computational methods to approximate them in high-dimensional settings, using stochastic approximation and entropic regularization. Finally, we showcase Wasserstein Singular Vectors on a single-cell RNA-sequencing dataset.
Fichier principal
Vignette du fichier
huizing22a.pdf (2.59 Mo) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03828733 , version 1 (25-10-2022)

Identifiants

  • HAL Id : hal-03828733 , version 1

Citer

Geert-Jan Huizing, Laura Cantini, Gabriel Peyré. Unsupervised Ground Metric Learning Using Wasserstein Singular Vectors. Proceedings of the 39th International Conference on Machine Learning,, Jul 2022, Baltimore, United States. ⟨hal-03828733⟩
20 Consultations
10 Téléchargements

Partager

Gmail Facebook X LinkedIn More