atlas

Research map / Image generation / Editing and style transfer

DragonDiffusion: Enabling Drag-style Manipulation on Diffusion Models

ICLR 2024 · Image generation › Editing and style transfer · approach: Inversion-based diffusion editing · 33 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 DragonDiffusion, a tuning-free drag-style image editing method that turns diffusion-feature correspondence into energy-based gradient guidance plus a memory-bank visual cross-attention for consistency.

Problem

Precise, fine-grained image editing (drag-style moving, resizing, dragging, appearance replacing, pasting) is hard with text-conditioned diffusion models, which lack a compact editable latent space, and prior GAN-based drag methods are limited in generation quality and generalization.

Approach

Reuses the pre-trained SD UNet decoder features as a tuning-free energy function measuring feature correspondence between source and target regions, converting editing operations into gradient guidance during DDIM sampling; a memory bank from DDIM inversion supplies keys/values for visual cross-attention to preserve content consistency.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use DragonDiffusion instead of DragDiffusionCelebA-HQ face manipulation · 17-point manipulationpoint distance: 22.95 -> 18.51Helped
Use DragonDiffusion instead of DragDiffusionCelebA-HQ face manipulation · 68-point manipulationpoint distance: 17.32 -> 13.94Helped
Use DragonDiffusion instead of DragDiffusionCelebA-HQ face manipulation · 17/68-point manipulationFID: 38.06/36.55 -> 35.75/34.58Helped
Use DragonDiffusion instead of DragGANCelebA-HQ face manipulation · 17-point manipulationpoint distance: 15.96 -> 18.51Hurt
Use DragonDiffusion instead of DragGANCelebA-HQ face manipulation · 68-point manipulationpoint distance: 10.60 -> 13.94Hurt
Use DragonDiffusion instead of DragGANCelebA-HQ face manipulation · 17/68-point manipulationFID: 39.27/39.50 -> 35.75/34.58Helped
Replace the DDIM inversion prior with random initialization of z_T (ablation)DragonDiffusion object moving · Object movingcontent consistencyHurt
Remove content consistency guidance (ablation)DragonDiffusion object moving · Object movinglocal distortionHurt
Remove visual cross-attention (ablation)DragonDiffusion object moving · Object movingcontent consistencyHurt
Use a separate generation branch for guidance instead of the memory bank (ablation)DragonDiffusion editing · Image editingediting qualityHurt
Add null-text optimization to a separate guidance branch (ablation)DragonDiffusion editing · Image editingguidance accuracyHelped

Surprising findings

Limitations

Experimental setup

Datasets: CelebA-HQ

Benchmarks: CelebA-HQ face keypoint manipulation (17 and 68 points)

Metrics: Euclidean distance to target points FID inference and preparing time user study votes

Baselines: DragGAN DragDiffusion UserControllableLT

Builds on: Stable Diffusion DIFT DDIM inversion classifier guidance DragGAN

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