atlas

Research map / Image generation / Editing and style transfer

Zero-shot Image Editing with Reference Imitation

NeurIPS 2024 · Image generation › Editing and style transfer · approach: Localized and attribute-specific editing · 17 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 imitative editing, where a masked region of a source image is filled by automatically finding and imitating the corresponding part of a reference image, trained self-supervised on video frame pairs with a dual U-Net diffusion model.

Problem

Local image editing typically relies on text prompts (inpainting) or carefully cropped reference objects (composition), which fail to capture fine-grained local parts or patterns; the paper targets editing a masked region using an unsegmented in-the-wild reference image without paired masks.

Approach

Train dual diffusion U-Nets (imitative and reference) self-supervised on two frames from the same video, masking one frame and recovering it from the other; reference U-Net keys/values are concatenated into the imitative U-Net attention, with SIFT-guided masking, strong augmentation, and optional depth conditioning for texture transfer.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace reference U-Net with CLIP Encoder (ablation)MimicBrush · Part compositionSSIM: 0.70 -> 0.66Hurt
replace reference U-Net with DINOv2 Encoder (ablation)MimicBrush · Part compositionSSIM: 0.70 -> 0.67Hurt
replace reference U-Net with CLIP Encoder (ablation)MimicBrush · Texture transferDINO-I: 49.83 -> 37.86Hurt
replace reference U-Net with DINOv2 Encoder (ablation)MimicBrush · Texture transferDINO-I: 49.83 -> 46.34Hurt
train using image data only (ablation)MimicBrush · Part compositionDINO-I: 56.48 -> 39.68Hurt
train using image data only (ablation)MimicBrush · Texture transferDINO-I: 49.83 -> 41.30Hurt
use weak augmentation (ablation)MimicBrush · Part compositionDINO-I: 56.48 -> 50.55Hurt
use weak augmentation (ablation)MimicBrush · Texture transferDINO-I: 49.83 -> 50.92Helped
use single-box masking (ablation)MimicBrush · Part compositionSSIM: 0.70 -> 0.66Hurt
use Mask Grid 0.25 (ablation)MimicBrush · Part compositionSSIM: 0.70 -> 0.68Hurt
use Mask Grid 0.50 (ablation)MimicBrush · Part compositionSSIM: 0.70 -> 0.68Hurt
use Mask Grid 0.75 (ablation)MimicBrush · Part compositionSSIM: 0.70 -> 0.67Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Pexels videos (100k) SAM dataset

Benchmarks: MimicBrush benchmark (part composition, texture transfer) DreamBooth inner-ID pairs

Metrics: SSIM PSNR LPIPS DINO-I CLIP-I CLIP-T user preference rank

Baselines: Paint-by-Example AnyDoor Firefly

Builds on: Stable Diffusion 1.5 inpainting Depth Anything AnyDoor Paint-by-Example

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