SMEM Algorithm for Mixture Models

Naonori Ueda, Ryohei Nakano, Zoubin Ghahramani, Geoffrey E. Hinton

Neural Computation · 2000 · 411 citations · 13 references

Concepts

TL;DR

Local maxima in finite mixture models arise when one region of the space contains too many components while another contains too few. The authors propose a split‑and‑merge expectation‑maximization (SMEM) algorithm to overcome this local‑maxima problem in parameter estimation of finite mixture models. SMEM repeatedly applies simultaneous split‑and‑merge operations guided by a new criterion that efficiently selects candidate components. Experiments on Gaussian mixtures, mixtures of factor analyzers, and real datasets show that SMEM improves training and held‑out likelihoods, and it also enhances performance in image compression and pattern recognition tasks.

Abstract

We present a split-and-merge expectation-maximization (SMEM) algorithm to overcome the local maxima problem in parameter estimation of finite mixture models. In the case of mixture models, local maxima often involve having too many components of a mixture model in one part of the space and too few in another, widely separated part of the space. To escape from such configurations, we repeatedly perform simultaneous split-and-merge operations using a new criterion for efficiently selecting the split-and-merge candidates. We apply the proposed algorithm to the training of gaussian mixtures and mixtures of factor analyzers using synthetic and real data and show the effectiveness of using the split-and-merge operations to improve the likelihood of both the training data and of held-out test data. We also show the practical usefulness of the proposed algorithm by applying it to image compression and pattern recognition problems.

References

13