Cellular Traffic Prediction via Byzantine-robust Asynchronous Federated Learning

Abstract

Network traffic prediction plays a crucial role in intelligent network operation. Traditional prediction methods often rely on centralized training, necessitating the transfer of vast amounts of traffic data to a central server. This approach can lead to latency and privacy concerns. To address these issues, federated learning integrated with differential privacy has emerged as a solution to improve data privacy and model robustness in distributed settings. Nonetheless, existing federated learning protocols are vulnerable to Byzantine attacks, which may significantly compromise model robustness. Developing a robust and privacy-preserving prediction model in the presence of Byzantine clients remains a significant challenge. To this end, we propose an asynchronous differential federated learning framework based on distributionally robust optimization. The proposed framework utilizes multiple clients to train the prediction model collaboratively with local differential privacy. In addition, regularization techniques have been employed to further improve the Byzantine robustness of the models. We have conducted extensive experiments on three real-world datasets, and the results elucidate that our proposed distributed algorithm can achieve superior performance over existing methods.

References in corpus (8)

• Privacy-preserving Traffic Flow Prediction: A Federated Learning Approach
• Dual Attention-Based Federated Learning for Wireless Traffic Prediction
• Active Learning for Wireless IoT Intrusion Detection
• GRNN: Generative Regression Neural Network -- A Data Leakage Attack for Federated Learning
• Learning Private Neural Language Modeling with Attentive Aggregation
• An Experimental Study of Byzantine-Robust Aggregation Schemes in Federated Learning
• Unsupervised Deep Learning for IoT Time Series
• Anomaly Detection in Event-triggered Traffic Time Series via Similarity Learning