Revisiting Data-Free Knowledge Distillation with Poisoned Teachers

Junyuan Hong, Yi Zeng, Shuyang Yu, Lingjuan Lyu, Ruoxi Jia +1 more
6/4/2023
cs.LGcs.CR

Official code: illidanlab/ABD (24 stars, MIT, last commit 2024-07-07).

Abstract

Data-free knowledge distillation (KD) helps transfer knowledge from a pre-trained model (known as the teacher model) to a smaller model (known as the student model) without access to the original training data used for training the teacher model. However, the security of the synthetic or out-of-distribution (OOD) data required in data-free KD is largely unknown and under-explored. In this work, we make the first effort to uncover the security risk of data-free KD w.r.t. untrusted pre-trained models. We then propose Anti-Backdoor Data-Free KD (ABD), the first plug-in defensive method for data-free KD methods to mitigate the chance of potential backdoors being transferred. We empirically evaluate the effectiveness of our proposed ABD in diminishing transferred backdoor knowledge while maintaining compatible downstream performances as the vanilla KD. We envision this work as a milestone for alarming and mitigating the potential backdoors in data-free KD. Codes are released at https://github.com/illidanlab/ABD.

View on arXivView PDF

Code Implementations(1)

illidanlab/ABDOfficial100%
MIT

[ICML2023] Revisiting Data-Free Knowledge Distillation with Poisoned Teachers

241PythonMay 31, 20232 years agoMIT
backdoor-attacksdata-freedistillation-modeltransfer-learning

Similar Papers

Cite this paper

@article{hong2023revisiting,
  title  = {Revisiting Data-Free Knowledge Distillation with Poisoned Teachers},
  author = {Junyuan Hong and Yi Zeng and Shuyang Yu and Lingjuan Lyu and Ruoxi Jia and Jiayu Zhou},
  year   = {2023},
  eprint = {2306.02368},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2306.02368v1},
  journal = {ICML 2023 2023}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2306.02368.svg)](https://codeofpaper.com/paper/2306.02368)

Discussion