Statistical Convergence of the EM Algorithm on Gaussian Mixture Models
Statistics Theory
2018-10-10 v1 Statistics Theory
Abstract
We study the convergence behavior of the Expectation Maximization (EM) algorithm on Gaussian mixture models with an arbitrary number of mixture components and mixing weights. We show that as long as the means of the components are separated by at least , where is the number of components and is the dimension, the EM algorithm converges locally to the global optimum of the log-likelihood. Further, we show that the convergence rate is linear and characterize the size of the basin of attraction to the global optimum.
Cite
@article{arxiv.1810.04090,
title = {Statistical Convergence of the EM Algorithm on Gaussian Mixture Models},
author = {Ruofei Zhao and Yuanzhi Li and Yuekai Sun},
journal= {arXiv preprint arXiv:1810.04090},
year = {2018}
}