Nov, 2023

基于能量的校准变分自编码器及其无测试时间消耗特性

TL;DR本文提出了一种新颖的能量校准生成模型,该模型利用条件 EBM 来增强变分自动编码器(VAEs)的性能,针对 VAEs 在生成方向上训练不足而导致生成结果模糊的问题,以及 EBMs 需要昂贵的马尔可夫链蒙特卡罗(MCMC)采样的问题,我们引入了条件 EBM 用于训练时校准生成方向,而不需要在测试时进行采样,这样的方法使得生成模型能够根据数据和校准样本进行自适应加权的训练,从而提高了效率和效果,而不需要在推理阶段进行 MCMC 采样,我们还展示了该方法可以扩展到校准归一化流和变分后验,并提出将该方法应用于基于神经传输先验和零射图像恢复,通过广泛的实验在图像生成和零射图像恢复等各种应用中证明了该方法的有效性,我们的方法在单步非对抗式生成方面展现了最先进的性能。