Randomly initialized EM algorithm for two-component Gaussian mixture achieves near optimality in O(√n) iterations
Not on the map. Why?
Abstract (arXiv)
We analyze the classical EM algorithm for parameter estimation in the symmetric two-component Gaussian mixtures in d dimensions. We show that, even in the absence of any separation between components, provided that the sample size satisfies n=Omega(d log3 d), the randomly initialized EM algorithm converges to an estimate in at most O(√n) iterations with high probability, which is at most O((fracd log3 nn)1/4) in Euclidean distance from the true parameter and within logarithmic factors of the minimax rate of (fracdn)1/4. Both the nonparametric statistical rate and the sublinear convergence rate are direct consequences of the zero Fisher information in the worst case. Refined pointwise guarantees beyond worst-case analysis and convergence to the MLE are also shown under mild conditions. This improves the previous result of Balakrishnan et al citeBWY17 which requires strong conditions on both the separation of the components and the quality of the initialization, and that of Daskalakis et al citeDTZ17 which requires sample splitting and restarting the EM iteration.
Key facts. Cited by 27 papers in the map (data through 14 Oct 2026). First posted on arXiv 28 Aug 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 3; 2022: 3; 2023: 9; 2024: 7; 2025: 3; 2026: 2.
27 papers.
Sort:
By area:
- 2023Demystifying Softmax Gating Function in Gaussian Mixture of Expertscited by 33
- 2023Over-Parameterization Exponentially Slows Down Gradient Descent for Learning a Single Neuroncited by 25
- 2023On the Convergence of Coordinate Ascent Variational Inferencecited by 18
- 2023How Over-Parameterization Slows Down Gradient Descent in Matrix Sensing: The Curses of Symmetry and Initializationcited by 18
- 2024How much is a noisy image worth? Data Scaling Laws for Ambient Diffusioncited by 18
- 2021Clustering a Mixture of Gaussians with Unknown Covariancecited by 14
- 2021Sharp global convergence guarantees for iterative nonconvex optimization: A Gaussian process perspectivecited by 10
- 2021The EM Algorithm is Adaptively-Optimal for Unbalanced Symmetric Gaussian Mixturescited by 6
- 2022Optimal Estimation and Computational Limit of Low-rank Gaussian Mixturescited by 6
- 2024Toward Global Convergence of Gradient EM for Over-Parameterized Gaussian Mixture Modelscited by 6
- 2023Can semi-supervised learning use all the data effectively? A lower bound perspectivecited by 5
- 2024Matrix Denoising with Doubly Heteroscedastic Noise: Fundamental Limits and Optimal Spectral Methodscited by 5
15 more
- 2024Semi-Supervised Sparse Gaussian Classification: Provable Benefits of Unlabeled Datacited by 3
- 2025Nonparametric MLE for Gaussian Location Mixtures: Certified Computation and Generic Behaviorcited by 3
- 2025Convergence Dynamics of Over-Parameterized Score Matching for a Single Gaussiancited by 3
- 2022Likelihood Adjusted Semidefinite Programs for Clustering Heterogeneous Datacited by 2
- 2023Sharp-SSL: Selective high-dimensional axis-aligned random projections for semi-supervised learningcited by 2
- 2025Dendrograms of Mixing Measures for Softmax-Gated Gaussian Mixture of Experts: Consistency Without Model Sweepscited by 2
- 2023Sharp analysis of EM for learning mixtures of pairwise differencescited by 1
- 2023EM for Mixture of Linear Regression with Clustered Datacited by 1
- 2023Joint Problems in Learning Multiple Dynamical Systemscited by 1
- 2024Factor Adjusted Spectral Clustering for Mixture Modelscited by 1
- 2024In-context Learning for Mixture of Linear Regressions: Existence, Generalization and Training Dynamicscited by 1
- 2022Mean Estimation in High-Dimensional Binary Markov Gaussian Mixture Modelscited by 0
- 2024Adaptive Mean Estimation in the Hidden Markov sub-Gaussian Mixture Modelcited by 0
- 2026One-Bit Clustering for Two Component Sub-Gaussian Mixture Modelscited by 0
- 2026Iterative Atom Refinement: A Monotonicity Principle for Dictionary Learningcited by 0
- 2026Iterative Atom Refinement: A Monotonicity Principle for Dictionary Learning20 Sep 2026
- 2026One-Bit Clustering for Two Component Sub-Gaussian Mixture Models20 Jun 2026
- 2025Convergence Dynamics of Over-Parameterized Score Matching for a Single Gaussian27 Nov 2025
- 2025Dendrograms of Mixing Measures for Softmax-Gated Gaussian Mixture of Experts: Consistency Without Model Sweeps14 Oct 2025
- 2025Nonparametric MLE for Gaussian Location Mixtures: Certified Computation and Generic Behavior26 Mar 2025
- 2024How much is a noisy image worth? Data Scaling Laws for Ambient Diffusion5 Nov 2024
- 2024In-context Learning for Mixture of Linear Regressions: Existence, Generalization and Training Dynamics18 Oct 2024
- 2024Semi-Supervised Sparse Gaussian Classification: Provable Benefits of Unlabeled Data5 Sep 2024
- 2024Factor Adjusted Spectral Clustering for Mixture Models22 Aug 2024
- 2024Toward Global Convergence of Gradient EM for Over-Parameterized Gaussian Mixture Models29 Jun 2024
- 2024Adaptive Mean Estimation in the Hidden Markov sub-Gaussian Mixture Model18 Jun 2024
- 2024Matrix Denoising with Doubly Heteroscedastic Noise: Fundamental Limits and Optimal Spectral Methods22 May 2024
15 more
- 2023Can semi-supervised learning use all the data effectively? A lower bound perspective30 Nov 2023
- 2023Joint Problems in Learning Multiple Dynamical Systems3 Nov 2023
- 2023How Over-Parameterization Slows Down Gradient Descent in Matrix Sensing: The Curses of Symmetry and Initialization3 Oct 2023
- 2023EM for Mixture of Linear Regression with Clustered Data22 Aug 2023
- 2023On the Convergence of Coordinate Ascent Variational Inference1 Jun 2023
- 2023Demystifying Softmax Gating Function in Gaussian Mixture of Experts5 May 2023
- 2023Sharp-SSL: Selective high-dimensional axis-aligned random projections for semi-supervised learning18 Apr 2023
- 2023Sharp analysis of EM for learning mixtures of pairwise differences20 Feb 2023
- 2023Over-Parameterization Exponentially Slows Down Gradient Descent for Learning a Single Neuron20 Feb 2023
- 2022Likelihood Adjusted Semidefinite Programs for Clustering Heterogeneous Data29 Sep 2022
- 2022Mean Estimation in High-Dimensional Binary Markov Gaussian Mixture Models6 Jun 2022
- 2022Optimal Estimation and Computational Limit of Low-rank Gaussian Mixtures22 Jan 2022
- 2021Clustering a Mixture of Gaussians with Unknown Covariance4 Oct 2021
- 2021Sharp global convergence guarantees for iterative nonconvex optimization: A Gaussian process perspective20 Sep 2021
- 2021The EM Algorithm is Adaptively-Optimal for Unbalanced Symmetric Gaussian Mixtures29 Mar 2021
- 2023Over-Parameterization Exponentially Slows Down Gradient Descent for Learning a Single Neuroncited by 25
- 2023On the Convergence of Coordinate Ascent Variational Inferencecited by 18
- 2023How Over-Parameterization Slows Down Gradient Descent in Matrix Sensing: The Curses of Symmetry and Initializationcited by 18
- 2024How much is a noisy image worth? Data Scaling Laws for Ambient Diffusioncited by 18
- 2021Clustering a Mixture of Gaussians with Unknown Covariancecited by 14
- 2021Sharp global convergence guarantees for iterative nonconvex optimization: A Gaussian process perspectivecited by 10
- 2021The EM Algorithm is Adaptively-Optimal for Unbalanced Symmetric Gaussian Mixturescited by 6
- 2022Optimal Estimation and Computational Limit of Low-rank Gaussian Mixturescited by 6
- 2024Toward Global Convergence of Gradient EM for Over-Parameterized Gaussian Mixture Modelscited by 6
- 2023Can semi-supervised learning use all the data effectively? A lower bound perspectivecited by 5
- 2024Matrix Denoising with Doubly Heteroscedastic Noise: Fundamental Limits and Optimal Spectral Methodscited by 5
- 2024Semi-Supervised Sparse Gaussian Classification: Provable Benefits of Unlabeled Datacited by 3
12 most cited of 24 All 24 (JSON)
- 2026Iterative Atom Refinement: A Monotonicity Principle for Dictionary Learning20 Sep 2026
- 2026One-Bit Clustering for Two Component Sub-Gaussian Mixture Models20 Jun 2026
- 2025Nonparametric MLE for Gaussian Location Mixtures: Certified Computation and Generic Behavior26 Mar 2025
- 2024How much is a noisy image worth? Data Scaling Laws for Ambient Diffusion5 Nov 2024
- 2024In-context Learning for Mixture of Linear Regressions: Existence, Generalization and Training Dynamics18 Oct 2024
- 2024Semi-Supervised Sparse Gaussian Classification: Provable Benefits of Unlabeled Data5 Sep 2024
- 2024Factor Adjusted Spectral Clustering for Mixture Models22 Aug 2024
- 2024Toward Global Convergence of Gradient EM for Over-Parameterized Gaussian Mixture Models29 Jun 2024
- 2024Adaptive Mean Estimation in the Hidden Markov sub-Gaussian Mixture Model18 Jun 2024
- 2024Matrix Denoising with Doubly Heteroscedastic Noise: Fundamental Limits and Optimal Spectral Methods22 May 2024
- 2023Can semi-supervised learning use all the data effectively? A lower bound perspective30 Nov 2023
- 2023Joint Problems in Learning Multiple Dynamical Systems3 Nov 2023
12 newest of 24 All 24 (JSON)