Clustering mixture models in almost-linear time via list-decodable mean estimation
Ilias Diakonikolas, Daniel M. Kane, Daniel Kongsgaard, Jerry Li, Kevin Tian
Abstract
We study the problem of list-decodable mean estimation, where an adversary can corrupt a majority of the dataset. Specifically, we are given a set T of n points in R d and a parameter 0 < α < 1 2 such that an α-fraction of the points in T are i.i.d. samples from a well-behaved distribution D and the remaining (1-α)-fraction are arbitrary. The goal is to output a small list of vectors, at least one of which is close to the mean of D. We develop new algorithms for listdecodable mean estimation, achieving nearly-optimal statistical guarantees, with running time O(n 1+ǫ0 d), for any fixed ǫ 0 > 0. All prior algorithms for this problem had additional polynomial factors in 1 α . We leverage this result, together with additional techniques, to obtain the first almost-linear time algorithms for clustering mixtures of k separated well-behaved distributions, nearly-matching the statistical guarantees of spectral methods. Prior clustering algorithms inherently relied on an application of k-PCA, thereby incurring runtimes of Ω(ndk). This marks the first runtime improvement for this basic statistical problem in nearly two decades. The starting point of our approach is a novel and simpler near-linear time robust mean estimation algorithm in the α → 1 regime, based on a one-shot matrix multiplicative weightsinspired potential decrease. We crucially leverage this new algorithmic framework in the context of the iterative multi-filtering technique of [DKS18, DKK20a], providing a method to simultaneously cluster and downsample points using one-dimensional projections -thus, bypassing the k-PCA subroutines required by prior algorithms.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f8a82125-92fb-49df-89c6-c9d0b131f4cfCited by top-tier papers19
- Statistical Query Lower Bounds for List-Decodable Linear RegressionIlias Diakonikolas, Daniel Kane, Ankit Pensia, Thanasis Pittas et al.NeurIPS 2021 · 28 citations
- Streaming Algorithms for High-Dimensional Robust StatisticsIlias Diakonikolas, Daniel M. Kane, Ankit Pensia, Thanasis PittasICML 2022 · 25 citations
- Robust Regression Revisited: Acceleration and Improved Estimation RatesArun Jambulapati, Jerry Li, Tselil Schramm, Kevin TianNeurIPS 2021 · 18 citations
- List-Decodable Sparse Mean Estimation via Difference-of-Pairs FilteringIlias Diakonikolas, Daniel Kane, Sushrut Karmalkar, Ankit Pensia et al.NeurIPS 2022 · 16 citations
- List-Decodable Sparse Mean EstimationShiwei Zeng, Jie ShenNeurIPS 2022 · 13 citations
Builds on12
- BREEDS: Benchmarks for Subpopulation ShiftShibani Santurkar, Dimitris Tsipras, Aleksander MadryICLR 2021 · 193 citations
- Robust Sub-Gaussian Principal Component Analysis and Width-Independent Schatten PackingArun Jambulapati, Jerry Li, Kevin TianNeurIPS 2020 · 45 citations
- List Decodable Learning via Sum of SquaresPrasad Raghavendra, Morris YauSODA 2020 · 44 citations
- List-Decodable Mean Estimation via Iterative Multi-FilteringIlias Diakonikolas, Daniel Kane, Daniel KongsgaardNeurIPS 2020 · 23 citations
- Robustly learning mixtures of k arbitrary GaussiansAinesh Bakshi, Ilias Diakonikolas, He Jia, Daniel M. Kane et al.STOC 2022 · 21 citations
Related papers
- List-Decodable Mean Estimation in Nearly-PCA TimeIlias Diakonikolas, Daniel Kane, Daniel Kongsgaard, Jerry Li et al.NeurIPS 2021 · 18 citations
- A Spectral Algorithm for List-Decodable Covariance Estimation in Relative Frobenius NormIlias Diakonikolas, Daniel Kane, Jasper C. H. Lee, Ankit Pensia et al.NeurIPS 2023 · 1 citation
- List-decodable covariance estimationMisha Ivkov, Pravesh K. KothariSTOC 2022 · 5 citations
- Robust Mixture Learning when Outliers Overwhelm Small GroupsDaniil Dmitriev, Rares-Darius Buhai, Stefan Tiegel, Alexander Wolters et al.NeurIPS 2024 · 2 citations
- List Decodable Mean Estimation in Nearly Linear TimeYeshwanth Cherapanamjeri, Sidhanth Mohanty, Morris YauFOCS 2020 · 13 citations
