e-means聚类提速10倍
摘要
e-means用指数移动平均做小批量更新,并按缩放定律自动设定批大小和步长,无需调参。只用1.2–3个epoch就达到Lloyd算法25个epoch的测试损失;GPU上端到端比Flash-KMeans快10–45倍,CPU上比Faiss快8–15倍,损失更低。
原文片段
E-means: 10x faster k-means at equal loss Hacker News 162 points · 11 comments https://news.ycombinator.com/item?id=50429734