Unlearning the Unpromptable: Prompt-free Instance Unlearning in Diffusion Models
Authors: Kyungryeol Lee, Kyeonghyun Lee, Seongmin Hong, Byung Hyun Lee, Se Young Chun
Organizations: Department of Electrical and Computer Engineering (ECE), Seoul National University (SNU), Seoul, 08826, Republic of Korea · Institute of New Media and Communications (INMC), SNU, Seoul, 08826, Republic of Korea · ECE, INMC, IPAI, SNU, Seoul, 08826, Republic of Korea
Machine unlearning aims to remove specific outputs from trained models, often at the concept level, such as forgetting all occurrences of a particular celebrity or filtering content via text prompts. However, many undesired outputs, such as an individual's face or generations culturally or factually misinterpreted, cannot often be specified by text prompts. We address this underexplored setting of instance unlearning for outputs that are undesired but unpromptable, where the goal is to forget target outputs selectively while preserving the rest. To this end, we introduce an effective surrogate-based unlearning method that leverages image editing, timestep-aware weighting, and gradient surgery to guide trained diffusion models toward forgetting specific outputs. Experiments on conditional (Stable Diffusion 3) and unconditional (DDPM-CelebA) diffusion models demonstrate that our prompt-free method uniquely unlearns unpromptable outputs, such as faces and culturally inaccurate depictions, with preserved integrity, unlike prompt-based and prompt-free baselines. Our proposed method would serve as a practical hotfix for diffusion model providers to ensure privacy protection and ethical compliance.
Figures & tables
Fig. 1 : Challenge and our solution for instance unlearning in diffusion models.
Fig. 2 : Cultural and semantic misrepresentation highlight the need for instance unlearning in commercial generative models.
Fig. 3 : Comparison between (a) prompt-based and (b) instance unlearning.
Fig. 4 : Surrogate-based unlearning ( θ† ) can be better than exact unlearing ( θ~ ) in mapping preservation, i.e. , the line is closer to the original ( θ∗ ).
Fig. 5 : Surrogate data construction of (left) CelebA with TediGAN, (middle) SD3 with SDEdit, and (right) manual editing.
Fig. 6 : Overview of SSCD to validate the success of forgetting.
Fig. 7 : Qualitative results on (a) forgetting target instance on (b) preserving model integrity on generation after unlearning.
Single Instance
Multiple Instances
SSCD < 0.4
LPIPS ↓
SSIM ↑
FIDpre↓
FIDreal↓
SSCD < 0.4
LPIPS ↓
SSIM ↑
FIDpre↓
FIDreal↓
NegGrad [ 39 ]
✓(0.18)
1.00
0.34
223.5
226.1
✓(0.14)
0.96
0.07
415.4
410.8
EraseDiff [ 40 ]
✓(0.03)
0.74
0.61
80.7
87.8
✓(0.04)
0.66
0.71
45.8
51.1
SISS [ 41 ]
✓(0.21)
0.43
0.84
15.7
21.3
✓(0.22)
0.51
0.81
26.8
31.5
Ours †
✓(0.17)
0.41
0.84
15.7
24.1
✓(0.16)
0.36
0.80
13.9
27.2
Ours
✓(0.19)
0.35
0.87
9.8
18.0
✓(0.21)
0.37
0.88
17.5
23.4
TABLE I : Quantitative results on forgetting single instance and multiple instances. Ours † refers to randomly sampling the surrogate data x0s from the remember dataset Dr at each update.
Fig. 8: Comparison of instance unlearning methods on (top) forgetting target instance and (bottom) preserving related instances.
Fig. 9 : Edited images (surrogates) with varying CLIP loss weights of TediGAN.
Fig. 10 : Trade-off between forgetting (SSCD) and model integrity (SSIM, LPIPS). We can select how the surrogate is close to the original instance by adjusting CLIP loss weight during editing.
CLIP weight
SSCD
LPIPS ↓
SSIM ↑
FID pre↓
FID real↓
0.5
0.434
0.305
0.900
8.08
16.25
1.0
0.403
0.304
0.901
8.04
16.33
2.0
0.330
0.307
0.900
8.10
16.62
4.0
0.302
0.315
0.896
8.34
17.09
8.0
0.278
0.325
0.891
8.80
17.62
16.0
0.228
0.338
0.885
8.94
17.48
TABLE II : Quantitative results with varying CLIP loss weights of TediGAN.
Condition
SSCD
LPIPS ↓
SSIM ↑
FIDpre↓
FIDreal↓
λ=0.95
0.279
0.174
0.897
7.82
24.06
λ=0.99
0.289
0.173
0.894
7.18
20.31
λ=1−2⋅10−5t
0.356
0.213
0.920
6.06
15.43
λ=1−5⋅10−5t
0.355
0.213
0.920
6.06
15.43
λ=1−10−4t
0.311
0.235
0.910
6.59
15.58
TABLE III : Ablation on timestep scheduling of λ .
Condition
SSCD
LPIPS ↓
SSIM ↑
FIDpre↓
FIDreal↓
no projection
0.309
0.275
0.899
7.68
14.72
project gf
0.309
0.274
0.900
7.65
14.70
project gf , gr
0.310
0.276
0.901
8.15
16.47
TABLE IV : Ablation on gradient projection.
Surrogate method
SSCD < 0.4
LPIPS ↓
SSIM ↑
Edit (TediGAN)
✓(0.35)
0.21
0.92
Flip
✓(0.40)
0.31
0.85
Add noise ( σ=50 )
✗ (0.54)
0.23
0.91
Add noise ( σ=150 )
✓(0.32)
0.28
0.89
TABLE V : Ablation on surrogate construction methods.
Fig. 11: Qualitative comparison of surrogates. σ=50 and σ=150 denote adding Gaussian noise with standard deviations of 50 and 150 to the target image.
SSCD
LPIPS ↓
SSIM ↑
FIDpre↓
FIDreal↓
Ours (InD)
0.34
0.30
0.90
8.7
16.6
Ours (OoD)
0.32
0.31
0.90
9.0
17.1
TABLE VI : Single instance unlearning for InD and OoD data.