arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.05967cs.CVcs.GRcs.LG
收藏

将扩散模型蒸馏为条件GAN

Distilling Diffusion Models into Conditional GANs

Pohang University of Science and Technology(浦项科技大学);Adobe Research(奥多比研究院);Seoul National University(首尔大学);Carnegie Mellon University(卡内基梅隆大学)

Minguk Kang, Richard Zhang, Connelly Barnes, Sylvain Paris, Suha Kwak, Jaesik Park, Eli Shechtman, Jun-Yan Zhu, Taesung Park

AI总结:

我们提出将扩散模型蒸馏为单步条件GAN的方法,通过E-LatentLPIPS损失和多尺度判别器,在零样本COCO上超越DMD等模型,加速推理并保持图像质量。

摘要:

我们提出了一种方法,将复杂的多步扩散模型蒸馏为单步条件GAN学生模型,从而大幅加速推理,同时保持图像质量。我们的方法将扩散蒸馏视为成对的图像到图像转换任务,利用扩散模型ODE轨迹的噪声-图像对。为了高效计算回归损失,我们提出了E-LatentLPIPS,一种直接在扩散模型潜在空间中操作的感知损失,利用增强集成。此外,我们调整扩散模型以构建多尺度判别器,并结合文本对齐损失,形成有效的基于条件GAN的公式。E-LatentLPIPS比许多现有蒸馏方法收敛更高效,即使考虑数据集构建成本。我们证明,我们的单步生成器在零样本COCO基准上优于最先进的单步扩散蒸馏模型——DMD、SDXL-Turbo和SDXL-Lightning。

英文摘要:

We propose a method to distill a complex multistep diffusion model into a single-step conditional GAN student model, dramatically accelerating inference, while preserving image quality. Our approach interprets diffusion distillation as a paired image-to-image translation task, using noise-to-image pairs of the diffusion model's ODE trajectory. For efficient regression loss computation, we propose E-LatentLPIPS, a perceptual loss operating directly in diffusion model's latent space, utilizing an ensemble of augmentations. Furthermore, we adapt a diffusion model to construct a multi-scale discriminator with a text alignment loss to build an effective conditional GAN-based formulation. E-LatentLPIPS converges more efficiently than many existing distillation methods, even accounting for dataset construction costs. We demonstrate that our one-step generator outperforms cutting-edge one-step diffusion distillation models -- DMD, SDXL-Turbo, and SDXL-Lightning -- on the zero-shot COCO benchmark.

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

↑