Learning Distances from Data with Normalizing Flows and Score Matching

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Sorrenson, Peter, Behrend-Uriarte, Daniel, Schnörr, Christoph, Köthe, Ullrich
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913866997301248
author Sorrenson, Peter
Behrend-Uriarte, Daniel
Schnörr, Christoph
Köthe, Ullrich
author_facet Sorrenson, Peter
Behrend-Uriarte, Daniel
Schnörr, Christoph
Köthe, Ullrich
contents Density-based distances (DBDs) provide a principled approach to metric learning by defining distances in terms of the underlying data distribution. By employing a Riemannian metric that increases in regions of low probability density, shortest paths naturally follow the data manifold. Fermat distances, a specific type of DBD, have attractive properties, but existing estimators based on nearest neighbor graphs suffer from poor convergence due to inaccurate density estimates. Moreover, graph-based methods scale poorly to high dimensions, as the proposed geodesics are often insufficiently smooth. We address these challenges in two key ways. First, we learn densities using normalizing flows. Second, we refine geodesics through relaxation, guided by a learned score model. Additionally, we introduce a dimension-adapted Fermat distance that scales intuitively to high dimensions and improves numerical stability. Our work paves the way for the practical use of density-based distances, especially in high-dimensional spaces.
format Preprint
id arxiv_https___arxiv_org_abs_2407_09297
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Learning Distances from Data with Normalizing Flows and Score Matching
Sorrenson, Peter
Behrend-Uriarte, Daniel
Schnörr, Christoph
Köthe, Ullrich
Machine Learning
Density-based distances (DBDs) provide a principled approach to metric learning by defining distances in terms of the underlying data distribution. By employing a Riemannian metric that increases in regions of low probability density, shortest paths naturally follow the data manifold. Fermat distances, a specific type of DBD, have attractive properties, but existing estimators based on nearest neighbor graphs suffer from poor convergence due to inaccurate density estimates. Moreover, graph-based methods scale poorly to high dimensions, as the proposed geodesics are often insufficiently smooth. We address these challenges in two key ways. First, we learn densities using normalizing flows. Second, we refine geodesics through relaxation, guided by a learned score model. Additionally, we introduce a dimension-adapted Fermat distance that scales intuitively to high dimensions and improves numerical stability. Our work paves the way for the practical use of density-based distances, especially in high-dimensional spaces.
title Learning Distances from Data with Normalizing Flows and Score Matching
topic Machine Learning
url https://arxiv.org/abs/2407.09297