Forgetting is Competition: Rethinking Unlearning as Representation Interference in Diffusion Models
About
Unlearning in text-to-image diffusion models often leads to uneven concept removal and unintended forgetting of unrelated capabilities. This complicates tasks such as copyright compliance, protected data mitigation, artist opt-outs, and policy-driven content updates. As models grow larger and adopt more diverse architectures, achieving precise and selective unlearning while preserving generative quality becomes increasingly challenging. We introduce SurgUn (pronounced as Surgeon), a surgical unlearning method that applies targeted weight-space updates to remove specific visual concepts in text-conditioned diffusion models. Our approach is motivated by retroactive interference theory, which holds that newly acquired memories can overwrite, suppress, or impede access to prior ones by competing for shared representational pathways. We adapt this principle to diffusion models by inducing retroactive concept interference, enabling focused destabilization of only the target concept while preserving unrelated capabilities through a novel training paradigm. SurgUn achieves high-precision unlearning across diverse settings. It performs strongly on compact U-Net based models such as Stable Diffusion v1.5, scales effectively to the larger U-Net architecture SDXL, and extends to SANA, representing an underexplored Diffusion Transformer based architecture for unlearning.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Style Unlearning | UnlearnCanvas | UA0.9779 | 25 | |
| Safety Unlearning Evaluation | Ring-A-Bell Nudity (test) | ASR9.22 | 21 | |
| Safety Unlearning Evaluation | Ring-A-Bell Violence (test) | ASR5.54 | 21 | |
| Generation Prevention | IP character | CLIPe0.17 | 16 | |
| Machine Unlearning | Sequential Unlearning Concepts (T1-T6) Stable Diffusion XL, SD v1.5, SANA | UA (T1)100 | 15 | |
| Object Unlearning | Object Unlearning | Unlearning Accuracy (UA)95.36 | 13 | |
| Concept Preservation | Related Concept Categories Church | Preservation Score96 | 9 | |
| Concept Preservation | Related Concept Categories Parachute | Preservation Score95.6 | 9 | |
| Concept Preservation | Related Concept Categories Gas pump | Preservation Score92.4 | 9 | |
| Concept Preservation | Related Concept Categories Average | Preservation Score90.5 | 9 |