Baum-Welch algorithm on directed acyclic graph for mixtures with latent Bayesian networks

Research output: Contribution to journalArticle

Abstract

We consider a mixture model with latent Bayesian network (MLBN) for a set of random vectors X(t), X(t) ∈ ℝdt , t = 1,…, T. Each X(t) is associated with a latent state st, given which X(t) is conditionally independent from other variables. The joint distribution of the states is governed by a Bayes net. Although specific types of MLBN have been used in diverse areas such as biomedical research and image analysis, the exact expectation–maximization (EM) algorithm for estimating the models can involve visiting all the combinations of states, yielding exponential complexity in the network size. A prominent exception is the Baum–Welch algorithm for the hidden Markov model, where the underlying graph topology is a chain. We hereby develop a new Baum–Welch algorithm on directed acyclic graph (BW-DAG) for the general MLBN and prove that it is an exact EM algorithm. BW-DAG provides insight on the achievable complexity of EM. For a tree graph, the complexity of BW-DAG is much lower than that of the brute-force EM.

Original languageEnglish (US)
Pages (from-to)303-314
Number of pages12
JournalStat
Volume6
Issue number1
DOIs
Publication statusPublished - Jan 1 2017

    Fingerprint

All Science Journal Classification (ASJC) codes

  • Statistics and Probability
  • Statistics, Probability and Uncertainty

Cite this