Multifactor expectation maximization for factor graphs

  • Authors:
  • Jason T. Rolfe;Matthew Cook

  • Affiliations:
  • Computation and Neural Systems, California Institute of Technology and Institute of Neuroinformatics, University of Zurich and ETH Zurich;Institute of Neuroinformatics, University of Zurich and ETH Zurich

  • Venue:
  • ICANN'10 Proceedings of the 20th international conference on Artificial neural networks: Part III
  • Year:
  • 2010

Quantified Score

Hi-index 0.01

Visualization

Abstract

Factor graphs allow large probability distributions to be stored efficiently and facilitate fast computation of marginal probabilities, but the difficulty of training them has limited their use. Given a large set of data points, the training process should yield factors for which the observed data has a high likelihood. We present a factor graph learning algorithm which on each iteration merges adjacent factors, performs expectation maximization on the resulting modified factor graph, and then splits the joined factors using non-negative matrix factorization. We show that this multifactor expectation maximization algorithm converges to the global maximum of the likelihood for difficult learning problems much faster and more reliably than traditional expectation maximization.