Federated Domain Generalization for Network Traffic Prediction via Spatial-Temporal Feature Learning

Shaoyin Chen, Yang Yang, Jingting Mei, Zhipeng Gao, Lanlan Rui, Peng Yu

Abstract: Network traffic prediction is crucial for network operation and management, forming the basis for utilizing big data in decision support. Traditional deep learning methods require extensive, assumed independently and identically distributed(IID) data. However, with IoT development, privacy protection gains importance, resulting in distributed data collection with varying distributions. This leads to a significant performance drop when applying a well-trained model to a new dataset, causing domain shift. To tackle domain shift from inconsistent data distributions and meet privacy protection needs, this paper proposes a spatial-temporal feature learning method within the federated domain generalization framework for network traffic prediction. The ultimately trained model effectively generalizes to an unseen domain, as confirmed by experimental results.