Pith. sign in

Erasing undesirable concepts in diffusion models with adversarial preservation

6 Pith papers cite this work. Polarity classification is still indexing.

6 Pith papers citing it
abstract

Diffusion models excel at generating visually striking content from text but can inadvertently produce undesirable or harmful content when trained on unfiltered internet data. A practical solution is to selectively removing target concepts from the model, but this may impact the remaining concepts. Prior approaches have tried to balance this by introducing a loss term to preserve neutral content or a regularization term to minimize changes in the model parameters, yet resolving this trade-off remains challenging. In this work, we propose to identify and preserving concepts most affected by parameter changes, termed as \textit{adversarial concepts}. This approach ensures stable erasure with minimal impact on the other concepts. We demonstrate the effectiveness of our method using the Stable Diffusion model, showing that it outperforms state-of-the-art erasure methods in eliminating unwanted content while maintaining the integrity of other unrelated elements. Our code is available at https://github.com/tuananhbui89/Erasing-Adversarial-Preservation.

citation-role summary

background 1

citation-polarity summary

years

2026 6

roles

background 1

polarities

background 1

representative citing papers

SHIFT: Steering Hidden Intermediates in Flow Transformers

cs.CV · 2026-04-10 · unverdicted · novelty 5.0

SHIFT learns and applies steering vectors to selected layers and timesteps in DiT models to suppress concepts, shift styles, or bias objects while keeping image quality and prompt adherence intact.

citing papers explorer

Showing 6 of 6 citing papers.