Hacker News·· 2 小时前AI 评分53
e-means 论文以更少训练轮次实现 10–45 倍 k-means 加速
E-means: 10x faster k-means at equal loss
AI 导读
e-means 通过指数移动平均更新、扩展规律自动调参、死簇分裂和专用 GPU 内核,在达到 Lloyd 算法 25 轮测试损失时仅需 1.2 至 3 轮。端到端测试中,GPU 相较 Flash-KMeans 缩短 10–45 倍耗时,CPU 相较 Faiss 缩短 8–15 倍耗时。
来源:Hacker News · e-means.github.io