Neighbor Embeddings Using Unbalanced Optimal Transport Metrics

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Rana, Muhammad, Hamm, Keaton
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908555801526272
author Rana, Muhammad
Hamm, Keaton
author_facet Rana, Muhammad
Hamm, Keaton
contents This paper proposes the use of the Hellinger--Kantorovich metric from unbalanced optimal transport (UOT) in a dimensionality reduction and learning (supervised and unsupervised) pipeline. The performance of UOT is compared to that of regular OT and Euclidean-based dimensionality reduction methods on several benchmark datasets including MedMNIST. The experimental results demonstrate that, on average, UOT shows improvement over both Euclidean and OT-based methods as verified by statistical hypothesis tests. In particular, on the MedMNIST datasets, UOT outperforms OT in classification 81\% of the time. For clustering MedMNIST, UOT outperforms OT 83\% of the time and outperforms both other metrics 58\% of the time.
format Preprint
id arxiv_https___arxiv_org_abs_2509_19226
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Neighbor Embeddings Using Unbalanced Optimal Transport Metrics
Rana, Muhammad
Hamm, Keaton
Machine Learning
This paper proposes the use of the Hellinger--Kantorovich metric from unbalanced optimal transport (UOT) in a dimensionality reduction and learning (supervised and unsupervised) pipeline. The performance of UOT is compared to that of regular OT and Euclidean-based dimensionality reduction methods on several benchmark datasets including MedMNIST. The experimental results demonstrate that, on average, UOT shows improvement over both Euclidean and OT-based methods as verified by statistical hypothesis tests. In particular, on the MedMNIST datasets, UOT outperforms OT in classification 81\% of the time. For clustering MedMNIST, UOT outperforms OT 83\% of the time and outperforms both other metrics 58\% of the time.
title Neighbor Embeddings Using Unbalanced Optimal Transport Metrics
topic Machine Learning
url https://arxiv.org/abs/2509.19226