Abstract:In this paper, we point out that suboptimal noise-data mapping leads to slow training of diffusion models. During diffusion training, current methods diffuse each image across the entire noise space, resulting in a mixture of all images at every point in the noise layer. We emphasize that this random mixture of noise-data mapping complicates the optimization of the denoising function in diffusion models. Drawing inspiration from the immiscibility phenomenon in physics, we propose Immiscible Diffusion, a simple and effective method to improve the random mixture of noise-data mapping. In physics, miscibility can vary according to various intermolecular forces. Thus, immiscibility means that the mixing of molecular sources is distinguishable. Inspired by this concept, we propose an assignment-then-diffusion training strategy to achieve Immiscible Diffusion. As one example, prior to diffusing the image data into noise, we assign diffusion target noise for the image data by minimizing the total image-noise pair distance in a mini-batch. The assignment functions analogously to external forces to expel the diffuse-able areas of images, thus mitigating the inherent difficulties in diffusion training. Our approach is remarkably simple, requiring only one line of code to restrict the diffuse-able area for each image while preserving the Gaussian distribution of noise. In this way, each image is preferably projected to nearby noise. Experiments demonstrate that our method can achieve up to 3x faster training for unconditional Consistency Models on the CIFAR dataset, as well as for DDIM and Stable Diffusion on CelebA and ImageNet dataset, and in class-conditional training and fine-tuning. In addition, we conducted a thorough analysis that sheds light on how it improves diffusion training speed while improving fidelity. The code is available at <a class="link-external link-https" href="https://yhli123.github.io/immiscible-diffusion" rel="external noopener nofollow">this https URL</a>

Consistent Diffusion Meets Tweedie: Training Exact Ambient Diffusion Models with Noisy Data

Ambient Diffusion: Learning Clean Distributions from Corrupted Data

An Expectation-Maximization Algorithm for Training Clean Diffusion Models from Corrupted Observations

GSURE-Based Diffusion Model Training with Corrupted Data

Slight Corruption in Pre-training Data Makes Better Diffusion Models

Learning Diffusion Model from Noisy Measurement using Principled Expectation-Maximization Method

Ambient Diffusion Posterior Sampling: Solving Inverse Problems with Diffusion Models trained on Corrupted Data

Integrating Amortized Inference with Diffusion Models for Learning Clean Distribution from Corrupted Images

Your Diffusion Model is Secretly a Noise Classifier and Benefits from Contrastive Training

Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment

How much is a noisy image worth? Data Scaling Laws for Ambient Diffusion

Consistent Diffusion: Denoising Diffusion Model with Data-Consistent Training for Image Restoration

DiracDiffusion: Denoising and Incremental Reconstruction with Assured Data-Consistency

Unmasking Bias in Diffusion Model Training

ACT-Diffusion: Efficient Adversarial Consistency Training for One-step Diffusion Models

Extracting Training Data from Diffusion Models

Training-Free Adaptive Diffusion with Bounded Difference Approximation Strategy

Unlearnable Examples for Diffusion Models: Protect Data from Unauthorized Exploitation

Towards Faster Training of Diffusion Models: An Inspiration of A Consistency Phenomenon

Erasing Undesirable Influence in Diffusion Models

Analyzing and Improving the Training Dynamics of Diffusion Models