Prediction of future traffic load is a crucial task to support the automatic Operations, Administration, and Management (OAM) of communication networks. Existing Machine Learning (ML) models require big data to accomplish this task. However, large data sets are not always available, due to the limited storage capacity and the high storage cost at Base Stations (BSs). To solve the problem, we leverage the spatial-temporal correlation among different BSs, which allows other BSs’ data to be used for the prediction of the target BS. One major challenge in realizing this idea is the imbalance of data amounts between neighbor BSs and a prediction target BS. If one simply aggregates the data from both neighbors and target, the target’s traffic features would be overwhelmed by the neighbors’ data. To address this challenge, we propose a Spatial-Temporal Transfer (STT) framework, which trains a base model with an aggregated data set from multiple BSs, and then carefully refines the base model to serve a target BS. To strike a perfect balance between general tendency and individual features, STT adopts an advanced transfer learning technique that exploits regularization on model parameters. Experiments show the efficiency of the proposed STT framework.

https://ieeexplore.ieee.org/document/9838413