arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 1802.04450cs.DCcs.MS
收藏

谱聚类算法在CPU-GPU平台上的高性能实现

A High Performance Implementation of Spectral Clustering on CPU-GPU Platforms

Yu Jin, Joseph F. JaJa

摘要:

谱聚类是一种最流行的图聚类算法,它在许多科学和工程应用中取得了最好的性能。然而,常用软件平台(如Matlab和Python)中的现有实现对于许多新兴的大数据应用程序来说并不能很好地扩展。在本文中,我们提出了一个快速实现的谱聚类算法的CPU-GPU异构平台上。我们的实现利用了多核CPU的计算能力和GPU的大规模多线程和SIMD功能。给定的输入数据点在高维空间中,我们提出了一个并行的计划来建立一个稀疏的相似性图表示在一个标准的稀疏表示格式。然后利用ARPACK软件和cuSPARSE库的反向通信接口计算拉普拉斯矩阵的最小$k$特征向量,其中$k$通常非常大.此外,我们在GPU上实现了一个非常快速的并行$k$ -means算法。我们的实现被证明是显着更快相比,最知名的Matlab和Python实现的每一步。此外,我们的算法扩展到具有非常大量的集群的问题。

英文摘要:

Spectral clustering is one of the most popular graph clustering algorithms, which achieves the best performance for many scientific and engineering applications. However, existing implementations in commonly used software platforms such as Matlab and Python do not scale well for many of the emerging Big Data applications. In this paper, we present a fast implementation of the spectral clustering algorithm on a CPU-GPU heterogeneous platform. Our implementation takes advantage of the computational power of the multi-core CPU and the massive multithreading and SIMD capabilities of GPUs. Given the input as data points in high dimensional space, we propose a parallel scheme to build a sparse similarity graph represented in a standard sparse representation format. Then we compute the smallest $k$ eigenvectors of the Laplacian matrix by utilizing the reverse communication interfaces of ARPACK software and cuSPARSE library, where $k$ is typically very large. Moreover, we implement a very fast parallelized $k$-means algorithm on GPUs. Our implementation is shown to be significantly faster compared to the best known Matlab and Python implementations for each step. In addition, our algorithm scales to problems with a very large number of clusters.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑