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

Concept Removal Guidance: Evidence-Calibrated Negative Guidance for Safe Diffusion Sampling

About

Text-to-image diffusion models remain vulnerable to adversarial prompts that elicit disallowed content, motivating reliable inference-time controls. A popular approach is negative guidance, which subtracts a negative prompt direction with a fixed weight. However, it often forces a safety-fidelity trade-off, causing artifacts or prompt drift when over-applied and failing under attacks when under-applied. Dynamic variants reweight guidance using posterior-odds signals, which can be brittle for open-vocabulary compositional prompts, while lightweight similarity-based methods ignore the evolving image evidence along the denoising trajectory. We introduce Concept Removal Guidance (CRG), a training-free method that estimates unwanted-concept presence at each diffusion step from the model's noise predictions, and adaptively calibrates negative guidance via a closed-form constrained update enforcing a target presence threshold while minimally perturbing the conditional trajectory. Across red-teaming benchmarks, CRG reduces attack success rates while preserving benign fidelity, and extends to additional suppression targets such as artist style and violence without fine-tuning or external classifiers.

Yoonseok Choi, Chaeyoung Oh, Hyunjun Choi, Seokin Seo, Kee-Eung Kim• 2026

Related benchmarks

TaskDatasetResultRank
Text-to-Image GenerationCOCO
FID37.63
104
Text-to-Image GenerationCOCO 30k
FID36.05
77
Concept RemovalRing-A-Bell Nudity
Attack Success Rate (ASR)2.1
52
Nudity RemovalMMA-Diffusion
ASR16.4
45
NSFW suppressionP4D
ASR0.06
41
Nudity Concept RemovalUnlearnAtk
ASR8.5
25
Nudity RemovalCOCO
CLIP Score32.07
18
Artist-style removalStable Diffusion Artist Style Removal (Van Gogh) v1.4
Accuracy90
18
Inference LatencyGeneric Prompts
Inference Latency (s)4.8
18
Artist-style removalStable Diffusion Artist Style Removal (Kelly McKernan) v1.4
Accuracy (Exact Match)15
18
Showing 10 of 17 rows

Other info

Follow for update