arXiv:2506.06278cs.LGcs.AI2025-06NeurIPS被引 14

用知识蒸馏让大模型删数据更牢靠,防微调反扑。

Distillation Robustifies Unlearning

  • 用未学习过数据的模型输出来训练新模型,保留行为但丢掉底层能力。
  • 在合成语言和算术任务上,仅用60%-80%算力就达到重训模型的鲁棒性。
  • 适合需要安全去标识、防数据回溯的工业级大模型应用。

当前大模型遗忘方法缺乏鲁棒性,仅需少量微调即可逆转效果。我们首先证明,即使在理想化的遗忘场景下——训练模型模仿从未接触过敏感信息的模型——其输入输出行为仍可被显著改变,而底层能力保持不变。基于此动态,我们提出核心结论:从已遗忘模型的输出中蒸馏训练一个随机初始化的学生模型,可转移行为但保留潜在能力,即蒸馏能增强遗忘的鲁棒性。据此提出UNDO(Unlearn-Noise-Distill-on-Outputs)方法,将遗忘后的模型与其加噪副本进行蒸馏。该方法在合成语言与算术任务上建立新的计算成本与鲁棒性权衡前沿,最强设置下仅需60-80%算力、0.01%预训练数据标注,即可达到完全数据过滤后重训模型的鲁棒性。在更真实的武器扩散代理(WMDP)基准上也验证了其有效性。由于蒸馏广泛应用于实践,先执行遗忘再蒸馏为实现可靠能力移除提供了便捷路径。

原文摘要 · Abstract (English)

Current LLM unlearning methods are not robust. A few steps of finetuning can revert their effects. We begin by showing that this is true even for an idealized form of unlearning: training to imitate a model that was never trained on unwanted information. This shows that training a model can drastically modify its input-output behavior while leaving its underlying capabilities intact. In light of this dynamic, we show our main result. Training a randomly initialized student on the outputs of an unlearned model transfers behaviors while leaving latent capabilities behind. In short, distillation robustifies unlearning. Based on this result, we propose Unlearn-Noise-Distill-on-Outputs (UNDO), a scalable method that distills an unlearned model into a noised copy of itself. UNDO introduces a tunable tradeoff between compute cost and robustness, establishing a new Pareto frontier on synthetic language and arithmetic tasks. At its strongest setting, UNDO matches the robustness of a model retrained from scratch with perfect data filtering while using only 60-80% of the compute and requiring only 0.01% of the pretraining data to be labeled. We also show that UNDO robustifies unlearning on the more realistic Weapons of Mass Destruction Proxy (WMDP) benchmark. Since distillation is widely used in practice, incorporating an unlearning step beforehand offers a convenient path to robust capability removal.

大模型遗忘知识蒸馏鲁棒性数据安全

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。