cs.LGMar 26, 2026

DRiffusion: Draft-and-Refine Process Parallelizes Diffusion Models with Ease

Authors: Runsheng Bai, Chengyu Zhang, Yangdong Deng

Organizations: CSAIL, MIT · IST, Nanjing University · School of Software, Tsinghua University; FuturististAI Lab, Shanghai Tsinghua International Innovation Center

Abstract

Diffusion models have achieved remarkable success in generating high-fidelity content but suffer from slow, iterative sampling, resulting in high latency that limits their use in interactive applications. We introduce DRiffusion, a parallel sampling framework that parallelizes diffusion inference through a draft-and-refine process. DRiffusion employs skip transitions to generate multiple draft states for future timesteps and computes their corresponding noises in parallel, which are then used in the standard denoising process to produce refined results. Theoretically, our method achieves an acceleration rate of 1n\tfrac{1}{n} or 2n+1\tfrac{2}{n+1}, depending on whether the conservative or aggressive mode is used, where nn denotes the number of devices. Empirically, DRiffusion attains 1.4×\times-3.7×\times speedup across multiple diffusion models while incur minimal degradation in generation quality: on MS-COCO dataset, both FID and CLIP remain largely on par with those of the original model, while PickScore and HPSv2.1 show only minor average drops of 0.17 and 0.43, respectively. These results verify that DRiffusion delivers substantial acceleration and preserves perceptual quality.

Figures & tables

Explore similar work

CardsList
  1. Dual-Rate Diffusion: Accelerating diffusion models with an interleaved heavy-light network

    May 18, 2026Grigory Bartosh, David Ruhe, Emiel Hoogeboom +3Diffusion SamplingDiffusion-Based Image Generation

  2. Accelerating Discrete Diffusion Models with Parallel-In-Time Sampling

    Jul 1, 2026Yu Yao, Huanjian Zhou, Andi Han +2Diffusion SamplingDiffusion Model Inference Acceleration