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

减少去噪扩散模型蒸馏中的空间拟合误差

Reducing Spatial Fitting Error in Distillation of Denoising Diffusion Models

Zhejiang University(浙江大学);Alibaba Group(阿里巴巴集团)

Shengzhe Zhou, Zejian Lee, Shengyuan Zhang, Lefan Hou, Changyuan Yang, Guang Yang, Zhiyuan Yang, Lingyun Sun

AI总结:

针对扩散模型蒸馏导致生成质量下降的问题,提出SFERD模型,利用教师注意力引导和语义梯度预测器减少空间拟合误差,实现一步生成高质量图像,在CIFAR-10和ImageNet上取得优异FID。

摘要:

去噪扩散模型在图像生成方面展现出了卓越的能力。然而,生成高质量样本需要大量的迭代次数。针对扩散模型的知识蒸馏是一种有效的方法,通过缩短采样过程来解决这一限制,但会导致生成质量的下降。基于我们使用偏差-方差分解和实验观察的分析,我们将这种退化归因于教师模型和学生模型训练中出现的空间拟合误差。因此,我们提出了空间拟合误差减少蒸馏模型(SFERD)。SFERD利用来自教师模型的注意力引导和设计的语义梯度预测器来减少学生的拟合误差。实证上,我们提出的模型在少量函数评估下促进了高质量样本的生成。我们在CIFAR-10上实现了5.31的FID,在ImageNet 64×64上仅用一步就实现了9.39的FID,优于现有的扩散方法。我们的研究通过强调模型固有的去噪能力,为扩散蒸馏提供了新的视角。项目链接:https://github.com/Sainzerjj/SFERD。

英文摘要:

Denoising Diffusion models have exhibited remarkable capabilities in image generation. However, generating high-quality samples requires a large number of iterations. Knowledge distillation for diffusion models is an effective method to address this limitation with a shortened sampling process but causes degraded generative quality. Based on our analysis with bias-variance decomposition and experimental observations, we attribute the degradation to the spatial fitting error occurring in the training of both the teacher and student model. Accordingly, we propose $\textbf{S}$patial $\textbf{F}$itting-$\textbf{E}$rror $\textbf{R}$eduction $\textbf{D}$istillation model ($\textbf{SFERD}$). SFERD utilizes attention guidance from the teacher model and a designed semantic gradient predictor to reduce the student's fitting error. Empirically, our proposed model facilitates high-quality sample generation in a few function evaluations. We achieve an FID of 5.31 on CIFAR-10 and 9.39 on ImageNet 64$\times$64 with only one step, outperforming existing diffusion methods. Our study provides a new perspective on diffusion distillation by highlighting the intrinsic denoising ability of models. Project link: \url{https://github.com/Sainzerjj/SFERD}.

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

↑