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>

Non-uniform Timestep Sampling: Towards Faster Diffusion Model Training

Adaptive Non-Uniform Timestep Sampling for Diffusion Model Training

Accelerating Parallel Sampling of Diffusion Models

Non-Uniform Diffusion Models

AutoDiffusion: Training-Free Optimization of Time Steps and Architectures for Automated Diffusion Model Acceleration

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

Choose Your Diffusion: Efficient and flexible ways to accelerate the diffusion model in fast high energy physics simulation

Towards More Accurate Diffusion Model Acceleration with A Timestep Aligner

A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training

Fast Sampling via Discrete Non-Markov Diffusion Models with Predetermined Transition Time

T-Stitch: Accelerating Sampling in Pre-Trained Diffusion Models with Trajectory Stitching

TMPQ-DM: Joint Timestep Reduction and Quantization Precision Selection for Efficient Diffusion Models

Towards Faster Non-Asymptotic Convergence for Diffusion-Based Generative Models

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

PFDiff: Training-free Acceleration of Diffusion Models through the Gradient Guidance of Past and Future

Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment

Fast constrained sampling in pre-trained diffusion models

Stochastic Runge-Kutta Methods: Provable Acceleration of Diffusion Models

Faster Diffusion: Rethinking the Role of the Encoder for Diffusion Model Inference

Contour wavelet diffusion: A fast and high‐quality image generation model

Towards Accurate Post-training Quantization for Diffusion Models