跳到正文
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 方法。

正文 · 原文

View PDF HTML (experimental)

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