Compositional Concepts Erasing: Disentangling Object-Style via Domain Manifold Unlearning
cs.CV, cs.AI
Submitted: 2026-10-10
Updated: 2026-10-10
Comments: 14 pages
License: http://creativecommons.org/licenses/by/4.0/
The gist: Diffusion models pose significant risks regarding the generation of illicit content and copyright violations.
Abstract
Diffusion models pose significant risks regarding the generation of illicit content and copyright violations. This has catalyzed the development of machine unlearning to erase harmful concepts or copyrighted styles from a model's generative repertoire without compromising its general performance. However, previous unlearning approaches on diffusion models have largely focused on either object-level or style-level removal and failed to address the challenge of object-style combination unlearning. At the same time, multiple unlearning requests often lead to catastrophic forgetting. To address these challenges, we propose Domain Manifold Unlearning (DMU), which uses an EMA-updated feature proxy extracted from intermediate diffusion features to represent the retained target-object direction. As a lightweight estimating representation of an object, DMU enables selective unlearning of domain-specific components in a specific object while preserving other semantic representations, thereby significantly enhancing robustness in multiple combination unlearning scenarios. Extensive experiments show that DMU significantly mitigates catastrophic forgetting, outperforming state-of-the-art methods of batch unlearning and sequential unlearning.
Sources
- Machine Unlearning Doesn't Do What You Think: Lessons for Generative AI Policy and Research
- Sculpting Memory: Multi-Concept Forgetting in Diffusion Models via Dynamic Mask and Concept-Aware Optimization
- Forget Many, Forget Right: Scalable and Precise Concept Unlearning in Diffusion Models
- Approximate Domain Unlearning for Vision-Language Models
- UnlearnCanvas: Stylized Image Dataset for Enhanced Machine Unlearning Evaluation in Diffusion Models
- Continual Unlearning for Text-to-Image Diffusion Models: A Regularization Perspective
- Continual Unlearning for Foundational Text-to-Image Models without Generalization Erosion
- Forget-It-All: Multi-Concept Machine Unlearning via Concept-Aware Neuron Masking
- No Concept Escapes the Audit: Auditing-Aware Unlearning for Verifiable Concept Erasure in Diffusion Models
- Your Unlearning Gives You Away: Identifying Erased Concepts in Diffusion Models
- Diffusion-Based Cross-Modal Feature Extraction for Multi-Label Classification
- SparseDiT: Token Sparsification for Efficient Diffusion Transformer
- V-Warper: Appearance-Consistent Video Diffusion Personalization via Value Warping
- SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation
- Separable Multi-Concept Erasure from Diffusion Models
- FADE: Frame-Aware Diffusion-Transformer-based Multi-Concept Erasure for Video Unlearning
- LLM Unlearning Reveals a Stronger-Than-Expected Coreset Effect in Current Benchmarks
- The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearning
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models