用双向边界稳定能量模型训练,提升生成质量
Exploring bidirectional bounds for minimax-training of Energy-based models
- 训练时同时最大化下界、最小化上界,避免传统方法的不稳定性
- 提出四种对数似然的上下界,涵盖奇异值、互信息等新思路
- 实测表明该方法显著提升密度估计与样本生成效果,适合生成建模研究者
能量模型(EBM)以优雅框架估计非归一化密度,但训练困难。近期工作将EBM与生成对抗网络关联,指出可通过变分下界进行极小化博弈训练。为避免最小化下界带来的不稳定性,本文提出使用双向边界:训练时同时最大化下界、最小化上界。本文从不同视角推导了四种对数似然的上下界:基于生成器雅可比奇异值的下界、基于互信息的下界;以及类似梯度惩罚的上界、基于扩散过程的上界。所有边界均提供可计算算法。通过对比分析不同边界优劣,实验表明双向边界能有效稳定EBM训练,在密度估计和样本生成上均取得高质量结果。
原文摘要 · Abstract (English)
Energy-based models (EBMs) estimate unnormalized densities in an elegant framework, but they are generally difficult to train. Recent work has linked EBMs to generative adversarial networks, by noting that they can be trained through a minimax game using a variational lower bound. To avoid the instabilities caused by minimizing a lower bound, we propose to instead work with bidirectional bounds, meaning that we maximize a lower bound and minimize an upper bound when training the EBM. We investigate four different bounds on the log-likelihood derived from different perspectives. We derive lower bounds based on the singular values of the generator Jacobian and on mutual information. To upper bound the negative log-likelihood, we consider a gradient penalty-like bound, as well as one based on diffusion processes. In all cases, we provide algorithms for evaluating the bounds. We compare the different bounds to investigate, the pros and cons of the different approaches. Finally, we demonstrate that the use of bidirectional bounds stabilizes EBM training and yields high-quality density estimation and sample generation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。