Recent progress in image generation models (IGMs) enables high-fidelity content creation but also amplifies risks, including the reproduction of copyrighted content and the generation of offensive content. Image Generation Model Unlearning (IGMU) mitigates these risks by removing harmful concepts without full retraining. Despite growing attention, the robustness under adversarial inputs, particularly image-side threats in black-box settings, remains underexplored. To bridge this gap, we present REFORGE, a black-box red-teaming framework that evaluates IGMU robustness via adversarial image prompts. REFORGE initializes stroke-based images and optimizes perturbations with a cross-attention-guided masking strategy that allocates noise to concept-relevant regions, balancing attack efficacy and visual fidelity. Extensive experiments across representative unlearning tasks and defenses demonstrate that REFORGE significantly improves attack success rate while achieving stronger semantic alignment and higher efficiency than involved baselines. These results expose persistent vulnerabilities in current IGMU methods and highlight the need for robustness-aware unlearning against multi-modal adversarial attacks. Our code is at: https://github.com/Imfatnoily/REFORGE.
Figures & tables
Fig. 1: Given that an unlearned IGM has undergone a concept-unlearning procedure (e.g., removal of the Van Gogh style), our adversarial image prompt Padv combined with the prompt Ptext can still bypass the unlearning mechanism, causing the erased style to re-emerge in the generated image I∗ .
Fig. 2: Overview of the ReForge framework. Sensitive parts are covered by .
Task
Method
ESD
UCE
AdvUnlearn
DoCo
MACE
ConceptPrune
Average
ASR
CLIP
ASR
CLIP
ASR
CLIP
ASR
CLIP
ASR
CLIP
ASR
CLIP
ASR
CLIP
Nudity
Text
32.00
24.62
54.66
25.22
4.66
19.89
76.00
26.06
24.00
19.34
94.66
26.16
47.66
23.55
SneakyPrompt
21.33
21.90
32.66
22.68
1.33
21.30
52.66
23.86
13.33
18.82
76.00
23.59
32.88
22.02
MMA
32.66
24.39
65.33
23.49
1.33
19.40
77.33
24.67
20.00
17.90
96.66
25.11
48.88
22.49
Ring-A-Bell
78.66
18.86
65.33
18.96
2.33
10.50
93.33
19.12
11.33
12.14
100.00
19.58
58.55
16.52
ReForge
65.33
25.83
69.33
26.15
62.66
22.33
89.33
24.46
14.66
17.95
98.00
26.46
66.55
24.19
TABLE I: Comparison of ASR (%) and CLIP Score across various unlearning tasks. For each method, the left column indicates ASR ( ↑ ) and the right indicates CLIP Score ( ↑ ). The best results are highlighted in bold , and the second-best are underlined .
Task
Nudity
Object-Parachute
Prompt
… perfect face, perfect and proportional body, black skin, sensual, in the style …
Parachute opening in mid-air.
Models
ESD
UCE
ESD
UCE
SneakyPrompt
MMA
Ring-A-Bell
ReForge
TABLE II: Generated images under different attacks.
Fig. 3: Ablation of key parameters: ASR (%) vs. (a) reference images, (b) timesteps, (c) layers, and (d) losses.
Advances in Generative AI (GenAI) have led to the development of various protection strategies to prevent the unauthorized use of images. These methods rely on adding imperceptible protective perturbations to images to thwart misuse such as style mimicry or deepfake manipulations. Although previous attacks on these protections required specialized, purpose-built methods, we demonstrate that this is no longer necessary. We show that off-the-shelf image-to-image GenAI models can be repurposed as generic ``denoisers" using a simple text prompt, effectively removing a wide range of protective perturbations. Across 8 case studies spanning 6 diverse protection schemes, our general-purpose attack not only circumvents these defenses but also outperforms existing specialized attacks while preserving the image's utility for the adversary. Our findings reveal a critical and widespread vulnerability in the current landscape of image protection, indicating that many schemes provide a false sense of security. We stress the urgent need to develop robust defenses and establish that any future protection mechanism must be benchmarked against attacks from off-the-shelf GenAI models. Code is available in this repository: https://github.com/mlsecviswanath/img2imgdenoiser
Xavier Pleimling, Sifat Muhammad Abdullah, Gunjan Balde +4
University of Texas at San Antonio San Antonio, TX, USA
Machine unlearning aims to remove specific concepts from pretrained text-to-image diffusion models, yet several white- and black-box attacks have been introduced to make the model generate such unlearned concepts. These attacks, nevertheless, do not assume a realistic threat model, i.e. they either assume access to the model weights, or result in gibberish adversarial prompts that could be easily detected even through naive rule-based safeguarding. We aim to address this gap in this paper. We introduce BEAP, a black-box, embedding-aware adversarial prompting attack that leverages a large language model (LLM) to iteratively generate effective adversarial prompts and exploit such hidden vulnerabilities. BEAP performs an embedding-aware search in text space, combining multiple reward signals: unlearned concept presence, text-image alignment, and image quality, to refine generated prompts. Unlike previous attack methods, BEAP keeps its prompts undetectable to safety filters while producing high-quality images. Across five unlearning methods, BEAP achieves a macro-averaged ASR of 97.8% under the held-out OpenNSFW2 evaluation, exceeding the white-box UDA baseline by 41.6 percentage points (56.2% to 97.8%). Counting unsuccessful searches at the full 100-query budget, BEAP uses 32.1 image-generation queries per evaluated prompt on average under this criterion.
Arian Komaei Koma, Seyed Amir Kasaei, AmirMahdi Sadeghzadeh +1
Department of Computer Engineering Sharif University of Technology
Machine unlearning has emerged as a critical post-hoc safety measure to erase sensitive concepts from Text-to-Image (T2I) models without prohibitive retraining. However, we reveal that current state-of-the-art (SOTA) approaches are brittle due to a severe lack of robustness to noise initialization. We call this phenomenon ``probabilistic forgetting'': suppressed concepts re-emerge under specific random initial noise conditions, despite appearing unlearned on other initializations. We trace this failure to the misalignment between standard Gaussian sampling during unlearning and the unlearning objective. Since the target concept manifests only in specific initial noise regions throughout the unlearning phase, uniform random sampling yields sparse, uninformative gradient updates that fail to drive robust erasure. To overcome this issue, we propose an adaptive, concept-conditioned sampling strategy that dynamically concentrates gradient updates on regions where the target concept manifests, down-weighting uninformative areas. We integrate our framework with six distinct SOTA unlearning methods across four diffusion backbones and evaluate it across safety, object, and artistic-style unlearning, as well as under black-box and white-box adversarial attacks. Our method reduces the conditional nudity re-emergence rate across random initializations by 67.2% on average over four baselines and lowers attack success rates across both adversarial evaluations. Across concept domains, Adaptive Noise Sampling strengthens adversarial robustness and non-target retention while preserving competitive generative quality and target-erasure performance.
Arian Komaei Koma, Seyed Amir Kasaei, Aida Aryafar +6
Sharif University of Technology · Hong Kong University of Science and Technology