A New Data-Free Backdoor Removal Method via Adversarial Self-Knowledge Distillation
In the context of Internet of Things edge devices, pretrained models are often sourced directly from cloud computing platforms due to the unavailability of training data. This lack of access during the training phase makes these models susceptible to backdoor attacks. To address this challenge, we introduce a novel data-free backdoor removal method that operates effectively even when only the poisoned model is accessible. Our innovative approach employs two end-to-end generators with identical architectures to create both clean and poisoned samples. These samples are crucial for transferring knowledge from the teacher model—the fixed poisoned model—to the student model, which is initialized with the poisoned model. Our method utilizes a channel shuffling technique during the distillation process to disrupt and eliminate the backdoor knowledge embedded in the teacher model. This process involves iterative updates of the generators and meticulous distillation of the student model, leading to efficient backdoor removal. We conducted extensive experiments on five sophisticated backdoor attacks across two benchmark datasets. The results demonstrate that our method not only significantly bolsters the model’s resistance to backdoor attacks but also maintains high recognition accuracy for clean samples, thereby outperforming existing methods. Additionally, the code for our method is available at <uri xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">https://github.com/gaoyafeiyoo/ADBR</uri>.
Authors
- Minglin Liu (ORCID: https://orcid.org/0000-0002-8477-0090)
- Xuexiang Li (ORCID: https://orcid.org/0000-0002-9309-8259)
- Celimuge Wu (ORCID: https://orcid.org/0000-0001-6853-5878)
- Xianfu Chen (ORCID: https://orcid.org/0000-0002-9453-4200)
- Jie Li (ORCID: https://orcid.org/0000-0002-4974-6116)
- Xu Zhou (ORCID: https://orcid.org/0000-0003-3418-1447)
- Yafei Gao
Institutions
- Michigan Technological University (US)
- Shanghai Jiao Tong University (CN)
- University of Electro-Communications (JP)
- Zhengzhou University (CN)
- Cloud Computing Center (CN)
Publication Details
- Journal
- IEEE Internet of Things Journal
- Published
- 2024-12-19
- DOI
- https://doi.org/10.1109/jiot.2024.3520642
- Primary Topic
- Adversarial Robustness in Machine Learning
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China