We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations

DBLP - CS Bibliography


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: Bringing UMAP Closer to the Speed of Light with GPU Acceleration

Abstract: The Uniform Manifold Approximation and Projection (UMAP) algorithm has become widely popular for its ease of use, quality of results, and support for exploratory, unsupervised, supervised, and semi-supervised learning. While many algorithms can be ported to a GPU in a simple and direct fashion, such efforts have resulted in inefficent and inaccurate versions of UMAP. We show a number of techniques that can be used to make a faster and more faithful GPU version of UMAP, and obtain speedups of up to 100x in practice. Many of these design choices/lessons are general purpose and may inform the conversion of other graph and manifold learning algorithms to use GPUs. Our implementation has been made publicly available as part of the open source RAPIDS cuML library(this https URL).
Subjects: Machine Learning (cs.LG); Data Structures and Algorithms (cs.DS); Machine Learning (stat.ML)
Cite as: arXiv:2008.00325 [cs.LG]
  (or arXiv:2008.00325v1 [cs.LG] for this version)

Submission history

From: Corey Nolet [view email]
[v1] Sat, 1 Aug 2020 19:35:56 GMT (2302kb,D)
[v2] Wed, 3 Mar 2021 02:27:07 GMT (691kb,D)
[v3] Mon, 29 Mar 2021 09:15:12 GMT (692kb,D)

Link back to: arXiv, form interface, contact.