Open Access Paper
12 November 2024 Research on robustness of federated learning based on pruning optimization
Lei Zhao, Yitong Chen
Author Affiliations +
Proceedings Volume 13395, International Conference on Optics, Electronics, and Communication Engineering (OECE 2024) ; 133952G (2024) https://doi.org/10.1117/12.3049006
Event: International Conference on Optics, Electronics, and Communication Engineering, 2024, Wuhan, China
Abstract
Federated learning has more flexible data ownership participants, therefore its data (sample feature vector or label) is more likely to be changed, and it is more vulnerable to data poisoning by malicious users, resulting in the final global model not getting the expected effect. This paper focuses on this data poisoning defense problem and applies the traditional centralized machine learning pruning optimization method to each client of federated learning. Each client needs to execute before each global iteration. Pruning optimization algorithm to remove abnormal data. The experimental results indicate that when the discrepancy between abnormal and normal samples is significant, the pruning optimization algorithm effectively eliminates the outliers, thereby minimizing their impact on the final federated learning model.
(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.
Lei Zhao and Yitong Chen "Research on robustness of federated learning based on pruning optimization", Proc. SPIE 13395, International Conference on Optics, Electronics, and Communication Engineering (OECE 2024) , 133952G (12 November 2024); https://doi.org/10.1117/12.3049006
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Machine learning

Data modeling

Mathematical optimization

Education and training

Statistical modeling

Adversarial training

Defense and security

Back to Top