Maximum likelihood estimation of Gaussian mixture models using stochastic search

作者:

Highlights:

摘要

Gaussian mixture models (GMM), commonly used in pattern recognition and machine learning, provide a flexible probabilistic model for the data. The conventional expectation–maximization (EM) algorithm for the maximum likelihood estimation of the parameters of GMMs is very sensitive to initialization and easily gets trapped in local maxima. Stochastic search algorithms have been popular alternatives for global optimization but their uses for GMM estimation have been limited to constrained models using identity or diagonal covariance matrices. Our major contributions in this paper are twofold. First, we present a novel parametrization for arbitrary covariance matrices that allow independent updating of individual parameters while retaining validity of the resultant matrices. Second, we propose an effective parameter matching technique to mitigate the issues related with the existence of multiple candidate solutions that are equivalent under permutations of the GMM components. Experiments on synthetic and real data sets show that the proposed framework has a robust performance and achieves significantly higher likelihood values than the EM algorithm.

论文关键词:Gaussian mixture models,Maximum likelihood estimation,Expectation–maximization,Covariance parametrization,Identifiability,Stochastic search,Particle swarm optimization

论文评审过程:Received 28 March 2011, Revised 16 December 2011, Accepted 30 December 2011, Available online 11 January 2012.

论文官网地址:https://doi.org/10.1016/j.patcog.2011.12.023