Persistently Trained, Diffusion-assisted Energy-based Models

Xinwei Zhang,Zhiqiang Tan,Zhijian Ou
DOI: https://doi.org/10.1002/sta4.625
2023-01-01
Stat
Abstract:Maximum likelihood (ML) learning for energy-based models (EBMs) is challenging, partly due to nonconvergence of Markov chain Monte Carlo. Several variations of ML learning have been proposed, but existing methods all fail to achieve both posttraining image generation and proper density estimation. We propose to introduce diffusion data and learn a joint EBM, called diffusion-assisted EBMs, through persistent training (i.e. using persistent contrastive divergence) with an enhanced sampling algorithm to properly sample from complex, multimodal distributions. We present results from a 2D illustrative experiment and image experiments and demonstrate that for the first time for image data, persistently trained EBMs can simultaneously achieve long-run stability, post-training image generation and superior out-of-distribution detection.
What problem does this paper attempt to address?