REFORGE: Multi-modal Attacks Reveal Vulnerable Concept Unlearning in Image Generation Models
Organizations: Yunnan University · Northeastern University · National University of Singapore
Abstract
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
| 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 | |
| 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 | ||||
Adversarial Attacks
Adversarial attacks exploit crafted inputs, poisoned data, or system interactions to manipulate AI predictions, behavior, privacy, or reliability.
See all 447 papers