Learning Multimodal Energy-Based Model with Multimodal Variational Auto-Encoder via MCMC Revision

TL;DR

This paper introduces a novel learning framework for multimodal energy-based models that combines MCMC refinement with variational auto-encoders, improving the quality and coherence of multimodal data synthesis.

Contribution

It proposes an integrated training approach that interweaves MLE updates with MCMC refinements for multimodal EBMs and VAEs, enhancing sampling effectiveness.

Findings

Achieves superior multimodal synthesis quality and coherence.

Demonstrates effective scalability and robustness through extensive experiments.

Provides ablation studies validating the proposed framework's components.

Abstract

Energy-based models (EBMs) are a flexible class of deep generative models and are well-suited to capture complex dependencies in multimodal data. However, learning multimodal EBM by maximum likelihood requires Markov Chain Monte Carlo (MCMC) sampling in the joint data space, where noise-initialized Langevin dynamics often mixes poorly and fails to discover coherent inter-modal relationships. Multimodal VAEs have made progress in capturing such inter-modal dependencies by introducing a shared latent generator and a joint inference model. However, both the shared latent generator and joint inference model are parameterized as unimodal Gaussian (or Laplace), which severely limits their ability to approximate the complex structure induced by multimodal data. In this work, we study the learning problem of the multimodal EBM, shared latent generator, and joint inference model. We present a learning framework that effectively interweaves their MLE updates with corresponding MCMC refinements in both the data and latent spaces. Specifically, the generator is learned to produce coherent multimodal samples that serve as strong initial states for EBM sampling, while the inference model is learned to provide informative latent initializations for generator posterior sampling. Together, these two models serve as complementary models that enable effective EBM sampling and learning, yielding realistic and coherent multimodal EBM samples. Extensive experiments demonstrate superior performance for multimodal synthesis quality and coherence compared to various baselines. We conduct various analyses and ablation studies to validate the effectiveness and scalability of the proposed multimodal framework.