Maximum likelihood estimation of Gaussian mixture models using stochastic search

详细信息	查看全文 | 推荐本文 |

• 作者：Ç ; a臒lar Ar谋^a ; ^{cari@ee.bilkent.edu.tr}  ; [Author Vitae] ; Selim Aksoy^b ; ^{saksoy@cs.bilkent.edu.tr}  ; [Author Vitae] ; Orhan Ar谋kan^a ; ^{oarikan@ee.bilkent.edu.tr}  ; [Author Vitae]
• 关键词：Gaussian mixture models ; Maximum likelihood estimation ; Expectation&ndash ; maximization ; Covariance parametrization ; Identifiability ; Stochastic search ; Particle swarm optimization
• 刊名：Pattern Recognition
• 出版年：2012
• 期刊代码：100_00313203
• 类别：cp
• 出版时间：July, 2012
• 卷：45
• 期：7
• 页码：2804-2816
• 文件大小：701 K

摘要

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.