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>

Improved Noise Schedule for Diffusion Training

Efficient Diffusion Training Via Min-SNR Weighting Strategy.

Error Bound Based Noise Schedule Design in Diffusion Models

Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment

Schedule On the Fly: Diffusion Time Prediction for Faster and Better Image Generation

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

Constant Rate Schedule: Constant-Rate Distributional Change for Efficient Training and Sampling in Diffusion Models

Training-Free Adaptive Diffusion with Bounded Difference Approximation Strategy

Denoising Task Difficulty-based Curriculum for Training Diffusion Models

Analyzing and Improving the Training Dynamics of Diffusion Models

A Fully Quantized Training Accelerator for Diffusion Network with Tensor Type & Noise Strength Aware Precision Scheduling

Diffusion Models With Learned Adaptive Noise

One More Step: A Versatile Plug-and-Play Module for Rectifying Diffusion Schedule Flaws and Enhancing Low-Frequency Controls

Golden Noise for Diffusion Models: A Learning Framework

A Noise is Worth Diffusion Guidance

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

Schedule Your Edit: A Simple yet Effective Diffusion Noise Schedule for Image Editing

Improving Efficiency of Diffusion Models via Multi-Stage Framework and Tailored Multi-Decoder Architectures

Score-Optimal Diffusion Schedules

Not All Noises Are Created Equally:Diffusion Noise Selection and Optimization