English

An Observation on Lloyd's k-Means Algorithm in High Dimensions

Machine Learning 2025-06-19 v1 Machine Learning

Abstract

Clustering and estimating cluster means are core problems in statistics and machine learning, with k-means and Expectation Maximization (EM) being two widely used algorithms. In this work, we provide a theoretical explanation for the failure of k-means in high-dimensional settings with high noise and limited sample sizes, using a simple Gaussian Mixture Model (GMM). We identify regimes where, with high probability, almost every partition of the data becomes a fixed point of the k-means algorithm. This study is motivated by challenges in the analysis of more complex cases, such as masked GMMs, and those arising from applications in Cryo-Electron Microscopy.

Keywords

Cite

@article{arxiv.2506.14952,
  title  = {An Observation on Lloyd's k-Means Algorithm in High Dimensions},
  author = {David Silva-Sánchez and Roy R. Lederman},
  journal= {arXiv preprint arXiv:2506.14952},
  year   = {2025}
}

Comments

27 pages, 3 figures, 4 supplemental figures

R2 v1 2026-07-01T03:22:43.523Z