atlas

Research map / Image generation / Fast diffusion sampling

AsyncDiff: Parallelizing Diffusion Models by Asynchronous Denoising

NeurIPS 2024 · Image generation › Fast diffusion sampling · approach: Training-free diffusion acceleration · 7 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes AsyncDiff, a training-free model-parallel scheme that splits the denoising U-Net into components on separate GPUs and uses previous-step hidden states to break sequential dependencies, plus stride denoising to cut communication.

Problem

Multi-step sequential denoising in diffusion models causes high cumulative latency and prevents parallel computation across devices; prior parallel methods (patch parallelism, Picard iterations) have low utilization, high memory, or quality loss.

Approach

Partition the noise prediction network into sequential components on different devices and feed each component the previous step's output from the preceding component, so the components run asynchronously on different timesteps; stride denoising further skips redundant computation and reduces broadcast frequency.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use asynchronous model parallelism with two componentsSD 2.1 · MS-COCO 2017 validationspeedup: 1.0x -> 1.8xHelped
Add stride denoising (S=2) (ablation)AsyncDiff on SD 2.1, 3 devices · SD 2.1speedup: 2.3x Faster -> 3.0x FasterHelped
Add stride denoising (S=2) (ablation)AsyncDiff on SD 2.1, 3 devices · SD 2.1CLIP Score: 31.56 -> 31.43Hurt
Add stride denoising (S=2) (ablation)AsyncDiff on SD 2.1, 4 devices · SD 2.1speedup: 2.6x Faster -> 4.0x FasterHelped
Add stride denoising (S=2) (ablation)AsyncDiff on SD 2.1, 4 devices · SD 2.1CLIP Score: 31.40 -> 31.22Hurt
Increase warm-up steps to 9 (ablation)AsyncDiff on SDXL · SDXLLPIPS: 0.3556 -> 0.2095Helped
Increase warm-up steps to 9 (ablation)AsyncDiff on SDXL · SDXLspeedup: 3.8x -> 2.8xHurt
Increase the number of model components and devices from N=2 to N=3 with stride 2 (ablation)AsyncDiff on SD 2.1 · MS-COCO 2017 validationspeedup: 3.0x -> 4.0xHelped
Increase the number of components from N=2 to N=4 with stride 1 (ablation)AsyncDiff on SDXL · MS-COCO 2017 validationspeedup: 1.7x -> 2.7xHelped

Surprising findings

Limitations

Experimental setup

Datasets: MS-COCO 2017 validation

Benchmarks: MS-COCO 2017 validation

Metrics: CLIP Score FID LPIPS latency speedup MACs DISTS MUSIQ CLIP-IQA NIQE

Baselines: DistriFusion Faster Diffusion ParaDiGMS Original model (sequential inference)

Builds on: DDIM DistriFusion Faster Diffusion ParaDiGMS

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy