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

DuoDiff:使用双骨干方法加速扩散模型

DuoDiff: Accelerating Diffusion Models with a Dual-Backbone Approach

University of Amsterdam(阿姆斯特丹大学)

Daniel Gallo Fernández, Răzvan-Andrei Matişan, Alejandro Monroy Muñoz, Ana-Maria Vasilcoiu, Janusz Partyka, Tin Hadži Veljković, Metod Jazbec

AI总结:

该研究发现自适应扩散模型采样中的“相变”,提出 DuoDiff 双骨干方法,在采样前期和后期分别使用浅、深去噪网络,从而同时提升推理速度与生成质量。

摘要:

扩散模型在图像生成方面取得了前所未有的性能,但由于其迭代采样过程,它们面临推理缓慢的问题。为解决这一问题,近期研究提出了提前退出方法,即根据每个采样步骤(估计)的难度,自适应地调整去噪网络的深度。本文发现,当前自适应扩散模型的采样过程中存在一个有趣的“相变”:去噪网络在初始采样步骤中始终提前退出,随后突然切换为使用完整网络。基于此,我们提出通过在初始采样步骤采用较浅的去噪网络、在后续步骤采用较深网络来加速生成。实验表明,我们的双骨干方法 DuoDiff 在推理速度和生成质量上均优于现有提前退出扩散方法。重要的是,DuoDiff 易于实现,并且可与现有扩散加速方法互补。

英文摘要:

Diffusion models have achieved unprecedented performance in image generation, yet they suffer from slow inference due to their iterative sampling process. To address this, early-exiting has recently been proposed, where the depth of the denoising network is made adaptive based on the (estimated) difficulty of each sampling step. Here, we discover an interesting "phase transition" in the sampling process of current adaptive diffusion models: the denoising network consistently exits early during the initial sampling steps, until it suddenly switches to utilizing the full network. Based on this, we propose accelerating generation by employing a shallower denoising network in the initial sampling steps and a deeper network in the later steps. We demonstrate empirically that our dual-backbone approach, DuoDiff, outperforms existing early-exit diffusion methods in both inference speed and generation quality. Importantly, DuoDiff is easy to implement and complementary to existing approaches for accelerating diffusion.

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

↑