vix.ing · top · new · best · stats

Erasing Concepts from Diffusion Models

2023/03/13 by Rohit Gandikota, Gandikota, Rohit, Joanna Materzynska +6 · 1 voice · 119 citations
Computer Science · Social Sciences · #Computer Vision and Pattern Recognition (cs.CV) #Digital Media Forensic Detection #FOS: Computer and information sciences #Generative Adversarial Networks and Image Synthesis #Law in Society and Culture #cs.CV

paper · pdf · doi:10.48550/arxiv.2303.07345

openalex publication_date 2023/03/13 · arxiv published 2023/03/13 · arxiv updated 2023/06/21 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Motivated by recent advancements in text-to-image diffusion, we study erasure of specific concepts from the model's weights. While Stable Diffusion has shown promise in producing explicit or realistic artwork, it has raised concerns regarding its potential for misuse. We propose a fine-tuning method that can erase a visual concept from a pre-trained diffusion model, given only the name of the style and using negative guidance as a teacher. We benchmark our method against previous approaches that remove sexually explicit content and demonstrate its effectiveness, performing on par with Safe Latent Diffusion and censored training. To evaluate artistic style removal, we conduct experiments erasing five modern artists from the network and conduct a user study to assess the human perception of the removed styles. Unlike previous methods, our approach can remove concepts from a diffusion model permanently rather than modifying the output at the inference time, so it cannot be circumvented even if a user has access to model weights. Our code, data, and results are available at https://erasing.baulab.info/

Cited by

Discussions

Related