cuSLINK：GPU 上的单连接凝聚聚类
机器学习
2023-06-29 v1 机器学习
摘要
在本文中，我们提出 cuSLINK，一种新颖且最先进的 GPU 上 SLINK 算法的重构，其仅需 空间并使用参数 来权衡空间与时间。我们还提出一组构成 cuSLINK 的新颖且可复用的构建模块。这些构建模块包括用于 -NN 图构建、生成树和树状图簇提取的高度优化计算模式。我们展示了如何利用我们的原语在 GPU 上端到端实现 cuSLINK，进一步赋能了曾难以处理的广泛真实世界数据挖掘与机器学习应用。除了作为流行 HDBSCAN 算法中的主要计算瓶颈外，我们的端到端 cuSLINK 算法的影响还涵盖大量重要应用，包括社交与计算机网络中的簇分析、自然语言处理和计算机视觉。用户可在 https://docs.rapids.ai/api/cuml/latest/api/#agglomerative-clustering 获取 cuSLINK。
引用
@article{arxiv.2306.16354,
title = {cuSLINK: Single-linkage Agglomerative Clustering on the GPU},
author = {Corey J. Nolet and Divye Gala and Alex Fender and Mahesh Doijade and Joe Eaton and Edward Raff and John Zedlewski and Brad Rees and Tim Oates},
journal= {arXiv preprint arXiv:2306.16354},
year = {2023}
}
备注
To appear in ECML PKDD 2023 by Springer Nature