FedAGCN: A traffic flow prediction framework based on federated learning and Asynchronous Graph Convolutional Network

Tao Qi,Lingqiang Chen,Guanghui Li,Yijing Li,Chenshu Wang
DOI: https://doi.org/10.1016/j.asoc.2023.110175
2023-03-10
Applied Soft Computing Journal
Abstract:Accurate and real-time traffic flow prediction is an essential component of the Intelligent Transportation System (ITS). Balancing the prediction accuracy and time cost of prediction models is a challenging topic. This paper proposes a deep learning framework (FedAGCN) based on federated learning and asynchronous graph convolutional networks to predict traffic flow accurately in real time. FedAGCN applies asynchronous spatial–temporal graph convolution to model the spatial–temporal dependence in traffic data. In order to reduce the time cost of the deep learning model, we propose a graph federated learning strategy GraphFed to train the model. Experiments were conducted on two public traffic datasets, and the results showed that FedAGCN effectively reduced the training and inference time of the model while maintaining considerable prediction accuracy.
What problem does this paper attempt to address?