Title
Joint Training of Variational Auto-Encoder and Latent Energy-Based Model
Abstract
This paper proposes a joint training method to learn both the variational auto-encoder (VAE) and the latent energy-based model (EBM). The joint training of VAE and latent EBM are based on an objective function that consists of three Kullback-Leibler divergences between three joint distributions on the latent vector and the image, and the objective function is of an elegant symmetric and anti-symmetric form of divergence triangle that seamlessly integrates variational and adversarial learning. In this joint training scheme, the latent EBM serves as a critic of the generator model, while the generator model and the inference model in VAE serve as the approximate synthesis sampler and inference sampler of the latent EBM. Our experiments show that the joint training greatly improves the synthesis quality of the VAE. It also enables learning of an energy function that is capable of detecting out of sample examples for anomaly detection.
Year
DOI
Venue
2020
10.1109/CVPR42600.2020.00800
CVPR
DocType
Citations 
PageRank 
Conference
1
0.35
References 
Authors
24
6
Name
Order
Citations
PageRank
Tian Han1236.21
Erik Nijkamp232615.58
Linqi Zhou310.69
Bo Pang412.38
Song-Chun Zhu56580741.75
Ying Nian Wu61652267.72