English

The Theory behind UMAP?

Machine Learning 2026-03-05 v1 Machine Learning Category Theory

Abstract

In 2018, McInnes et al. introduced a dimensionality reduction algorithm called UMAP, which enjoys wide popularity among data scientists. Their work introduces a finite variant of a functor called the metric realization, based on an unpublished draft by Spivak. This draft contains many errors, most of which are reproduced by McInnes et al. and subsequent publications. This article aims to repair these errors and provide a self-contained document with the full derivation of Spivak's functors and McInnes et al.'s finite variant. We contribute an explicit description of the metric realization and related functors. At the end, we discuss the UMAP algorithm, as well as claims about properties of the algorithm and the correspondence of McInnes et al.'s finite variant to the UMAP algorithm.

Cite

@article{arxiv.2603.03375,
  title  = {The Theory behind UMAP?},
  author = {David Wegmann},
  journal= {arXiv preprint arXiv:2603.03375},
  year   = {2026}
}

Comments

This article is derived from my masters thesis

R2 v1 2026-07-01T11:01:52.681Z