arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2302.09057cs.LGcs.AIcs.CVcs.ITmath.IT
收藏

一致扩散模型:通过学习一致性来减轻采样漂移

Consistent Diffusion Models: Mitigating Sampling Drift by Learning to be Consistent

University of Texas at Austin(德克萨斯大学奥斯汀分校);University of California, Berkeley(加州大学伯克利分校);Massachusetts Institute of Technology(麻省理工学院)

Giannis Daras, Yuval Dagan, Alexandros G. Dimakis, Constantinos Daskalakis

摘要:

不完美的分数匹配导致扩散模型的训练分布和采样分布之间的偏移。由于生成过程的递归性质,先前步骤中的错误会产生偏离训练分布的采样迭代。然而,通过去噪得分匹配(DSM)的标准训练目标仅被设计为在非漂移数据上进行优化。为了在漂移数据上进行训练,我们建议强制执行\emph{consistency}属性,该属性声明模型对自己生成的数据的预测在时间上是一致的。从理论上讲,我们表明,如果在一些非漂移点上(通过DSM)完美地学习了分数,并且如果到处都强制执行一致性属性,那么在任何地方都可以准确地学习分数。从经验上讲,我们表明,我们的新的训练目标产生了最先进的结果,在CIFAR-10和AFHQ和FFHQ的基线改进的条件和无条件生成。我们开源我们的代码和模型:https://github.com/giannisdaras/cdm

英文摘要:

Imperfect score-matching leads to a shift between the training and the sampling distribution of diffusion models. Due to the recursive nature of the generation process, errors in previous steps yield sampling iterates that drift away from the training distribution. Yet, the standard training objective via Denoising Score Matching (DSM) is only designed to optimize over non-drifted data. To train on drifted data, we propose to enforce a \emph{consistency} property which states that predictions of the model on its own generated data are consistent across time. Theoretically, we show that if the score is learned perfectly on some non-drifted points (via DSM) and if the consistency property is enforced everywhere, then the score is learned accurately everywhere. Empirically we show that our novel training objective yields state-of-the-art results for conditional and unconditional generation in CIFAR-10 and baseline improvements in AFHQ and FFHQ. We open-source our code and models: https://github.com/giannisdaras/cdm

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

↑