arXiv:2507.04119cs.LGcs.AI2025-07ICML被引 6

对抗非可迁移教师,让无数据蒸馏摆脱分布外陷阱

When Data-Free Knowledge Distillation Meets Non-Transferable Teacher: Escaping Out-of-Distribution Trap is All You Need

  • 用生成器合成数据,通过鲁棒性区分真假样本
  • 识别并过滤分布外伪造样本,提升知识迁移效果
  • 适合关注模型安全与蒸馏鲁棒性的研究者

数据无监督知识蒸馏(DFKD)在不访问真实分布内(ID)数据的情况下,将知识从教师模型转移到学生模型。现有方法通常假设教师可信,但对不可迁移学习(NTL)教师的鲁棒性与安全性未充分探索。本文首次研究使用非可迁移教师进行DFKD,其中从分布内到分布外(OOD)的知识迁移被禁止。我们发现,此类教师会诱导生成器将注意力从有用的分布内知识转移到误导性的分布外知识,阻碍分布内知识传递而强化分布外知识转移。为此,我们提出对抗陷阱逃脱(ATEsc)策略,通过分析生成样本的鲁棒性差异:脆弱样本视为分布内样例用于正常蒸馏,稳健样本视为分布外样例用于遗忘其知识。大量实验表明,该方法有效提升DFKD在面对非可迁移教师时的性能。代码已公开于https://github.com/tmllab/2025_ICML_ATEsc。

原文摘要 · Abstract (English)

Data-free knowledge distillation (DFKD) transfers knowledge from a teacher to a student without access the real in-distribution (ID) data. Its common solution is to use a generator to synthesize fake data and use them as a substitute for real ID data. However, existing works typically assume teachers are trustworthy, leaving the robustness and security of DFKD from untrusted teachers largely unexplored. In this work, we conduct the first investigation into distilling non-transferable learning (NTL) teachers using DFKD, where the transferability from an ID domain to an out-of-distribution (OOD) domain is prohibited. We find that NTL teachers fool DFKD through divert the generator's attention from the useful ID knowledge to the misleading OOD knowledge. This hinders ID knowledge transfer but prioritizes OOD knowledge transfer. To mitigate this issue, we propose Adversarial Trap Escaping (ATEsc) to benefit DFKD by identifying and filtering out OOD-like synthetic samples. Specifically, inspired by the evidence that NTL teachers show stronger adversarial robustness on OOD samples than ID samples, we split synthetic samples into two groups according to their robustness. The fragile group is treated as ID-like data and used for normal knowledge distillation, while the robust group is seen as OOD-like data and utilized for forgetting OOD knowledge. Extensive experiments demonstrate the effectiveness of ATEsc for improving DFKD against NTL teachers. Code is released at https://github.com/tmllab/2025_ICML_ATEsc.

知识蒸馏无数据训练模型安全对抗鲁棒性

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