Robust Sparse Mean Estimation via Sum of Squares

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Diakonikolas, Ilias, Kane, Daniel M., Karmalkar, Sushrut, Pensia, Ankit, Pittas, Thanasis
Natura: Preprint
Pubblicazione: 2022
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913416581480448
author Diakonikolas, Ilias
Kane, Daniel M.
Karmalkar, Sushrut
Pensia, Ankit
Pittas, Thanasis
author_facet Diakonikolas, Ilias
Kane, Daniel M.
Karmalkar, Sushrut
Pensia, Ankit
Pittas, Thanasis
contents We study the problem of high-dimensional sparse mean estimation in the presence of an $ε$-fraction of adversarial outliers. Prior work obtained sample and computationally efficient algorithms for this task for identity-covariance subgaussian distributions. In this work, we develop the first efficient algorithms for robust sparse mean estimation without a priori knowledge of the covariance. For distributions on $\mathbb R^d$ with "certifiably bounded" $t$-th moments and sufficiently light tails, our algorithm achieves error of $O(ε^{1-1/t})$ with sample complexity $m = (k\log(d))^{O(t)}/ε^{2-2/t}$. For the special case of the Gaussian distribution, our algorithm achieves near-optimal error of $\tilde O(ε)$ with sample complexity $m = O(k^4 \mathrm{polylog}(d))/ε^2$. Our algorithms follow the Sum-of-Squares based, proofs to algorithms approach. We complement our upper bounds with Statistical Query and low-degree polynomial testing lower bounds, providing evidence that the sample-time-error tradeoffs achieved by our algorithms are qualitatively the best possible.
format Preprint
id arxiv_https___arxiv_org_abs_2206_03441
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Robust Sparse Mean Estimation via Sum of Squares
Diakonikolas, Ilias
Kane, Daniel M.
Karmalkar, Sushrut
Pensia, Ankit
Pittas, Thanasis
Data Structures and Algorithms
Machine Learning
Statistics Theory
We study the problem of high-dimensional sparse mean estimation in the presence of an $ε$-fraction of adversarial outliers. Prior work obtained sample and computationally efficient algorithms for this task for identity-covariance subgaussian distributions. In this work, we develop the first efficient algorithms for robust sparse mean estimation without a priori knowledge of the covariance. For distributions on $\mathbb R^d$ with "certifiably bounded" $t$-th moments and sufficiently light tails, our algorithm achieves error of $O(ε^{1-1/t})$ with sample complexity $m = (k\log(d))^{O(t)}/ε^{2-2/t}$. For the special case of the Gaussian distribution, our algorithm achieves near-optimal error of $\tilde O(ε)$ with sample complexity $m = O(k^4 \mathrm{polylog}(d))/ε^2$. Our algorithms follow the Sum-of-Squares based, proofs to algorithms approach. We complement our upper bounds with Statistical Query and low-degree polynomial testing lower bounds, providing evidence that the sample-time-error tradeoffs achieved by our algorithms are qualitatively the best possible.
title Robust Sparse Mean Estimation via Sum of Squares
topic Data Structures and Algorithms
Machine Learning
Statistics Theory
url https://arxiv.org/abs/2206.03441