Learning Gaussian mixture models with entropy-based criteria.
In this paper, we address the problem of estimating the parameters of Gaussian mixture models. Although the expectation-maximization (EM) algorithm yields the maximum-likelihood (ML) solution, its sensitivity to the selection of the starting parameters is well-known and it may converge to the boundary of the parameter space. Furthermore, the resulting mixture depends on the number of selected components, but the optimal number of kernels may be unknown beforehand. We introduce the use of the entropy of the probability density function (pdf) associated to each kernel to measure the quality of a given mixture model with a fixed number of kernels. We propose two methods to approximate the entropy of each kernel and a modification of the classical EM algorithm in order to find the optimum number of components of the mixture. Moreover, we use two stopping criteria: a novel global mixture entropy-based criterion called Gaussianity deficiency (GD) and a minimum description length (MDL) principle-based one. Our algorithm, called entropy-based EM (EBEM), starts with a unique kernel and performs only splitting by selecting the worst kernel attending to GD. We have successfully tested it in probability density estimation, pattern classification, and color image segmentation. Experimental results improve the ones of other state-of-the-art model order selection methods.
Penalver Benavent
Antonio
A
Departamento de Estadística, Matemáticas e Informática, Universidad Miguel Hernández, Elche 03202, Spain. a.penalver@umh.es
Escolano Ruiz
Francisco
F
Saez
Juan Manuel
JM
D000465:Algorithms; D001185:Artificial Intelligence; D003627:Data Interpretation, Statistical; D019277:Entropy; D015233:Models, Statistical; D016571:Neural Networks, Computer; D016011:Normal Distribution