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.
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.
- 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.