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

ScaleErasure: Inference-Time Minimal Intervention for Precise Concept Erasure in Next-Scale Autoregressive Image Generation

About

Concept erasure aims to prevent image generative models from producing unsafe content while preserving their general generative capability. Meanwhile, next-scale autoregressive (AR) image generation has recently emerged as a new generative paradigm characterized by next-scale prediction, for which concept erasure remains largely unexplored. In this paradigm, semantic information is highly compressed at early scales, leading to severe entanglement between unsafe and unrelated semantics. In this paper, we propose ScaleErasure, an inference-time concept erasure method that performs minimal intervention. ScaleErasure precisely selects and guides predicted logits that are most relevant to the unsafe concept, thereby enabling effective erasure under severe semantic entanglement. Specifically, ScaleErasure performs two additional forward passes conditioned on the unsafe concept and the corresponding safe concept, and leverages their outputs to guide the target logits away from unsafe concepts toward safe concepts. To enable precise and minimal intervention, logits selection and guidance are conducted across three dimensions: scales, tokens, and bit channels. Experiments demonstrate that ScaleErasure outperforms adapted baselines in the next-scale AR paradigm, achieving more precise concept erasure while largely preserving general generative capability. The code is available at https://github.com/coziiizz/ScaleErasure.

Cong Wang, Haiyu Wu, Zhiwei Jiang, Zifeng Cheng, Fei Shen, Yafeng Yin, Qing Gu• 2026

Related benchmarks

TaskDatasetResultRank
Nudity ErasureI2P
Total Count182
52
Concept ErasureBroader Unsafe Categories
Hate Residual Percentage14.29
7
Concept ErasureUser Study 20 image groups
Erasure Effectiveness876
7
Concept ErasureCopyrighted Characters
Snoopy Erasure Rate10.68
7
Copyrighted Concept ErasureSMP
Pikachu CLIP-E22.48
7
Concept ErasureArtistic Styles
Rembrandt Score8.23
7
Concept ErasureI2P RAB attack subset
Female Detection Count6
4
Concept ErasureI2P UD attack
Female Detection Count11
4
Showing 8 of 8 rows

Other info

Follow for update