Poster
Improving Explorability in Variational Inference with Annealed Variational Objectives
Chin-Wei Huang · Shawn Tan · Alexandre Lacoste · Aaron Courville

Wed Dec 5th 10:45 AM -- 12:45 PM @ Room 210 #23

Despite the advances in the representational capacity of approximate distributions for variational inference, the optimization process can still limit the density that is ultimately learned. We demonstrate the drawbacks of biasing the true posterior to be unimodal, and introduce Annealed Variational Objectives (AVO) into the training of hierarchical variational methods. Inspired by Annealed Importance Sampling, the proposed method facilitates learning by incorporating energy tempering into the optimization objective. In our experiments, we demonstrate our method's robustness to deterministic warm up, and the benefits of encouraging exploration in the latent space.

Author Information

Chin-Wei Huang (MILA)
Shawn Tan (Mila)
Alexandre Lacoste (Element AI)
Aaron Courville (U. Montreal)

More from the Same Authors