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.
References
[1]
Lv, Y., Duan, Y., Kang, W., Li, Z. and Wang, F.Y. (2014) Traffic Flow Prediction with Big Data: A Deep Learning Approach. IEEE Transactions on Intelligent Transportation Systems, 16, 865-873. https://doi.org/10.1109/TITS.2014.2345663
[2]
Moorthy, C.K. and Ratcliffe, B.G. (1988) Short Term Traffic Forecasting Using Time Series Methods. Transportation Planning and Technology, 12, 45-56. https://doi.org/10.1080/03081068808717359
[3]
Van Der Voort, M., Dougherty, M. and Watson, S. (1996) Combining Kohonen Maps with ARIMA Time Series Models to Forecast Traffic Flow. Transportation Research Part C: Emerging Technologies, 4, 307-318. https://doi.org/10.1016/S0968-090X(97)82903-8
[4]
Lippi, M., Bertini, M. and Frasconi, P. (2013) Short-Term Traffic Flow Forecasting: An Experimental Comparison of Time-Series Analysis and Supervised Learning. IEEE Transactions on Intelligent Transportation Systems, 14, 871-882. https://doi.org/10.1109/TITS.2013.2247040
[5]
Zarei, N., Ghayour, M.A. and Hashemi, S. (2013) Road Traffic Prediction Using Context-Aware Random Forest Based on Volatility Nature of Traffic Flows. Intelligent Information and Database Systems: 5th Asian Conference, ACIIDS 2013, Kuala Lumpur, 18-20 March 2013, 196-205. https://doi.org/10.1007/978-3-642-36546-1_21
[6]
Cai, P., Wang, Y., Lu, G., Chen, P., Ding, C. and Sun, J. (2016) A Spatiotemporal Correlative k-Nearest Neighbor Model for Short-Term Traffic Multistep Forecasting. Transportation Research Part C: Emerging Technologies, 62, 21-34. https://doi.org/10.1016/j.trc.2015.11.002
[7]
Reza, S., Oliveira, H.S., Machado, J.J. and Tavares, J.M.R. (2021) Urban Safety: An Image-Processing and Deep-Learning-Based Intelligent Traffic Management and Control System. Sensors, 21, Article No. 7705. https://doi.org/10.3390/s21227705
[8]
Guo, S., Lin, Y., Feng, N., Song, C. and Wan, H. (2019) Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting. Proceedings of the AAAI Conference on Artificial Intelligence, 1, 922-929. https://doi.org/10.1609/aaai.v33i01.3301922
[9]
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
[10]
Yan, H., Ma, X. and Pu, Z. (2021) Learning Dynamic and Hierarchical Traffic Spatiotemporal Features with Transformer. IEEE Transactions on Intelligent Transportation Systems, 23, 22386-22399. https://doi.org/10.1109/TITS.2021.3102983
[11]
Li, Y., Yu, R., Shahabi, C. and Liu, Y. (2017) Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting.
[12]
Yu, B., Yin, H. and Zhu, Z. (2017) Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting. Proceedings of the 27th International Joint Conference on Artificial Intelligence, Stockholm, 13-19 July 2018, 3634-3640. https://doi.org/10.24963/ijcai.2018/505
[13]
Wu, Z., Pan, S., Long, G., Jiang, J. and Zhang, C. (2019) Graph Wavenet for Deep Spatial-Temporal Graph Modeling. Proceedings of the 28th International Joint Conference on Artificial Intelligence Main Track, Macao, 10-16 August 2019, 1907-1913. https://doi.org/10.24963/ijcai.2019/264
[14]
Bai, L., Yao, L., Li, C., Wang, X. and Wang, C. (2020) Adaptive Graph Convolutional Recurrent Network for Traffic Forecasting. NeurIPS 2020, 6-12 December 2020, 17804-17815.
[15]
Ta, X., Liu, Z., Hu, X., Yu, L., Sun, L. and Du, B. (2022) Adaptive Spatio-Temporal Graph Neural Network for Traffic Forecasting. Knowledge-Based Systems, 242, Article ID: 108199. https://doi.org/10.1016/j.knosys.2022.108199
[16]
Zhang, J., Wang, F.Y., Wang, K., Lin, W.H., Xu, X. and Chen, C. (2011) Data-Driven Intelligent Transportation Systems: A Survey. IEEE Transactions on Intelligent Transportation Systems, 12, 1624-1639. https://doi.org/10.1109/TITS.2011.2158001
[17]
Xie, Y., Zhang, Y. and Ye, Z. (2007) Short-Term Traffic Volume Forecasting Using Kalman Filter with Discrete Wavelet Decomposition. Computer-Aided Civil and Infrastructure Engineering, 22, 326-334. https://doi.org/10.1111/j.1467-8667.2007.00489.x
[18]
Goodfellow, I., Bengio, Y. and Courville, A. (2016) Deep Learning. MIT Press, Cambridge.
[19]
Wu, Z., Pan, S., Chen, F., Long, G., Zhang, C. and Philip, S.Y. (2020) A Comprehensive Survey on Graph Neural Networks. IEEE Transactions on Neural Networks and Learning Systems, 32, 4-24. https://doi.org/10.1109/TNNLS.2020.2978386
[20]
Zhao, L., Song, Y., Zhang, C., Liu, Y., Wang, P., Lin, T. and Li, H. (2019) T-gcn: A Temporal Graph Convolutional Network for Traffic Prediction. IEEE Transactions on Intelligent Transportation Systems, 21, 3848-3858. https://doi.org/10.1109/TITS.2019.2935152
