An Observation on Lloyd's k-Means Algorithm in High Dimensions
Silva-Sánchez, David, Lederman, Roy R.
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.
Jun-19-2025
- Country:
- North America > United States > Connecticut > New Haven County > New Haven (0.04)
- Genre:
- Research Report (1.00)
- Technology: