跳到正文
原文
arXiv 机器学习· Mi Li, Eibe Frank, Bernhard Pfahringer·· 10 小时前AI 评分33

利用 GPU 加速 FastDEC 聚类算法

Accelerating FastDEC Clustering Algorithm with GPUs

AI 导读

研究人员提出 GPU 加速的 FastDEC 聚类算法实现 SymWarp-FastDEC,在保持聚类结果与原版完全一致的同时显著提升效率。该方案核心包含利用搜索对称性的 SymTrikNN 自 k-NN 算子,以及消除瓶颈的 warp 并行算法 WP-DC。相比采用 FAISS GpuIndexFlatL2 的变体,其端到端性能提升约 1.7x 至 4.9x,且优势随数据维度增加而进一步扩大。

来源:arXiv 机器学习 · arxiv.org