论文

扩散遗忘中共现的相关保留概念

Co-occurring Associated REtained concepts in Diffusion Unlearning

模型训练模型编辑与遗忘

摘要

忘却已成为减少传播模型中有害内容生成的关键技术。然而,现有的方法常常不仅删除目标概念,而且删除良性的共现概念。如图 1 所示,忘记裸体会无意中抑制人的概念,从而阻止模型生成人的图像。我们将这些不希望被抑制的共现概念定义为必须保留的 CARE(共现关联保留概念)。然后,我们引入了 CARE 分数,这是一个通用指标,可以直接量化它们在不学习任务中的保存情况。在此基础上,我们提出了ReCARE(CARE的鲁棒擦除),这是一个明确保护CARE同时仅擦除目标概念的框架。 ReCARE 自动构建 CARE 集,这是从目标图像中提取的良性共现标记的精选词汇表,并在训练过程中利用该词汇表实现稳定的忘却。跨各种目标概念(裸体、梵高风格和 Tench 物体)的广泛实验表明,ReCARE 在平衡稳健的概念擦除、整体实用性和 CARE 保存方面实现了整体最先进的性能。