提出新方法解决模型遗忘中的过度遗忘和知识复活问题
Unlearning's Blind Spots: Over-Unlearning and Prototypical Relearning Attack
- 设计量化指标检测遗忘区域附近的过度遗忘
- 发现仅用少量样本即可恢复遗忘知识的攻击漏洞
- 通过可插拔模块同时防御过度遗忘与知识复活
机器遗忘(MU)旨在不重新训练的情况下移除指定数据,但现有方法忽视了两个关键盲点:过度遗忘导致遗忘区域邻近数据性能下降,以及事后重学攻击可轻易恢复被遗忘的知识。针对类别级遗忘,我们首次提出量化指标OU@epsilon,衡量遗忘类附近区域的附带损害。进一步揭示一种新型重学威胁——原型重学攻击,仅需少数样本即可利用遗忘类的类别原型重建原始性能。为应对两类问题,我们提出Spotter,一个可插拔目标函数:(i) 在遗忘类附近区域施加掩码知识蒸馏惩罚以抑制OU@epsilon;(ii) 引入类内分散损失打散遗忘类嵌入,抵消原型重学攻击。Spotter在CIFAR、TinyImageNet和CASIA-WebFace上均达到当前最佳效果,为遗忘盲点提供实用解决方案。
原文摘要 · Abstract (English)
Machine unlearning (MU) aims to expunge a designated forget set from a trained model without costly retraining, yet the existing techniques overlook two critical blind spots: "over-unlearning" that deteriorates retained data near the forget set, and post-hoc "relearning" attacks that aim to resurrect the forgotten knowledge. Focusing on class-level unlearning, we first derive an over-unlearning metric, OU@epsilon, which quantifies collateral damage in regions proximal to the forget set, where over-unlearning mainly occurs. Next, we expose an unforeseen relearning threat on MU, i.e., the Prototypical Relearning Attack, which exploits the per-class prototype of the forget class with just a few samples, and easily restores the pre-unlearning performance. To counter both blind spots in class-level unlearning, we introduce Spotter, a plug-and-play objective that combines (i) a masked knowledge-distillation penalty on the nearby region of forget classes to suppress OU@epsilon, and (ii) an intra-class dispersion loss that scatters forget-class embeddings, neutralizing Prototypical Relearning Attacks. Spotter achieves state-of-the-art results across CIFAR, TinyImageNet, and CASIA-WebFace datasets, offering a practical remedy to unlearning's blind spots.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。