摘要
arXiv:2504.09039v1 类型: cross
摘要: 文本到图像(T2I)扩散模型在从文本提示生成高质量图像方面取得了显著成功。然而,在需要选择性遗忘的场景中,如移除版权内容、减少偏见或消除有害概念,它们储存大量知识的能力引发了担忧。尽管现有的遗忘方法可以移除某些概念,但在多概念遗忘方面,由于不稳定性、残留知识的持久性以及生成质量的下降,它们存在困难。为了解决这些挑战,我们提出了一个名为“动态掩码结合概念意识损失”的新颖遗忘框架,该框架适用于扩散模型中的多概念遗忘。我们的“动态掩码”机制根据当前的优化状态动态更新梯度掩码,允许选择性权重修改,从而防止与其他无关知识的干扰。此外,我们的“概念意识损失”明确了遗忘过程,通过 superclass 衔接实现语义一致性,而基于知识蒸馏的正则化损失确保在顺序遗忘过程中之前遗忘的概念仍然被遗忘。我们进行了广泛实验来评估我们的方式。结果表明,我们的方法在遗忘效果、输出保真度和语义连贯性方面优于现有的遗忘技术,特别是在多概念场景中。我们的工作提供了一个针对生成模型的原理性和灵活性强的稳定和高保真遗忘框架。代码将公开发布。