Research & Papers

New theory proves k-means works for missing data with √n rates

Finally, a theoretical guarantee for k-means clustering when data is incomplete—with √n rates and normality.

Deep Dive

Classical k-means clustering fails on incomplete data, and existing missing-data adaptations lack asymptotic theoretical guarantees. Xin Guan's paper fills this gap by rigorously proving the statistical properties of k-means under general missing mechanisms, with deeper results for the Missing Completely at Random (MCAR) case. The work establishes a √n-excess risk bound, consistency of estimated cluster centers, and—under MCAR—a √n-convergence rate and asymptotic normality. These results provide the first theoretical foundation for applying k-means to incomplete datasets with confidence.

The analysis reveals a critical condition: to achieve √n-rate convergence to the true cluster centers, all k true centers must be distinct in every dimension. This poses significant challenges in high-dimensional regimes, where dimensions may be redundant or not all discriminative. The paper also gives a sufficient condition involving missing probability and cluster separation. Numerical simulations on synthetic incomplete datasets support the theory, confirming that the proposed missing-data k-means can match the performance of fully observed clustering under the right conditions.

Key Points
  • Establishes √n-excess risk bounds and consistency for k-means under general missing mechanisms
  • Under MCAR, proves √n-convergence rate and asymptotic normality of cluster centers
  • Requires true cluster centers to be distinct in every dimension for convergence—highlighting high-dimensional challenges

Why It Matters

Rigorous theory enables confident use of k-means on real-world incomplete data, critical for analytics with missing values.

📬 Get the top 10 AI stories daily