Joint Client Selection and Model Compression for Efficient FL in UAV-Assisted Wireless Networks
Luo Chen, Ruyan Wang, Yaping Cui, Peng He, Ang Duan
Abstract
Deploying federated learning (FL) applications in unmanned aerial vehicle (UAV)-assisted wireless networks can enable ground terminals (GTs) to perform complex machine learning tasks with their own data. However, the FL is inefficient in practice due to the massive model parameters and device heterogeneity. In this paper, we propose a joint client selection and model compression scheme for FL (csmcFL) to improve training efficiency. Specifically, the average throughput of users is first improved by optimizing the UAV deployment location based on user communication fairness. Then, a low-rank decomposition of the fully connected layer in the CNN is performed to compress the model parameters, and partial devices are screened to implement model compression through the client selection strategy to alleviate the excessive aggregation time due to device heterogeneity. We perform extensive simulation experiments in different data distribution scenarios, and the experimental results show that the proposed scheme significantly reduces the data volume of the transmitted model while achieving higher model accuracy compared to the baseline scheme.