Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

EGLOCE: Training-Free Energy-Guided Latent Optimization for Concept Erasure

About

As text-to-image diffusion models grow increasingly prevalent, the ability to remove specific concepts-mostly explicit content and many copyrighted characters or styles-has become essential for safety and compliance. Existing unlearning approaches often require costly re-training, modify parameters at the cost of degradation of unrelated concept fidelity, or depend on indirect inference-time adjustment that compromise the effectiveness of concept erasure. Inspired by the success of energy-guided sampling for preservation of the condition of diffusion models, we introduce Energy-Guided Latent Optimization for Concept Erasure (EGLOCE), a training-free approach that removes unwanted concepts by re-directing noisy latent during inference. Our method employs a dual-objective framework: a repulsion energy that steers generation away from target concepts via gradient descent in latent space, and a retention energy that preserves semantic alignment to the original prompt. Combined with previous approaches that either require erroneous modified model weights or provide weak inference-time guidance, EGLOCE operates entirely at inference and enhances erasure performance, enabling plug-and-play integration. Extensive experiments demonstrate that EGLOCE improves concept removal while maintaining image quality and prompt alignment across baselines, even with adversarial attacks. To the best of our knowledge, our work is the first to establish a new paradigm for safe and controllable image generation through dual energy-based guidance during sampling.

Junyeong Ahn, Seojin Yoon, Sungyong Baik• 2026

Related benchmarks

TaskDatasetResultRank
Nudity ErasureI2P--
38
NSFW suppressionUnlearn DiffAtk
ASR12
18
NSFW suppressionRing-a-Bell
ASR2.5
18
Style RemovalStyle Removal Van Gogh
Acce0.00e+0
16
NSFW suppressionP4D
ASR0.9
16
Artist-style removalKelly McKernan artist style unlearning
Accuracy (e)100
9
Nudity RemovalCOCO
FID13.65
8
Nudity RemovalMMA-Diffusion
ASR89.2
8
Object RemovalEnglish Springer
Accuracy (Specific)10
8
Object RemovalChurch
Accuracy (Acce)70
8
Showing 10 of 10 rows

Other info

Follow for update