工具slug: flash-kmeans

Flash-KMeans

面向 GPU 的高性能 k-means 实现,作为 e-means 的对比基线

收录情报动态

共 1 篇 · 第 1/1 页
Hacker News AIT2·基座大模型
78 分
e-means 让 k-means 提速 45 倍且损失更低

Meta 与 École Polytechnique 等机构研究者提出 e-means,在同等或更低测试损失下仅需 1.2 至 3 个 epoch 即可达到 Lloyd 算法 25 个 epoch 的效果。端到端实测在 GPU 上相对 Flash-KMeans 实现 10 至 45 倍加速,CPU 上相对 Faiss 提速 8 至 15 倍,同时打破 k-means 中速度与质量此消彼长的传统权衡。