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

Obliviate: Erasing Concepts from Autoregressive Image Generation Models

About

The widespread adoption of generative AI models has intensified concerns about misuse, including the creation of unsafe or disturbing imagery. To mitigate such issues, several concept erasure approaches have been proposed to remove harmful content from multimodal generative models. Yet concept erasure for autoregressive image generation remains largely unexplored, despite the growing relevance of these models in recent trends toward unified multimodal architectures. In this work, we fill this gap by introducing Obliviate, a guidance-based concept erasure method for autoregressive image generation. Our method builds on three key design choices: KL-based supervision over visual token distributions, trajectory-level updates over full autoregressive rollouts, and aligned visual prefixes for stable target construction. We evaluate Obliviate on three state-of-the-art autoregressive text-to-image models, Liquid, Emu3-Gen, and Janus-Pro, covering the erasure of explicit content, graphic violence, and branded imagery. Obliviate consistently outperforms current alternatives, reducing nudity on the defensive RAB benchmark from 91.58 to 3.15 while preserving overall model utility.

Hossein Shakibania, Jonas Henry Grebe, Tobias Braun, Ege Aktemur, Saleh Aslani, Mehmet G\"orkem Yi\u{g}it, Marcus Rohrbach• 2026

Related benchmarks

TaskDatasetResultRank
Nudity ErasureI2P--
52
Nudity ErasureRAB
CDR1.05
16
Image GenerationGeneral Utility Evaluation
FID12.31
16
Concept ErasureGory Content T2I-RP
CDR77.83
16
Artistic Style ErasureVan Gogh 500 Qwen-generated prompts (test)
CLIPE Score22.87
16
Nudity ErasureT2I-RP
CDR18.11
16
Nudity ErasureMMA-Diff
CDR2.8
16
Concept ErasureAugmented Unbranding Coca-Cola (test)
CDR5.22
16
Showing 8 of 8 rows

Other info

Follow for update