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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Nudity Erasure | I2P | Total Count182 | 52 | |
| Concept Erasure | Broader Unsafe Categories | Hate Residual Percentage14.29 | 7 | |
| Concept Erasure | User Study 20 image groups | Erasure Effectiveness876 | 7 | |
| Concept Erasure | Copyrighted Characters | Snoopy Erasure Rate10.68 | 7 | |
| Copyrighted Concept Erasure | SMP | Pikachu CLIP-E22.48 | 7 | |
| Concept Erasure | Artistic Styles | Rembrandt Score8.23 | 7 | |
| Concept Erasure | I2P RAB attack subset | Female Detection Count6 | 4 | |
| Concept Erasure | I2P UD attack | Female Detection Count11 | 4 |