提出DELETE方法,实现对任意类别知识的彻底遗忘且保留其余知识。
Decoupled Distillation to Erase: A General Unlearning Method for Any Class-centric Tasks
- 分离遗忘与保留项,通过掩码蒸馏同时优化两者
- 无需原始数据或干预,在多个任务上达顶尖效果
- 适用于人脸识别、后门防御等各类分类任务
本文提出DELETE方法,一种通用且强大的类别中心型任务遗忘机制。我们构建理论框架,将遗忘损失分解为遗忘与保留两项。分析指出,以往方法多隐式优化遗忘项而缺乏保留项监督,导致预训练模型分布失衡,难以充分保留剩余类知识。为此,我们引入“暗知识”改进保留项,设计掩码蒸馏遗忘方法:通过掩码分离遗忘与保留输出,同步优化二者,确保目标类别被彻底遗忘,同时有效保留其余类知识。该方法无需访问剩余数据或额外干预,已在多个基准测试中达到最先进性能。此外,DELETE具备通用性,可应用于人脸识别、后门防御、语义分割等下游任务并表现优异。
原文摘要 · Abstract (English)
In this work, we present DEcoupLEd Distillation To Erase (DELETE), a general and strong unlearning method for any class-centric tasks. To derive this, we first propose a theoretical framework to analyze the general form of unlearning loss and decompose it into forgetting and retention terms. Through the theoretical framework, we point out that a class of previous methods could be mainly formulated as a loss that implicitly optimizes the forgetting term while lacking supervision for the retention term, disturbing the distribution of pre-trained model and struggling to adequately preserve knowledge of the remaining classes. To address it, we refine the retention term using "dark knowledge" and propose a mask distillation unlearning method. By applying a mask to separate forgetting logits from retention logits, our approach optimizes both the forgetting and refined retention components simultaneously, retaining knowledge of the remaining classes while ensuring thorough forgetting of the target class. Without access to the remaining data or intervention (i.e., used in some works), we achieve state-of-the-art performance across various benchmarks. What's more, DELETE is a general solution that can be applied to various downstream tasks, including face recognition, backdoor defense, and semantic segmentation with great performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。