TriMap: Large-scale Dimensionality Reduction Using Triplets

Ehsan Amid, Manfred K. Warmuth

Keywords: Dimensionality Reduction, Triplets, Data Visualization, t-SNE, LargeVis, UMAP

TL;DR: A significantly faster dimensionality reduction method based on triplets that preserves the global structure of the data better than t-SNE, LargeVis, and UMAP.

Abstract: We introduce ``TriMap''; a dimensionality reduction technique based on triplet constraints that preserves the global accuracy of the data better than the other commonly used methods such as t-SNE, LargeVis, and UMAP. To quantify the global accuracy, we introduce a score which roughly reflects the relative placement of the clusters rather than the individual points. We empirically show the excellent performance of TriMap on a large variety of datasets in terms of the quality of the embedding as well as the runtime. On our performance benchmarks, TriMap easily scales to millions of points without depleting the memory and clearly outperforms t-SNE, LargeVis, and UMAP in terms of runtime.

Code: https://drive.google.com/file/d/1oR9DXPmkDt8601N8CQ1l0CVgX6AgTnya/view?usp=sharing

Community Implementations: [ 8 code implementations](https://www.catalyzex.com/paper/arxiv:1910.00204/code)