Random swap EM algorithm for finite mixture models in image segmentation

  • Authors:
  • Qinpei Zhao;Ville Hautamäki;Ismo Kärkkäinen;Pasi Fränti

  • Affiliations:
  • Department of Computer Science, University of Joensuu, Finland;Department of Computer Science, University of Joensuu, Finland;Institute for Infocomm Research, A*STAR, Singapore;Department of Computer Science, University of Joensuu, Finland and School of Computer Science, Fudan University, China

  • Venue:
  • ICIP'09 Proceedings of the 16th IEEE international conference on Image processing
  • Year:
  • 2009

Quantified Score

Hi-index 0.00

Visualization

Abstract

The Expectation-Maximization (EM) algorithm is a popular tool in estimating model parameters, especially mixture models. As the EM algorithm is a hill-climbing approach, problems such as local maxima, plateau and ridges may appear. In the case of mixture models, these problems involve the initialization of the algorithm and the structure of the data set. We propose a random swap EM algorithm (RSEM) to overcome these problems in Gaussian mixture models. Random swaps are repeatedly performed in our method, which can break the configuration of the local maxima and other problems. Compared to the strategies in other methods, the proposed algorithm has relative improvements on log-likelihood value in most cases and less variance than other algorithms. We also apply RSEM to the image segmentation problem.