Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SPEED: Scalable, Precise, and Efficient Concept Erasure for Diffusion Models

About

Erasing concepts from large-scale text-to-image (T2I) diffusion models has become increasingly crucial due to the growing concerns over copyright infringement, offensive content, and privacy violations. In scalable applications, fine-tuning-based methods are time-consuming to precisely erase multiple target concepts, while real-time editing-based methods often degrade the generation quality of non-target concepts due to conflicting optimization objectives. To address this dilemma, we introduce SPEED, an efficient concept erasure approach that directly edits model parameters. SPEED searches for a null space, a model editing space where parameter updates do not affect non-target concepts, to achieve scalable and precise erasure. To facilitate accurate null space optimization, we incorporate three complementary strategies: Influence-based Prior Filtering (IPF) to selectively retain the most affected non-target concepts, Directed Prior Augmentation (DPA) to enrich the filtered retain set with semantically consistent variations, and Invariant Equality Constraints (IEC) to preserve key invariants during the T2I generation process. Extensive evaluations across multiple concept erasure tasks demonstrate that SPEED consistently outperforms existing methods in non-target preservation while achieving efficient and high-fidelity concept erasure, successfully erasing 100 concepts within only 5 seconds. Our code and models are available at: https://github.com/Ouxiang-Li/SPEED.

Ouxiang Li, Yuan Wang, Xinting Hu, Houcheng Jiang, Yanbin Hao, Fuli Feng• 2025

Related benchmarks

TaskDatasetResultRank
Art Style ErasureArtist style and content prompts 5 groups SD v1.4 based (test)
CS Style31.402
40
Concept ErasureVan Gogh style
FID19.18
39
Style ErasureMonet
Contrastive Similarity (CS)28.94
28
Style ErasureMS-COCO
CS Score26.55
28
Style ErasurePicasso
Contrastive Similarity (CS)27.96
28
Style ErasurePaul Gauguin
CS29.75
28
Style ErasureCaravaggio
CS26.24
28
Art Style ErasureArtist style and content prompt groups v1.4 (test)
Lgene76.7
16
Concept ErasureConcept prompts (Snoopy, Mickey, Spongebob, Pikachu, Hello Kitty) and MS-COCO
Snoopy Concept Score (CS)23.69
15
Concept ErasureMMA
ASR (%)24
14
Showing 10 of 46 rows

Other info

Follow for update