计算机工程与应用 ›› 2012, Vol. 48 ›› Issue (34): 28-32.

• 博士论坛 • 上一篇    下一篇

基于分裂EM算法的GMM参数估计

钟金琴1,3,辜丽川2,檀结庆3,李莹莹3   

  1. 1.安徽大学 电子与信息系,合肥 230031
    2.安徽农业大学 计算机信息学院,合肥 230036
    3.合肥工业大学 计算机与信息学院,合肥 230009
  • 出版日期:2012-12-01 发布日期:2012-11-30

Estimating parameters of GMM based on split EM

ZHONG Jinqin1,3, GU Lichuan2, TAN Jieqing3, LI Yingying3   

  1. 1.Department of Electronic and Information, Anhui University, Hefei 230031, China
    2.School of Information and Computer, Anhui Agriculture University, Hefei 230036, China
    3.School of Computer and Information, Hefei University of Technology, Hefei 230009, China
  • Online:2012-12-01 Published:2012-11-30

摘要: 期望最大化(Expectation Maximization,EM)算法是一种求参数极大似然估计的迭代算法,常用来估计混合密度分布模型的参数。EM算法的主要问题是参数初始化依赖于先验知识且在迭代过程中容易收敛到局部极大值。提出一种新的基于分裂EM算法的GMM参数估计算法,该方法从一个确定的单高斯分布开始,在EM优化过程中逐渐分裂并估计混合分布的参数,解决了参数迭代收敛到局部极值问题。大量的实验表明,与现有的其他参数估计算法相比,算法具有较好的运算效率和估算准确性。

关键词: 高斯混合模型, 期望最大化, 参数估计, 模式分类

Abstract: The expectation maximization algorithm has been classically used to find the maximum likelihood estimates of parameters in mixture probabilistic models. Problems of the EM algorithm are that parameters initialization depends on some prior knowledge, and it is easy to converge to a local maximum in the iteration process. In this paper, a new method of estimating the parameter of GMM based on split EM is proposed, it starts from a single mixture component, sequentially split and estimates the parameter of the mixture components during expectation maximization steps. Extensive experiments show the advantages and efficiency of the proposed method.

Key words: Gaussian Mixture Model(GMM), Expectation Maximization(EM), parameters estimation, pattern classification