MarkTechPost· Michal Sutter·· 5 days agoAI score50
Meta FAIR Releases e-means: An Algorithm Beating k-means in 10x Fewer Epochs
Meta FAIR Releases e-means: An Algorithm Beating k-means in 10x Fewer Epochs
AI summary
Meta FAIR researchers have released e-means, a mini-batch k-means algorithm that updates centroids with exponential moving averages and reaches Lloyd's k-means 25-epoch loss in 1.4 to 2.6 epochs. It ships in FAISS 1.16.0 under faiss.contrib.e_means and runs on CPU or GPU with PyTorch tensors or NumPy. The authors self-report 10-45x GPU wall-clock speedup over Flash-KMeans at equal loss and 8-15x CPU speedup over Faiss k-means, with two fitted scaling laws setting batch size and initial step size instead of manual tuning.
Source: MarkTechPost · marktechpost.com