Clustering a mixture of Gaussians with unknown covariance
We investigate a clustering problem with data from a mixture of Gaussians that share a common but unknown, and potentially ill-conditioned, covariance matrix. We start by considering Gaussian mixtures with two equally-sized components and derive a Max-Cut integer program based on maximum likelihood estimation. We prove its solutions achieve the optimal misclassification rate when the number of samples grows linearly in the dimension, up to a logarithmic factor. However, solving the Max-Cut problem appears to be computationally intractable. To overcome this, we develop an efficient spectral algorithm that attains the optimal rate but requires a quadratic sample size. Although this sample complexity is worse than that of the Max-Cut problem, we conjecture that no polynomial-time method can perform better. Furthermore, we gather numerical and theoretical evidence that supports the existence of a statistical-computational gap. Finally, we generalize the Max-Cut program to a k-means program that handles multi-component mixtures with possibly unequal weights.
Authors
- Damek Davis (ORCID: https://orcid.org/0000-0003-2105-4641)
- Mateo Díaz
- Kaizheng Wang (ORCID: https://orcid.org/0000-0001-6011-493X)
Institutions
- Johns Hopkins University (US)
- Columbia University (US)
- University of Pennsylvania (US)
Publication Details
- Journal
- Bernoulli
- Published
- 2025-04-09
- DOI
- https://doi.org/10.3150/24-bej1798
- Citations
- 2
- Primary Topic
- Bayesian Methods and Mixture Models
- Type
- article
- Field-Weighted Citation Impact
- 2.48