Research map / Image generation / Autoregressive and latent generation
Attentive Eraser: Unleashing Diffusion Model’s Object Removal Potential via Self-Attention Redirection Guidance
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.
Attentive Eraser is a tuning-free method that redirects diffusion self-attention (Attention Activation and Suppression plus Self-Attention Redirection Guidance) to remove masked objects and repaint plausible background in pre-trained Stable Diffusion models.
Problem
Diffusion-based object removal suffers from random artifacts, incomplete removal, and poor repainting of the masked foreground after object deletion, and existing fixes need costly fine-tuning or heavy prompt engineering.
Approach
Modify the self-attention maps inside a pre-trained SD U-net using the object mask so foreground tokens attend to background and not to themselves (AAS, with similarity suppression), then use this redirected prediction as a PAG-style guidance term during sampling (SARG).
Key results
- On OpenImages V5 (10k images), SD2.1+SIP with SARG reaches Local FID 5.835 and CLIP consensus 0.0734, versus 15.58 and 0.1347 without SARG, and best-in-class local FID among compared methods (LAMA 6.091).
- In a user study with 10 participants, SD2.1+SIP w/ SARG was preferred 44.9% of the time, versus 19.7% for LAMA and 10% for SD2.1-inpainting.
- In GPT-4o pairwise evaluation against LAMA over 3000 image pairs, the proposed method was selected 74.47% of the time.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Add SARG to SD2.1+SIP (ablation) | SD2.1+SIP · OpenImages V5 test | Local FID: 15.58 -> 5.835 | Helped |
| Add SARG to SD2.1+SIP (ablation) | SD2.1+SIP · OpenImages V5 test | FID: 5.98 -> 7.352 | Hurt |
| Add SARG to SD2.1+SIP (ablation) | SD2.1+SIP · OpenImages V5 test | LPIPS: 0.2998 -> 0.3113 | Hurt |
| Add SARG to SD2.1+SIP (ablation) | SD2.1+SIP · OpenImages V5 test | CLIP consensus: 0.1347 -> 0.0734 | Helped |
| Add SARG to SD2.1+SIP (ablation) | SD2.1+SIP · OpenImages V5 test | CLIP score: 22.05 -> 23.56 | Helped |
| Apply AAS alone, without SARG guidance (ablation) | Attentive Eraser · Object removal | object removal effectiveness | Hurt |
| Add SARG to AAS (ablation) | Attentive Eraser · Object removal | object removal effectiveness | Helped |
| Add Similarity Suppression (SS) (ablation) | Attentive Eraser · Object removal with similar background objects | similar-object generation | Helped |
| Decrease suppression factor λ (ablation) | Attentive Eraser · Object removal with similar background objects | similar-object generation | Helped |
| Increase removal guidance scale s (ablation) | Attentive Eraser · Object removal | removal intensity | Helped |
| Use SD2.1+SIP with SARG instead of LAMA | Object removal methods · OpenImages V5 test | Local FID: 6.091 -> 5.835 | Helped |
| Use SD2.1+SIP with SARG instead of LAMA | Object removal methods · OpenImages V5 test | LPIPS: 0.2189 -> 0.3113 | Hurt |
Surprising findings
- Global metrics FID and LPIPS do not reflect removal effectiveness; the proposed method is only average on them (FID 7.352 vs. best 3.805).
Limitations
- Guidance requires two U-Net noise predictions per step, increasing inference time.
- When the mask is very large, few background references exist, causing reconstruction artifacts.
- Global metrics FID and LPIPS do not fully capture removal quality, and the method is only average on them.
Experimental setup
Datasets: OpenImages V5
Benchmarks: OpenImages V5
Metrics: FID LPIPS Local FID CLIP consensus CLIP score user preference GPT-4o selection rate
Baselines: SD-Inpaint LAMA Inst-Inpaint PowerPaint SD2.1 with SIP
Builds on: Perceptual Attention Guidance (PAG) Blended Latent Diffusion (BLD) DiffEdit DDIM inversion Self-Attention Guidance (SAG)
Closest papers by meaning
- CLIPAway: Harmonizing focused embeddings for removing objects via diffusion models (NeurIPS 2024 · similarity 0.86)
- Erase Diffusion: Empowering Object Removal Through Calibrating Diffusion Pathways (CVPR 2025 · similarity 0.86)
- Diffusion Handles Enabling 3D Edits for Diffusion Models by Lifting Activations to 3D (CVPR 2024 · similarity 0.85)
- Concept Pinpoint Eraser for Text-to-image Diffusion Models via Residual Attention Gate (ICLR 2025 · similarity 0.85)
- Towards Understanding Cross and Self-Attention in Stable Diffusion for Text-Guided Image Editing (CVPR 2024 · similarity 0.85)
- Fine-Grained Erasure in Text-to-Image Diffusion-based Foundation Models (CVPR 2025 · similarity 0.85)
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.
