Minimalist Concept Erasure in Generative Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Yang, Jin, Er, Dong, Yanfei, Wu, Yixuan, Torr, Philip, Khakzar, Ashkan, Stegmaier, Johannes, Kawaguchi, Kenji
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913948280815616
author Zhang, Yang
Jin, Er
Dong, Yanfei
Wu, Yixuan
Torr, Philip
Khakzar, Ashkan
Stegmaier, Johannes
Kawaguchi, Kenji
author_facet Zhang, Yang
Jin, Er
Dong, Yanfei
Wu, Yixuan
Torr, Philip
Khakzar, Ashkan
Stegmaier, Johannes
Kawaguchi, Kenji
contents Recent advances in generative models have demonstrated remarkable capabilities in producing high-quality images, but their reliance on large-scale unlabeled data has raised significant safety and copyright concerns. Efforts to address these issues by erasing unwanted concepts have shown promise. However, many existing erasure methods involve excessive modifications that compromise the overall utility of the model. In this work, we address these issues by formulating a novel minimalist concept erasure objective based \emph{only} on the distributional distance of final generation outputs. Building on our formulation, we derive a tractable loss for differentiable optimization that leverages backpropagation through all generation steps in an end-to-end manner. We also conduct extensive analysis to show theoretical connections with other models and methods. To improve the robustness of the erasure, we incorporate neuron masking as an alternative to model fine-tuning. Empirical evaluations on state-of-the-art flow-matching models demonstrate that our method robustly erases concepts without degrading overall model performance, paving the way for safer and more responsible generative models.
format Preprint
id arxiv_https___arxiv_org_abs_2507_13386
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Minimalist Concept Erasure in Generative Models
Zhang, Yang
Jin, Er
Dong, Yanfei
Wu, Yixuan
Torr, Philip
Khakzar, Ashkan
Stegmaier, Johannes
Kawaguchi, Kenji
Computer Vision and Pattern Recognition
Machine Learning
Recent advances in generative models have demonstrated remarkable capabilities in producing high-quality images, but their reliance on large-scale unlabeled data has raised significant safety and copyright concerns. Efforts to address these issues by erasing unwanted concepts have shown promise. However, many existing erasure methods involve excessive modifications that compromise the overall utility of the model. In this work, we address these issues by formulating a novel minimalist concept erasure objective based \emph{only} on the distributional distance of final generation outputs. Building on our formulation, we derive a tractable loss for differentiable optimization that leverages backpropagation through all generation steps in an end-to-end manner. We also conduct extensive analysis to show theoretical connections with other models and methods. To improve the robustness of the erasure, we incorporate neuron masking as an alternative to model fine-tuning. Empirical evaluations on state-of-the-art flow-matching models demonstrate that our method robustly erases concepts without degrading overall model performance, paving the way for safer and more responsible generative models.
title Minimalist Concept Erasure in Generative Models
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2507.13386