arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2306.01654cs.LGcs.CVstat.ML

关于扩散建模中闭式IPM-GAN判别器引导的洞见

Insights into Closed-form IPM-GAN Discriminator Guidance for Diffusion Modeling

Microsoft Research #9 VIGYAN(微软研究院)

Aadithya Srikanth, Siddarth Asokan, Nishanth Shetty, Chandra Sekhar Seelamantula

AI总结:

本文提出理论框架分析GAN判别器对朗之万采样的影响,证明IPM-GAN优化是平滑分数匹配,统一两类方法,并验证闭式核判别器引导可提升多种扩散模型的生成质量。

摘要:

扩散模型是一种最先进的生成建模框架,它通过朗之万采样将噪声转化为图像,该过程由分数(即数据分布对数的梯度)引导。近期研究经实验证实,使用分类器网络(通常是在生成对抗网络(GAN)设置下训练的判别器)进行引导时,生成质量可得到提升。本文提出了一个理论框架,用于分析GAN判别器对基于朗之万的采样的影响,并证明IPM-GAN优化可被视为一种平滑分数匹配方法,其中数据分布与生成器分布的分数会与IPM相关的核函数进行卷积。所提方法实现了基于分数的训练与IPM-GAN优化的统一。基于这些洞见,我们证明基于核的闭式判别器引导应用于基线扩散模型之上时,可带来CLIP-FID和KID指标层面的提升。我们在去噪扩散隐式模型(DDIM)和潜扩散模型(LDM)设置下,在多个标准数据集上验证了这些结果。我们还表明,所提方法可与现有的加速扩散技术结合,以改进潜空间图像生成。

英文摘要:

Diffusion models are a state-of-the-art generative modeling framework that transform noise to images via Langevin sampling, guided by the score, which is the gradient of the logarithm of the data distribution. Recent works have shown empirically that the generation quality can be improved when guided by classifier network, which is typically the discriminator trained in a generative adversarial network (GAN) setting. In this paper, we propose a theoretical framework to analyze the effect of the GAN discriminator on Langevin-based sampling, and show that the IPM-GAN optimization can be seen as one of smoothed score-matching, wherein the scores of the data and the generator distributions are convolved with the kernel function associated with the IPM. The proposed approach serves to unify score-based training and optimization of IPM-GANs. Based on these insights, we demonstrate that closed-form kernel-based discriminator guidance, results in improvements (in terms of CLIP-FID and KID metrics) when applied atop baseline diffusion models. We demonstrate these results on the denoising diffusion implicit model (DDIM) and latent diffusion model (LDM) settings on various standard datasets. We also show that the proposed approach can be combined with existing accelerated-diffusion techniques to improve latent-space image generation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑