Evaluations of Multi-Step Traffic Flow Prediction Models Based on Graph Neural Networks

Yaopeng Xie,Chengjie Jin
DOI: https://doi.org/10.1109/cisce62493.2024.10653257
2024-01-01
Abstract:Accurate prediction of traffic flow is crucial for alleviating traffic congestion, optimizing resource allocation and enhancing the efficiency of transport systems. In recent years, graph neural networks (GNNs) have been widely used in the field of traffic flow prediction due to their advantages in processing graph-structured data. This study provides an in-depth comparative analysis of the performance of several representative GNN models in a multi-step traffic flow prediction task. Through a unified training system, the experiments on real-world traffic flow datasets (PEMS 04 and 08) reveal the performance differences of five typical models: STGCN, DCRNN, GWNET, AGCRN and MTGNN. The results show that although the compared five models all exhibit excellent performance compared to traditional methods (HA, ARIMA and LSTM), there are significant differences in their efficiencies under specific conditions. The speed of STGCN, AGCRN and MTGNN are much faster than that of DCRNN and GWNET. Finally, how to further improve these models is also discussed with the consideration of node-specific prediction tasks.
What problem does this paper attempt to address?