Bringing UMAP Closer to the Speed of Light with GPU Acceleration
| dc.contributor.author | Nolet, Corey J. | |
| dc.contributor.author | Lafargue, Victor | |
| dc.contributor.author | Raff, Edward | |
| dc.contributor.author | Nanditale, Thejaswi | |
| dc.contributor.author | Oates, Tim | |
| dc.contributor.author | Zedlewski, John | |
| dc.contributor.author | Patterson, Joshua | |
| dc.date.accessioned | 2020-09-11T17:18:18Z | |
| dc.date.available | 2020-09-11T17:18:18Z | |
| dc.date.issued | 2020-08-01 | |
| dc.description.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 | en_US |
| dc.description.sponsorship | We extend our sincerest gratitude to all of those who helped enable our research, especially Philip Hynsu Cho and Dante Gama Dessavre from the RAPIDS cuML team. We would also like to thank the Clara Genomics team at Nvidia, especially Avantika Lal, Johnny Israeli, Raghav Mani, and Neha Tadimeti | en_US |
| dc.description.uri | https://arxiv.org/abs/2008.00325 | en_US |
| dc.format.extent | 19 pages | en_US |
| dc.genre | journal articles preprints | en_US |
| dc.identifier | doi:10.13016/m2sqzx-hybf | |
| dc.identifier.citation | Corey J. Nolet, Victor Lafargue, Edward Raff, Thejaswi Nanditale, Tim Oates, John Zedlewski and Joshua Patterson, Bringing UMAP Closer to the Speed of Light with GPU Acceleration, https://arxiv.org/abs/2008.00325 | en_US |
| dc.identifier.uri | http://hdl.handle.net/11603/19642 | |
| dc.language.iso | en_US | en_US |
| dc.relation.isAvailableAt | The University of Maryland, Baltimore County (UMBC) | |
| dc.relation.ispartof | UMBC Computer Science and Electrical Engineering Department Collection | |
| dc.relation.ispartof | UMBC Faculty Collection | |
| dc.rights | This item is likely protected under Title 17 of the U.S. Copyright Law. Unless on a Creative Commons license, for uses protected by Copyright Law, contact the copyright holder or the author. | |
| dc.title | Bringing UMAP Closer to the Speed of Light with GPU Acceleration | en_US |
| dc.type | Text | en_US |