iCreatAI News
Skip to content
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