No Concept Escapes the Audit: Auditing-Aware Unlearning for Verifiable Concept Erasure in Diffusion Models
cs.LG, cs.AI
Submitted: 2026-10-04
Updated: 2026-10-11
Code: https://github.com/fastai/imagenette
License: http://creativecommons.org/licenses/by/4.0/
The gist: Text-to-image diffusion models can generate prohibited content, which motivates concept erasure through machine unlearning.
Terminology
Abstract
Text-to-image diffusion models can generate prohibited content, which motivates concept erasure through machine unlearning. Most erasure methods intervene at the text interface, through prompt modification or localized updates to text-conditioning weights, and they are evaluated by what the model outputs for given prompts. Such evaluation cannot see what the network still encodes. Latent-space auditing, which bypasses text conditioning and probes the denoising network directly, shows that erased concepts remain recoverable from internal representations. We find that this also holds for methods built to be robust against adversarial prompts, and that the problem grows with the number of erased concepts. We propose Auditing-Aware Unlearning for Verifiable Concept Erasure in Diffusion Models (AVCE), a framework that grounds erasure in the model's latent representations. AVCE audits the embedding neighborhood of each concept and condenses the discovered vulnerable directions into an anchor at the weakest geometric point. It edits cross-attention and self-attention projections in closed form at this anchor, then fine-tunes the two pathways with pathway-level auditing losses, using orthogonal gradient projection to consolidate multiple concepts. Experiments on SD v1.5, SDXL, and Flux 1.0 across object, explicit-content, and artistic-style unlearning show that AVCE reduces attack success rates by 5.07x and improves auditing scores by 3.84x over the strongest baseline, while preserving competitive generation quality.
Sources
- The Malicious Use of Artificial Intelligence: Forecasting, Prevention, and Mitigation
- Fantastic Targets for Concept Erasure in Diffusion Models and Where To Find Them
- ConceptPrune: Concept Editing in Diffusion Models via Skilled Neuron Pruning
- Prompting4Debugging: Red-Teaming Text-to-Image Diffusion Models by Finding Problematic Prompts
- Forget Many, Forget Right: Scalable and Precise Concept Unlearning in Diffusion Models
- SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation
- Certified Data Removal from Machine Learning Models
- Knowledge Unlearning for Mitigating Privacy Risks in Language Models
- Diffusion Models for Medical Image Analysis: A Comprehensive Survey
- One Image is Worth a Thousand Words: A Usability Preservable Text-Image Collaborative Erasing Framework
- AEGIS: Adversarial Target-Guided Retention-Data-Free Robust Concept Erasure from Diffusion Models
- Position: Bridge the Gaps between Machine Unlearning and AI Regulation
- A Survey of Machine Unlearning
- GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Models
- Circumventing Concept Erasure Methods For Text-to-Image Generative Models
- SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis
- Unlearning or Concealment? A Critical Analysis and Evaluation Metrics for Unlearning in Diffusion Models
- Ring-A-Bell! How Reliable are Concept Removal Methods for Diffusion Models?
- CE-SDWV: Effective and Efficient Concept Erasure for Text-to-Image Diffusion Models via a Semantic-Driven Word Vocabulary
- Machine Unlearning: A Comprehensive Survey
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks