Unsupervised Ground Metric Learning

Janis Auffenberg, Jonas Bresch, Oleh Melnyk, Gabriele Steidl
7/17/2025
math.OCcs.LGmath.NA

No official code found. Best implementation: CSDUlm/wsingular (10 stars, MIT, last commit 2024-09-02). 1 more below.

Abstract

Data classification without access to labeled samples remains a challenging problem. It usually depends on an appropriately chosen distance between features, a topic addressed in metric learning. Recently, Huizing, Cantini and Peyré proposed to simultaneously learn optimal transport (OT) cost matrices between samples and features of the dataset. This leads to the task of finding positive eigenvectors of a certain nonlinear function that maps cost matrices to OT distances. Having this basic idea in mind, we consider both the algorithmic and the modeling part of unsupervised metric learning. First, we examine appropriate algorithms and their convergence. In particular, we propose to use the stochastic random function iteration algorithm and prove that it converges linearly for our setting, although our operators are not paracontractive as it was required for convergence so far. Second, we ask the natural question if the OT distance can be replaced by other distances. We show how Mahalanobis-like distances fit into our considerations. Further, we examine an approach via graph Laplacians. In contrast to the previous settings, we have just to deal with linear functions in the wanted matrices here, so that simple algorithms from linear algebra can be applied.

View on arXivView PDF

Code Implementations(2)

CSDUlm/wsingularCommunity65%
MIT

Python package for the ICML 2022 paper "Unsupervised Ground Metric Learning Using Wasserstein Singular Vectors".

101PythonMay 17, 20222 years agoMIT
icmlicml-2022machine-learningmetric-learningoptimal-transport+1 more
kiradust/tree-wsvCommunity62%
MIT

Repository for the 2025 ICLR paper "Fast unsupervised ground metric learning with tree-Wasserstein distance."

10PythonMar 1, 20251 years agoMIT

Similar Papers

Cite this paper

@article{auffenberg2025unsupervised,
  title  = {Unsupervised Ground Metric Learning},
  author = {Janis Auffenberg and Jonas Bresch and Oleh Melnyk and Gabriele Steidl},
  year   = {2025},
  eprint = {2507.13094},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2507.13094v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2507.13094.svg)](https://codeofpaper.com/paper/2507.13094)

Discussion