Calculating Posterior Distributions and Modal Estimates in Markov Mixture Models

bayesiangibbs-samplermarkov-switchingmcmcdata-augmentationmixture-modelem-algorithm

Summary

Develops a full Bayesian Gibbs sampling approach for Markov mixture models (hidden Markov models) — finite mixture distributions where the component populations are selected by an unobserved Markov chain. The central innovation is drawing the entire latent state sequence Sn=(s1,,sn)S_n=(s_1,\ldots,s_n) from its joint distribution in a single Gibbs block via a forward filter–backward sampling algorithm, rather than from nn individual full conditionals. This reduces state-related Gibbs blocks from nn to 1 and dramatically improves convergence. The paper also derives stochastic expectation-maximization (EM) variants (SEM and MCEM) using the same joint-sampling routine for computing posterior modes.

Key Claims

  1. Joint state sampling (main contribution, eq. 5–8): The joint p(SnYn,θ)p(S_n|Y_n,\theta) factorizes as p(snYn,θ)t=1n1p(stYn,St+1,θ)p(s_n|Y_n,\theta)\prod_{t=1}^{n-1}p(s_t|Y_n,S^{t+1},\theta); each factor equals (row tt of filter FF) ×\times (column st+1s_{t+1} of PP), normalized — requiring only forward-filter output and one transition probability per step.
  2. Forward filter: n×mn\times m matrix FF where row t(Ft1P)dtt\propto (F_{t-1}P)\odot d_t (dtd_t = row vector of component observation densities f(ytYt1,θst)f(y_t|Y_{t-1},\theta_{s_t}) across all mm states); initialized from the chain's stationary distribution; cost O(nm2)O(nm^2).
  3. Backward sampler: Draw sns_n from final row of FF; for t=n1,,1t=n-1,\ldots,1 draw st[Ft]stPst,st+1s_t\propto [F_t]_{s_t}\cdot P_{s_t,s_{t+1}} (eq. 7); cost O(nm)O(nm). Total: one block, replacing nn blocks.
  4. Convergence improvement: Albert-Chib (1993) and McCulloch-Tsay (1994) used single-move state sampling (nn blocks, high autocorrelation). Joint sampling requires truly one additional block for all states.
  5. Dirichlet conjugacy for PP: With prior piD(αi1,,αim)p_i\sim\mathcal{D}(\alpha_{i1},\ldots,\alpha_{im}), the posterior conditional is piSnD(αi1+ni1,,αim+nim)p_i|S_n\sim\mathcal{D}(\alpha_{i1}+n_{i1},\ldots,\alpha_{im}+n_{im}) (eq. 10), where nijn_{ij} counts iji\to j transitions in the sampled SnS_n.
  6. SEM (Celeux-Diebolt 1985): one draw of SnS_n per iteration; iterates form an aperiodic irreducible Markov chain converging to the posterior mode — a stochastic alternative to standard EM.
  7. MCEM (Monte Carlo EM; Wei-Tanner 1990): NN draws of SnS_n per iteration to approximate the Q-function; averages over Sn,jS_{n,j}, j=1,,Nj=1,\ldots,N; deterministic convergence to mode as NN\to\infty. Recommended strategy: SEM burn-in, MCEM for final iterations.
  8. Three applications: (i) Poisson fetal movement (n=240n=240, 2/3 states: λ^10.26\hat\lambda_1\approx0.26, λ^22.29\hat\lambda_2\approx2.29, p^110.98\hat p_{11}\approx0.98 — high persistence); (ii) fourth-order autoregressive AR(4) US gross national product (GNP) 1951:2–1984:4 (n=134n=134, 4 states, state-dependent intercepts, common AR and variance: AR(2)+4 states appears parsimonious); (iii) bivariate Gaussian mixture (n=300n=300, 3 components: Bayes more accurate than SEM; label-switching observed between states 2 and 3).

Concepts Introduced or Extended

Entities Mentioned

Quotes

"This new result is extremely significant. Instead of nn additional blocks in the Gibbs sampler (the number required if each state is sampled from its full conditional distribution), truly one additional block is required. This dramatically improves the convergence of the Markov chain induced by the Gibbs sampling algorithm."

My Take

Chib (1996) is the discrete-state counterpart of the Carter-Kohn (1994) simulation smoother for Gaussian state-space models: both replace nn single-move draws with a joint draw via a forward-backward pass. The algorithm became standard for Bayesian Markov-switching estimation and the forward-filter / backward-sampler structure reappears in all subsequent Markov-switching vector-autoregression (MS-VAR) implementations (Kim-Nelson 1999; Frühwirth-Schnatter 2006; Sims-Zha 2006). The SEM/MCEM discussion is valuable historically but now largely superseded — full Bayes with the Chib (1995) marginal likelihood provides cleaner model selection. The label-switching problem (noted in the bivariate Gaussian example) remains an open challenge for mixture posteriors.