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

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.

Ashutosh Ranjan, Vivek Srivastava, Shirish Karande, Murari Mandal• 2026

Related benchmarks

TaskDatasetResultRank
Style UnlearningUnlearnCanvas
UA0.9779
25
Safety Unlearning EvaluationRing-A-Bell Nudity (test)
ASR9.22
21
Safety Unlearning EvaluationRing-A-Bell Violence (test)
ASR5.54
21
Generation PreventionIP character
CLIPe0.17
16
Machine UnlearningSequential Unlearning Concepts (T1-T6) Stable Diffusion XL, SD v1.5, SANA
UA (T1)100
15
Object UnlearningObject Unlearning
Unlearning Accuracy (UA)95.36
13
Concept PreservationRelated Concept Categories Church
Preservation Score96
9
Concept PreservationRelated Concept Categories Parachute
Preservation Score95.6
9
Concept PreservationRelated Concept Categories Gas pump
Preservation Score92.4
9
Concept PreservationRelated Concept Categories Average
Preservation Score90.5
9
Showing 10 of 15 rows

Other info

Follow for update