Meta FAIR 提出 e-means:1/10 轮次超越 k-means,GPU 提速最高 45 倍

burny_tech · x · 2026-11-10

Meta FAIR(作者含 Hugo Touvron、Hervé Jégou、Matthijs Douze 等)发布论文《e-means: outperforming k-means (in one-tenth the epochs)》,用深度学习思路改造经典聚类算法。

方法:将 k-means 的质心更新改为基于 mini-batch 的指数滑动平均(EMA)优化,借助现代深度学习框架实现;并建立了衰减率与 batch size 对最终性能影响的 scaling law,可直接按公式设定超参,无需网格搜索。

结果:

聚类广泛用于检索、量化与训练数据筛选,对 LLM 数据管线有直接价值。代码开源于 FAISS 库。

所属事件:Meta FAIR 推出 e-means 聚类算法,GPU 提速最高 45 倍(3 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →