atlas

Research map / Image generation / Text-to-image diffusion

CCM: Real-Time Controllable Visual Content Creation Using Text-to-Image Consistency Models

ICML 2024 · Image generation › Text-to-image diffusion · approach: Layout-to-image control

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.

Trains ControlNet for text-to-image latent consistency models directly via consistency training (plus a unified multi-condition adapter to transfer DM ControlNets), enabling real-time controllable generation in 1-4 steps.

Problem

Adding new conditional controls (e.g., sketch, depth, pose) to pre-trained text-to-image consistency models for fast controllable generation, since ControlNets trained for diffusion models transfer imperfectly and lose low-level detail and realism.

Approach

Treat the frozen text-to-image consistency model plus a trainable ControlNet as a new conditional consistency model and train the ControlNet from scratch with the consistency training loss; additionally train a unified multi-condition adapter with consistency training to better adapt diffusion-trained ControlNets to CMs.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Train CM’s ControlNet with consistency training instead of transferring DM’s ControlNet directlyControllable generation with CM · Sketch2Image, Depth2Image, Mask2Image, 16×SRFID: 30.71 -> 10.39; 26.08 -> 12.94; 14.67 -> 5.44; 21.32 -> 7.60Helped
Add a unified adapter to DM’s ControlNetDM’s ControlNet+CM · Sketch2Image, Depth2Image, Mask2Image, 16×SRFID: 30.71 -> 20.43; 26.08 -> 19.75; 14.67 -> 13.95; 21.32 -> 13.73Helped
Train CM’s ControlNet with consistency training rather than use DM’s ControlNet with CMControllable generation with CM · Sketch2Image, Depth2Image, Mask2Image, 16×SRFidelity: 0.083 -> 0.095; 0.193 -> 0.169; 0.431 -> 0.082; 0.237 -> 0.118Mixed
Use L1 distance instead of MSE distance (ablation)CM’s ControlNet consistency training · Sketch2ImageFID: 20.16 -> 9.30Helped
Use L1 distance instead of Charbonnier distance (ablation)CM’s ControlNet consistency training · Sketch2ImageFID: 11.91 -> 9.30Helped
Use L1 distance instead of randomly picking a distance function (ablation)CM’s ControlNet consistency training · Sketch2ImageFID: 11.37 -> 9.30Helped
Set consistency-training hyperparameter N to 50 rather than 100, 200, 500, 1000, or use a progressive schedule (ablation)CM’s ControlNet consistency training · Sketch2ImageFID: 9.68; 10.31; 10.25; 10.37; 10.21 -> 9.30Helped
Use the general prompt “A high-quality and professional image” rather than paired image captions (ablation)CM’s ControlNet · Sketch2ImageFID: 10.39 -> 11.25Hurt

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet21K WebVision LAION (filtered)

Benchmarks: FID (10k samples) Fidelity on sketch, depth, mask, and 16x super-resolution conditions

Metrics: FID Fidelity (L1 distance between condition extractor output and NFEs running time

Baselines: DM's ControlNet+DM (DDIM 50 steps) DM's ControlNet+CM DM's ControlNet+CM+Adapter

Builds on: Consistency Models ControlNet Latent Consistency Models Stable Diffusion Consistency distillation

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