REFORGE: Multi-modal Attacks Reveal Vulnerable Concept Unlearning in Image Generation Models

Yong Zou, Haoran Li, Fanxiao Li, Shenyang Wei, Yunyun Dong +3 more
3/17/2026
cs.CVcs.AIcs.CR

Official code: Imfatnoily/REFORGE (1 star, MIT, last commit 2025-12-30).

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.

View on arXivView PDF

Code Implementations(1)

Imfatnoily/REFORGEOfficial100%
MIT

REFORGE: Multi-modal Attacks Reveal Vulnerable Concept Unlearning in Image Generation Models

10PythonDec 29, 20259 months agoMIT

Similar Papers

Cite this paper

@article{zou2026reforge,
  title  = {REFORGE: Multi-modal Attacks Reveal Vulnerable Concept Unlearning in Image Generation Models},
  author = {Yong Zou and Haoran Li and Fanxiao Li and Shenyang Wei and Yunyun Dong and Li Tang and Wei Zhou and Renyang Liu},
  year   = {2026},
  eprint = {2603.16576},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.16576v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2603.16576.svg)](https://codeofpaper.com/paper/2603.16576)

Discussion