Improved Contrastive Divergence Training of Energy Based Models
Authors: Yilun Du, Shuang Li, Joshua B. Tenenbaum, Igor Mordatch
Keywords: Contrastive Divergence, Energy Based Modeling
Abstract: We propose several different techniques to improve contrastive divergence training of energy-based models (EBMs). We first show that a gradient term neglected in the popular contrastive divergence formulation is both tractable to estimate and is important to avoid training instabilities in previous models. We further highlight how data augmentation, multi-scale processing, and reservoir sampling can be used to improve model robustness and generation quality. Thirdly, we empirically evaluate stability of model architectures and show improved performance on a host of benchmarks and use cases, such as image generation, OOD detection, and compositional generation.