English

List-Decodable Covariance Estimation

Data Structures and Algorithms 2022-06-23 v1 Machine Learning Statistics Theory Machine Learning Statistics Theory

Abstract

We give the first polynomial time algorithm for \emph{list-decodable covariance estimation}. For any α>0\alpha > 0, our algorithm takes input a sample YRdY \subseteq \mathbb{R}^d of size ndpoly(1/α)n\geq d^{\mathsf{poly}(1/\alpha)} obtained by adversarially corrupting an (1α)n(1-\alpha)n points in an i.i.d. sample XX of size nn from the Gaussian distribution with unknown mean μ\mu_* and covariance Σ\Sigma_*. In npoly(1/α)n^{\mathsf{poly}(1/\alpha)} time, it outputs a constant-size list of k=k(α)=(1/α)poly(1/α)k = k(\alpha)= (1/\alpha)^{\mathsf{poly}(1/\alpha)} candidate parameters that, with high probability, contains a (μ^,Σ^)(\hat{\mu},\hat{\Sigma}) such that the total variation distance TV(N(μ,Σ),N(μ^,Σ^))<1Oα(1)TV(\mathcal{N}(\mu_*,\Sigma_*),\mathcal{N}(\hat{\mu},\hat{\Sigma}))<1-O_{\alpha}(1). This is the statistically strongest notion of distance and implies multiplicative spectral and relative Frobenius distance approximation for parameters with dimension independent error. Our algorithm works more generally for (1α)(1-\alpha)-corruptions of any distribution DD that possesses low-degree sum-of-squares certificates of two natural analytic properties: 1) anti-concentration of one-dimensional marginals and 2) hypercontractivity of degree 2 polynomials. Prior to our work, the only known results for estimating covariance in the list-decodable setting were for the special cases of list-decodable linear regression and subspace recovery due to Karmarkar, Klivans, and Kothari (2019), Raghavendra and Yau (2019 and 2020) and Bakshi and Kothari (2020). These results need superpolynomial time for obtaining any subconstant error in the underlying dimension. Our result implies the first polynomial-time \emph{exact} algorithm for list-decodable linear regression and subspace recovery that allows, in particular, to obtain 2poly(d)2^{-\mathsf{poly}(d)} error in polynomial-time. Our result also implies an improved algorithm for clustering non-spherical mixtures.

Keywords

Cite

@article{arxiv.2206.10942,
  title  = {List-Decodable Covariance Estimation},
  author = {Misha Ivkov and Pravesh K. Kothari},
  journal= {arXiv preprint arXiv:2206.10942},
  year   = {2022}
}

Comments

Abstract slightly clipped. To appear at STOC 2022

R2 v1 2026-06-24T11:59:49.084Z