You Can't Have It Both Ways: Concept Entanglement Limits Diffusion Model Unlearning
cs.AI
Submitted: 2026-09-28
Updated: 2026-09-28
Comments: 40th Conference on Neural Information Processing Systems (NeurIPS 2026)
Code: https://github.com/Jazzssmine/concept-entangle
License: http://creativecommons.org/licenses/by/4.0/
The gist: Concept unlearning in text-to-image diffusion models aims to suppress a target concept (e.g., horse) while preserving related but distinct content (e.g., donkey), yet existing methods either leak
Terminology
Abstract
Concept unlearning in text-to-image diffusion models aims to suppress a target concept (e.g., horse) while preserving related but distinct content (e.g., donkey), yet existing methods either leak under indirect prompts or visibly degrade other concepts. We show that these failure modes stem from the geometry of concept representations rather than from any particular algorithm. Formalizing concepts as activation-space regions, we prove that the overlap between a target and other concepts lower-bounds the damage any robust erasure must inflict on them, with the trade-off scaling linearly in the degree of overlap. Across thirteen unlearning methods, including methods designed to preserve non-target concepts, no method achieves both strong erasure and strong neighbor preservation: STEREO nearly eliminates indirect leakage but cuts neighbor generation by more than 75%, while sparse inference-time methods preserve neighbors but leak. Damage increases with our overlap measure, monotonically so for STEREO; the κ-scaling reproduces on SDXL, and neighbor-selective damage recurs on FLUX. Perfect unlearning is the wrong target for entangled concepts; methods should be evaluated on the Pareto frontier our theorem establishes.
Sources
- Erasing Undesirable Concepts in Diffusion Models with Adversarial Preservation
- SAeUron: Interpretable Concept Unlearning in Diffusion Models with Sparse Autoencoders
- SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation
- Selective Fine-Tuning for Targeted and Robust Concept Unlearning
- Get What You Want, Not What You Don't: Image Content Suppression for Text-to-Image Diffusion Models
- Erasing Undesirable Influence in Diffusion Models
- When and why vision-language models behave like bags-of-words, and what to do about it?
- Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems