Communication-Efficient and Secure Federated Learning Based on Adaptive One-Bit Compressed Sensing

Di Xiao,Xue Tan,Min Li
DOI: https://doi.org/10.1007/978-3-031-22390-7_29
2022-01-01
Abstract:Federated learning (FL) enables multiple clients to jointly train a global model without exposing local private datasets. The main challenges for FL include communication efficiency, privacy and robustness. In order to reduce the transmission cost of FL, there are currently some attempts to apply compressed sensing (CS) and quantization strategies as well as a combination of both. Insufficiently, they ignore privacy and robustness. Differential privacy (DP) can guarantee privacy by adding noise to gradient updates, but the convergence or performance of the model will be reduced. In order to overcome the shortcomings of existing methods, we propose the CS-DP-SignSGD algorithm to address the above challenges. Specifically, the model updates are compressed and quantized, which greatly reduces the transmission cost. The one-bit quantization strategy SignSGD based on the majority voting mechanism, which has natural Byzantine robustness, is used. Since CS reduces the model size, CS-DP-SignSGD adds Gaussian noise to the compressed model, which can reduce the amount of added noise and improve model quality without losing privacy. More notably, we propose an adaptive threshold selection scheme to sparsely represent gradients in the CS process, which achieves a certain degree of protection for the sparse representation strategy. Finally, we carry out privacy and convergence analyses of the algorithm. Experiments on the MNIST and Fashion-MNIST datasets demonstrate the effectiveness of our proposed method.
What problem does this paper attempt to address?