One-Shot Backdoor Removal for Federated Learning

Zijie Pan,Zuobin Ying,Yajie Wang,Chuan Zhang,Chunhai Li,Liehuang Zhu
DOI: https://doi.org/10.1109/jiot.2024.3438150
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:Federated learning is a distributed machine learning approach that enables multiple participants to collaboratively train a model without sharing their data, thus preserving privacy. However, the decentralized nature of federated learning also makes it susceptible to backdoor attacks, where malicious participants can embed hidden vulnerabilities within the model. Addressing these threats efficiently and effectively is crucial, especially given the impracticality of iterative and resource-intensive detection methods in federated learning environments. This paper presents a novel framework for one-shot backdoor removal in federated learning. Our approach integrates advanced anomaly detection techniques with a unique model update aggregation strategy, allowing for the identification and neutralization of backdoor influences in a single update cycle without the need for extensive data access or communication between participants. Extensive experiments across various federated architectures and data distributions demonstrate that our method effectively mitigates backdoor threats while maintaining model performance and scalability. This work not only enhances the security of federated models but also contributes to the broader applicability of federated learning in sensitive and critical domains.
What problem does this paper attempt to address?