Settling the robust learnability of mixtures of Gaussians
Allen Liu, Ankur Moitra
Abstract
This work represents a natural coalescence of two important lines of work -learning mixtures of Gaussians and algorithmic robust statistics. In particular, we give the first provably robust algorithm for learning mixtures of any constant number of Gaussians. We require only mild assumptions on the mixing weights and that the total variation distance between components is bounded away from zero. At the heart of our algorithm is a new method for proving a type of dimension-independent polynomial identifiability -which we call robust identifiability -through applying a carefully chosen sequence of differential operations to certain generating functions that not only encode the parameters we would like to learn but also the system of polynomial equations we would like to solve. We show how the symbolic identities we derive can be directly used to analyze a natural sum-of-squares relaxation.
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.
Cited by top-tier papers25
- Polynomial Time and Private Learning of Unbounded Gaussian Mixture ModelsJamil Arbas, Hassan Ashtiani, Christopher LiawICML 2023 · 32 citations
- Streaming Algorithms for High-Dimensional Robust StatisticsIlias Diakonikolas, Daniel M. Kane, Ankit Pensia, Thanasis PittasICML 2022 · 25 citations
- Robustly learning mixtures of k arbitrary GaussiansAinesh Bakshi, Ilias Diakonikolas, He Jia, Daniel M. Kane et al.STOC 2022 · 21 citations
- Learning Quantum Hamiltonians at Any Temperature in Polynomial TimeAinesh Bakshi, Allen Liu, Ankur Moitra, Ewin TangSTOC 2024 · 14 citations
- Nearly-Linear Time and Streaming Algorithms for Outlier-Robust PCAIlias Diakonikolas, Daniel Kane, Ankit Pensia, Thanasis PittasICML 2023 · 11 citations
Builds on4
- Robustly learning mixtures of k arbitrary GaussiansAinesh Bakshi, Ilias Diakonikolas, He Jia, Daniel M. Kane et al.STOC 2022 · 21 citations
- Online and Distribution-Free Robustness: Regression and Contextual Bandits with Huber ContaminationSitan Chen, Frederic Koehler, Ankur Moitra, Morris YauFOCS 2021 · 14 citations
- Robust linear regression: optimal rates in polynomial timeAinesh Bakshi, Adarsh PrasadSTOC 2021 · 13 citations
- Robust Learning of Mixtures of GaussiansDaniel M. KaneSODA 2021 · 12 citations
Related papers
- Outlier-Robust Clustering of Gaussians and Other Non-Spherical MixturesAinesh Bakshi, Ilias Diakonikolas, Samuel B. Hopkins, Daniel Kane et al.FOCS 2020 · 13 citations
- Robust Model Selection and Nearly-Proper Learning for GMMsAllen Liu, Jerry Li, Ankur MoitraNeurIPS 2022 · 5 citations
- List-decodable covariance estimationMisha Ivkov, Pravesh K. KothariSTOC 2022 · 5 citations
- Sample-Efficient Private Learning of Mixtures of GaussiansHassan Ashtiani, Mahbod Majid, Shyam NarayananNeurIPS 2024
- Polynomial-Time Power-Sum Decomposition of PolynomialsMitali Bafna, Jun-Ting Hsieh, Pravesh K. Kothari, Jeff XuFOCS 2022 · 4 citations
