e-means paper trick: IVF approximate assignment cuts clustering cost 34x at 10B vectors

antoine_chaffin · x · 2026-11-10

Researcher antoinechaffin highlights a method from the e-means paper: index-accelerated approximate assignment. With millions of centroids, exact distance computation becomes prohibitive, so centroids are indexed with IVF (bucketed by meta-centroids) and treated as approximate nearest-neighbor search.

GPU hours to reach Lloyd's test loss on DINO10B subsets: no gain at 100M vectors, 4.3x cheaper at 1B, and 34x cheaper at 10B — training 16.7M centroids in under one GPU-day. He also submitted a PR to TACHIOM supporting the same technique (cpu, HNSW, non-batched) as an alternative to TAC clustering.

Original post →

More from Infra

Infra channel →