Skip to main navigation Skip to search Skip to main content

Robustly Learning Mixtures of k Arbitrary Gaussians

  • Ainesh Bakshi
  • , Ilias Diakonikolas
  • , He Jia
  • , Daniel Kane
  • , Pravesh Kothari
  • , Santosh Vempala

Research output: Contribution to journalArticlepeer-review

Abstract

We give a polynomial-time algorithm for the problem of robustly estimating a mixture of k arbitrary Gaussians in ℝd, for any fixed k, in the presence of a constant fraction of arbitrary corruptions. This resolves the main open problem in several previous works on algorithmic robust statistics, which addressed the special cases of robustly estimating (a) a single Gaussian, (b) a mixture of TV-distance separated Gaussians, and (c) a uniform mixture of two Gaussians. Our main tools are an efficient partial clustering algorithm that relies on the sum-of-squares method, and a novel tensor decomposition algorithm that allows errors in both Frobenius norm and low-rank terms.

Original languageEnglish (US)
Article number18
JournalJournal of the ACM
Volume73
Issue number3
DOIs
StatePublished - Jun 17 2026

All Science Journal Classification (ASJC) codes

  • Software
  • Control and Systems Engineering
  • Information Systems
  • Hardware and Architecture
  • Artificial Intelligence

Keywords

  • Gaussian Mixture Model
  • Robust Statistics
  • Sum-of-Squares
  • Tensor Decomposition

Fingerprint

Dive into the research topics of 'Robustly Learning Mixtures of k Arbitrary Gaussians'. Together they form a unique fingerprint.

Cite this