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

EAR: Erasing Concepts from Unified Autoregressive Models

About

Autoregressive (AR) models have achieved unified and strong performance across both visual understanding and image generation tasks. However, removing undesired concepts from AR models while maintaining overall generation quality remains an open challenge. In this paper, we propose Erasure Autoregressive Model (EAR), a fine-tuning method for effective and utility-preserving concept erasure in AR models. Specifically, we introduce Windowed Gradient Accumulation (WGA) strategy to align patch-level decoding with erasure objectives, and Thresholded Loss Masking (TLM) strategy to protect content unrelated to the target concept during fine-tuning. Furthermore, we propose a novel benchmark, Erase Concept Generator and Visual Filter (ECGVF), aim at provide a more rigorous and comprehensive foundation for evaluating concept erasure in AR models. Specifically, we first employ structured templates across diverse large language models (LLMs) to pre-generate a large-scale corpus of target-replacement concept prompt pairs. Subsequently, we generate images from these prompts and subject them to rigorous filtering via a visual classifier to ensure concept fidelity and alignment. Extensive experimental results conducted on the ECGVF benchmark with the AR model Janus-Pro demonstrate that EAR achieves marked improvements in both erasure effectiveness and model utility preservation. Code is available at: https://github.com/immc-lab/ear/

Haipeng Fan, Shiyuan Zhang, Baohunesitu, Zihang Guo, Huaiwen Zhang• 2025

Related benchmarks

TaskDatasetResultRank
Nudity ErasureI2P--
52
Concept ErasureGory Content T2I-RP
CDR88.06
16
Nudity ErasureRAB
CDR11.58
16
Concept ErasureAugmented Unbranding Coca-Cola (test)
CDR85.61
16
Nudity ErasureT2I-RP
CDR28.33
16
Artistic Style ErasureVan Gogh 500 Qwen-generated prompts (test)
CLIPE Score22.59
16
Image GenerationGeneral Utility Evaluation
FID31.63
16
Nudity ErasureMMA-Diff
CDR0.8
16
Showing 8 of 8 rows

Other info

Follow for update