ScaleErasure: Inference-Time Minimal Intervention for Precise Concept Erasure in Next-Scale Autoregressive Image Generation
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 38 of its 42 references that are in the map. First posted on arXiv 28 Jun 2026.
Builds on
38 of 42 references.
References · 38
- 2017Attention Is All You Needcited by 59,932
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2022Classifier-Free Diffusion Guidancecited by 5,717
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,005
- 2025FLUX.1 Kontext: Flow Matching for In-Context Image Generation and Editing in Latent Spacecited by 960
- 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 760
- 2023MasaCtrl: Tuning-Free Mutual Self-Attention Control for Consistent Image Synthesis and Editingcited by 722
26 more
- 2024Autoregressive Image Generation without Vector Quantizationcited by 644
- 2023Erasing Concepts from Diffusion Modelscited by 530
- 2022Safe Latent Diffusion: Mitigating Inappropriate Degeneration in Diffusion Modelscited by 501
- 2023Unified Concept Editing in Diffusion Modelscited by 377
- 2023Ablating Concepts in Text-to-Image Diffusion Modelscited by 312
- 2024MACE: Mass Concept Erasure in Diffusion Modelscited by 298
- 2024Infinity: Scaling Bitwise AutoRegressive Modeling for High-Resolution Image Synthesiscited by 297
- 2023Ring-A-Bell! How Reliable are Concept Removal Methods for Diffusion Models?cited by 239
- 2022What the DAAM: Interpreting Stable Diffusion Using Cross Attentioncited by 233
- 2022Splicing ViT Features for Semantic Appearance Transfercited by 209
- 2023Selective Amnesia: A Continual Learning Approach to Forgetting in Deep Generative Modelscited by 175
- 2023One-Dimensional Adapter to Rule Them All: Concepts, Diffusion Models and Erasing Applicationscited by 164
- 2024Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Modelscited by 164
- 2024Randomized Autoregressive Visual Generationcited by 125
- 2024Reliable and Efficient Concept Erasure of Text-to-Image Diffusion Modelscited by 123
- 2024Image and Video Tokenization with Binary Spherical Quantizationcited by 110
- 2024EraseAnything: Enabling Concept Erasure in Rectified Flow Transformerscited by 93
- 2024Tuning Language Models by Proxycited by 88
- 2024Reversing the Forget-Retain Objectives: An Efficient LLM Unlearning Framework from Logit Differencecited by 85
- 2023Focus on Your Instruction: Fine-grained and Multi-instruction Image Editing by Attention Modulationcited by 61
- 2024GuardT2I: Defending Text-to-Image Models from Adversarial Promptscited by 61
- 2025Set You Straight: Auto-Steering Denoising Trajectories to Sidestep Unwanted Conceptscited by 61
- 2024R.A.C.E.: Robust Adversarial Concept Erasure for Secure Text-to-Image Diffusion Modelcited by 56
- 2025Training-Free Text-Guided Image Editing with Visual Autoregressive Modelcited by 21
- 2023ORES: Open-vocabulary Responsible Visual Synthesiscited by 14
- 2024On Giant's Shoulders: Effortless Weak to Strong by Dynamic Logits Fusioncited by 9
Built on by
Not cited yet.