t-SNE-CUDA: GPU-Accelerated t-SNE and its Applications to Modern Data

DM Chan, R Rao, F Huang… - 2018 30th International …, 2018 - ieeexplore.ieee.org
2018 30th International Symposium on Computer Architecture and …, 2018ieeexplore.ieee.org
Modern datasets and models are notoriously difficult to explore and analyze due to their
inherent high dimensionality and massive numbers of samples. Existing visualization
methods which employ dimensionality reduction to two or three dimensions are often
inefficient and/or ineffective for these datasets. This paper introduces T-SNE-CUDA, a GPU-
accelerated implementation of t-distributed Symmetric Neighbour Embedding (t-SNE) for
visualizing datasets and models. T-SNE-CUDA significantly outperforms current …
Modern datasets and models are notoriously difficult to explore and analyze due to their inherent high dimensionality and massive numbers of samples. Existing visualization methods which employ dimensionality reduction to two or three dimensions are often inefficient and/or ineffective for these datasets. This paper introduces T-SNE-CUDA, a GPU-accelerated implementation of t-distributed Symmetric Neighbour Embedding (t-SNE) for visualizing datasets and models. T-SNE-CUDA significantly outperforms current implementations with 50-700x speedups on the CIFAR-10 and MNIST datasets. These speedups enable, for the first time, visualization of the neural network activations on the entire ImageNet dataset - a feat that was previously computationally intractable. We also demonstrate visualization performance in the NLP domain by visualizing the GloVe embedding vectors. From these visualizations, we can draw interesting conclusions about using the L2 metric in these embedding spaces. T-SNE-CUDA is publicly available at https://github.com/CannyLab/tsne-cuda.
ieeexplore.ieee.org
以上显示的是最相近的搜索结果。 查看全部搜索结果