Multiscale Information Enhanced Spatial-Temporal Graph Convolutional Network for Multivariate Traffic Flow Forecasting Via Magnifying Perceptual Scope

Xinyu Zheng,Haidong Shao,Shen Yan,Yiming Xiao,Bin Liu
DOI: https://doi.org/10.1016/j.engappai.2024.109010
IF: 8
2024-01-01
Engineering Applications of Artificial Intelligence
Abstract:Graph Convolutional Networks (GCNs), which can model data in non-Euclidean space, have received extensive attention in multivariate traffic flow forecasting in recent years. However, most of the existing GCN forecasting methods rely on adjacency matrices defined by road network knowledge, which cannot accurately capture the inherent spatial-temporal dependencies of complex multivariate traffic flows. In this paper, a multiscale information enhanced spatial-temporal GCN is proposed, which utilizes an adaptive graph learning layer to automatically extract the dependencies between variables and constructs adaptive adjacency matrices for different datasets. In addition, an information enhancement module is introduced to magnify perceptual scope of the model and strengthen the connection between information, so that the spatial-temporal graph convolution module can better capture the spatial-temporal dependencies in the traffic flow. We conducted experiments on multivariate traffic flow data from three different real-world scenarios to evaluate the effectiveness of the proposed method. Comparative experimental results consistently show that the proposed method outperforms existing GCNs, achieving an average improvement of 21.16% on Mean Absolute Error (MAE), 18.87% on Root Mean Square Error (RMSE), and 8.27% on R2 score (R2) in the three datasets, respectively, and the results demonstrate its excellent forecasting performance in the field of multivariate traffic flow forecasting.
What problem does this paper attempt to address?