Federated Learning Poisoning Attack Detection: Reconfiguration Algorithm TopK-FLcredit

Zhang Hong,Hongjiao Li
DOI: https://doi.org/10.1109/acie61839.2024.00009
2024-01-26
Abstract:Federated learning (FL) can train models between distributed clients without compromising the privacy of the training dataset, while the invisibility of the client dataset is highly susceptible to data poisoning attacks. For data poisoning attack in federated learning, a reconstruction algorithm called TopK-FLcredit is proposed for detecting federated learning poisoning attacks in Non-IID .The method uses the TopK algorithm to retain client-side key gradient information and reconstructs the abnormal client gradients to ensure normal client participation in the FL system while resisting poisoning attack. The accuracy of the proposed method was verified under different proportions of poisoning attacks using the MNIST dataset as an example. The experimental results show that the accuracy of the proposed reconstruction algorithm for poisoning attack detection is improved by 0.6%-6.9% compared to the poisoning attack detection algorithm Contra. Then, WTopK-FLcredit algorithm is further proposed, which uses dimensional difference calculation to replace single element difference calculation. Compared to the TopK-FLcredit algorithm, WTopK-FLcredit has a smaller computational complexity and the accuracy improvement of 0.65% to 6.1% over the Contra algorithm.
Computer Science
What problem does this paper attempt to address?