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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Text-to-Image Generation | COCO | FID37.63 | 104 | |
| Text-to-Image Generation | COCO 30k | FID36.05 | 77 | |
| Concept Removal | Ring-A-Bell Nudity | Attack Success Rate (ASR)2.1 | 52 | |
| Nudity Removal | MMA-Diffusion | ASR16.4 | 45 | |
| NSFW suppression | P4D | ASR0.06 | 41 | |
| Nudity Concept Removal | UnlearnAtk | ASR8.5 | 25 | |
| Nudity Removal | COCO | CLIP Score32.07 | 18 | |
| Artist-style removal | Stable Diffusion Artist Style Removal (Van Gogh) v1.4 | Accuracy90 | 18 | |
| Inference Latency | Generic Prompts | Inference Latency (s)4.8 | 18 | |
| Artist-style removal | Stable Diffusion Artist Style Removal (Kelly McKernan) v1.4 | Accuracy (Exact Match)15 | 18 |