arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.11713cs.CV

Latent Denoising Diffusion GAN:更快采样、更高图像质量

Latent Denoising Diffusion GAN: Faster sampling, Higher image quality

Yokohama National University(横滨国立大学)

Luan Thanh Trinh, Tomoki Hamagami

AI总结:

针对扩散模型推理慢且质量不及 GAN 的问题,本文提出潜空间去噪扩散 GAN,并用 Weighted Learning 提升多样性与画质,在三个数据集上达到扩散模型中的最优速度并全面超越前代方法。

摘要:

扩散模型正成为生成高保真且多样化图像的强大方案,在许多情况下往往优于 GAN。然而,其缓慢的推理速度阻碍了其在实时应用中的潜力。为解决这一问题,DiffusionGAN 利用条件 GAN 大幅减少去噪步骤并加快推理。其改进版本 Wavelet Diffusion 通过将数据转换到小波空间进一步加速了该过程,从而提升了效率。尽管如此,这些模型在速度和图像质量方面仍不及 GAN。为弥合这些差距,本文提出 Latent Denoising Diffusion GAN,它采用预训练自动编码器将图像压缩到紧凑的潜空间中,显著提高推理速度和图像质量。此外,我们提出一种 Weighted Learning 策略以增强多样性和图像质量。在 CIFAR-10、CelebA-HQ 和 LSUN-Church 数据集上的实验结果证明,我们的模型在扩散模型中实现了最先进的运行速度。与其前身 DiffusionGAN 和 Wavelet Diffusion 相比,我们的模型在所有评估指标上均表现出显著提升。代码和预训练检查点:\url{https://github.com/thanhluantrinh/LDDGAN.git}

英文摘要:

Diffusion models are emerging as powerful solutions for generating high-fidelity and diverse images, often surpassing GANs under many circumstances. However, their slow inference speed hinders their potential for real-time applications. To address this, DiffusionGAN leveraged a conditional GAN to drastically reduce the denoising steps and speed up inference. Its advancement, Wavelet Diffusion, further accelerated the process by converting data into wavelet space, thus enhancing efficiency. Nonetheless, these models still fall short of GANs in terms of speed and image quality. To bridge these gaps, this paper introduces the Latent Denoising Diffusion GAN, which employs pre-trained autoencoders to compress images into a compact latent space, significantly improving inference speed and image quality. Furthermore, we propose a Weighted Learning strategy to enhance diversity and image quality. Experimental results on the CIFAR-10, CelebA-HQ, and LSUN-Church datasets prove that our model achieves state-of-the-art running speed among diffusion models. Compared to its predecessors, DiffusionGAN and Wavelet Diffusion, our model shows remarkable improvements in all evaluation metrics. Code and pre-trained checkpoints: \url{https://github.com/thanhluantrinh/LDDGAN.git}

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

↑