Chen, C., Li, K., Teo, S.G., Zou, X., Wang, K., Wang, J. and Zeng, Z. (2019) Gated Residual Recurrent Graph Neural Networks for Traffic Prediction. Proceedings of the AAAI Conference on Artificial Intelligence, 1, 485-492.
https://doi.org/10.1609/aaai.v33i01.3301485

has been cited by the following article:

• TITLE: Multi-Head Attention Spatial-Temporal Graph Neural Networks for Traffic Forecasting

  AUTHORS: Xiuwei Hu, Enlong Yu, Xiaoyu Zhao

  KEYWORDS: Traffic Prediction, Intelligent Traffic System, Multi-Head Attention, Graph Neural Networks

  JOURNAL NAME: Journal of Computer and Communications, Vol.12 No.3, March 15, 2024

  ABSTRACT: Accurate traffic prediction is crucial for an intelligent traffic system (ITS). However, the excessive non-linearity and complexity of the spatial-temporal correlation in traffic flow severely limit the prediction accuracy of most existing models, which simply stack temporal and spatial modules and fail to capture spatial-temporal features effectively. To improve the prediction accuracy, a multi-head attention spatial-temporal graph neural network (MSTNet) is proposed in this paper. First, the traffic data is decomposed into unique time spans that conform to positive rules, and valuable traffic node attributes are mined through an adaptive graph structure. Second, time and spatial features are captured using a multi-head attention spatial-temporal module. Finally, a multi-step prediction module is used to achieve future traffic condition prediction. Numerical experiments were conducted on an open-source dataset, and the results demonstrate that MSTNet performs well in spatial-temporal feature extraction and achieves more positive forecasting results than the baseline methods.