Communication-Efficient Wireless Traffic Prediction with Federated Learning

• DOI: 10.3390/math12162539
• Publication Date: 2024-08-19T09:19:36Z
• Authors: Fuwei Gao; Chuanting Zhang; Jingping Qiao; Kaiqiang Li; Yi Cao
• Keywords: federated learning; deep neural networks; gradient compression; QA1-939; 0202 electrical engineering, electronic engineering, information engineering; 02 engineering and technology; spatial–temporal data analysis; Mathematics; wireless traffic prediction

Wireless traffic prediction is essential to developing intelligent communication networks that facilitate efficient resource allocation. Along this line, decentralized wireless traffic prediction under the paradigm of federated learning is becoming increasingly significant. Compared to traditional centralized learning, federated learning satisfies network operators’ requirements for sensitive data protection and reduces the consumption of network resources. In this paper, we propose a novel communication-efficient federated learning framework, named FedCE, by developing a gradient compression scheme and an adaptive aggregation strategy for wireless traffic prediction. FedCE achieves gradient compression through top-K sparsification and can largely relieve the communication burdens between local clients and the central server, making it communication-efficient. An adaptive aggregation strategy is designed by quantifying the different contributions of local models to the global model, making FedCE aware of spatial dependencies among various local clients. We validate the effectiveness of FedCE on two real-world datasets. The results demonstrate that FedCE can improve prediction accuracy by approximately 27% with only 20% of communications in the baseline method.