CVMay 22, 2024

Directly Denoising Diffusion Models

arXiv:2405.13540v23 citationsh-index: 2ICML
Originality Incremental advance
AI Analysis

This work addresses the challenge of efficient image generation for AI applications, presenting an incremental improvement over existing diffusion models.

The paper tackles the problem of generating realistic images with few-step sampling in diffusion models, achieving FID scores of 2.57 and 2.33 on CIFAR-10 in one-step and two-step sampling, respectively, and reducing it to 1.79 with 1000 steps.

In this paper, we present the Directly Denoising Diffusion Model (DDDM): a simple and generic approach for generating realistic images with few-step sampling, while multistep sampling is still preserved for better performance. DDDMs require no delicately designed samplers nor distillation on pre-trained distillation models. DDDMs train the diffusion model conditioned on an estimated target that was generated from previous training iterations of its own. To generate images, samples generated from the previous time step are also taken into consideration, guiding the generation process iteratively. We further propose Pseudo-LPIPS, a novel metric loss that is more robust to various values of hyperparameter. Despite its simplicity, the proposed approach can achieve strong performance in benchmark datasets. Our model achieves FID scores of 2.57 and 2.33 on CIFAR-10 in one-step and two-step sampling respectively, surpassing those obtained from GANs and distillation-based models. By extending the sampling to 1000 steps, we further reduce FID score to 1.79, aligning with state-of-the-art methods in the literature. For ImageNet 64x64, our approach stands as a competitive contender against leading models.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes