arXiv cs.AI· Yang Xiao, Kaiyuan Deng, Gen Li, Zinan Ling, Zhiwen Fan, Xiaolong Ma, Bo Hui·· 3 天前AI 评分18
扩散模型组合概念擦除:DMU 通过域流形遗忘解耦物体与风格
Compositional Concepts Erasing: Disentangling Object-Style via Domain Manifold Unlearning
AI 导读
针对扩散模型难以擦除"物体-风格"组合概念且多次遗忘请求易致灾难性遗忘的问题,研究者提出域流形遗忘(DMU),利用从扩散中间特征提取的 EMA 更新特征代理表示待保留的目标物体方向,实现对特定物体中域相关组件的选择性遗忘,同时保留其他语义表示。实验显示 DMU 显著缓解灾难性遗忘,在批量遗忘与顺序遗忘上均优于现有 SOTA 方法。
正文 · 原文
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.
| Comments: | 14 pages |
| Subjects: | Computer Vision and Pattern Recognition (cs.CV); Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2610.14514 [cs.CV] |
| (or arXiv:2610.14514v1 [cs.CV] for this version) | |
| https://doi.org/10.48550/arXiv.2610.14514 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Yang Xiao [view email]
[v1]
Sat, 10 Oct 2026 03:22:31 UTC (23,478 KB)
来源:arXiv cs.AI · arxiv.org