Oct, 2023

神经扩散模型

TL;DR传播模型是生成任务中表现出色的模型,然而大部分传播模型仅允许对数据分布进行线性变换,相比之下,更广泛的转换可能有助于更高效地训练生成分布并消除真实负对数似然和变分近似之间的差距。在本文中,我们介绍了神经传播模型(NDMs),这是传统传播模型的泛化,可以定义和学习数据的时间相关的非线性转换。我们展示了如何在无需模拟的情况下使用变分上界来优化NDMs。此外,我们导出了NDMs的时间连续形式,可以使用现成的数值ODE和SDE求解器进行快速可靠的推断。最后,我们通过在标准图像生成基准测试上的实验,包括CIFAR-10、ImageNet的降采样版本和CelebA-HQ,展示了可学习转换的NDMs的实用性。NDMs在似然度和生成高质量样本方面优于传统传播模型。