Abstract
Traffic prediction is the core task of intelligent transportation system (ITS) and accurate traffic prediction can greatly improve the utilization of public resources. Dynamic interaction of multiple spatial relationships will influence the accuracy of traffic prediction. However, many existing methods only consider static spatial relationships, which restricts the accuracy of the prediction. To address the above problem, in this article, we propose the Dynamic Multi-Graph Fusion Network (DMGF-Net) to model the spatial-temporal correlations in traffic network. In the DMGF-Net, the fusion graph is designed to leverage and extract the various spatial correlations between different regions by fusing spatial graph, semantic graph, and spatial-semantic graph. Further, to dynamically learn the importance of different neighbors, we design the Dynamic Spatial-Temporal Unit (DSTU), which can adjust the aggregation weights of different neighbors by combining the convolution operation and the attention mechanism. It can selectively aggregate spatial-temporal features from different neighbors. Extensive experiments on three datasets demonstrate that effectiveness of our model, especially on PEMS08, our model achieves an increase of about 8.55% and 7.55% in terms of MAE and RMSE than the static model STGCN.
- [1] . 2020. Adaptive graph convolutional recurrent network for traffic forecasting. In Proceedings of the Neural Information Processing Systems (2020).Google Scholar
- [2] . 2019. Stg2seq: Spatial-temporal graph to sequence model for multi-step passenger demand forecasting. In Proceedings of the International Joint Conference on Artificial Intelligence, (2019).Google ScholarCross Ref
- [3] . 2014. Learning phrase representations using rnn encoder-decoder for statistical machine translation. In Proceedings of the 2014 Conference on Empirical Methods in Natural Language Processing (EMNLP). 1724–1734.Google ScholarCross Ref
- [4] . 2018. Deep multi-view spatial-temporal network for taxi demand prediction. In Proceedings of the 32nd AAAI Conferenceon Artificial Intelligence, AAAI 2018 (2018), 2588–2595.Google ScholarCross Ref
- [5] . 2012. Comparison of two new ARIMA-ANN and ARIMA-Kalman hybrid methods for wind speed prediction. Applied Energy 98 (2012), 415–424.Google Scholar
- [6] . 2019. Spatiotemporal multi-graph convolution network for ride-hailing demand forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence (2019).Google ScholarDigital Library
- [7] . 2019. Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence (2019).Google ScholarDigital Library
- [8] . 2016. DNN-based prediction model for spatio-temporal data. In Proceedings of the 24th ACM SIGSPATIAL International Conference on Advances in Geographic Information Systems (Burlingame, California) (SIGSPACIAL’16).Google ScholarDigital Library
- [9] . 2020. Predicting citywide crowd flows in irregular regions using multi-view graph convolutional networks. IEEE Transactions on Knowledge and Data Engineering 34, 5 (2020), 2348–2359.Google Scholar
- [10] . 2018. Diffusion convolutional recurrent neural network: Data-driven traffic forecasting. In Proceedings of the 6th International Conference on Learning Representations (ICLR), 2018.Google Scholar
- [11] . 2020. T-GCN: A temporal graph convolutional network for traffic prediction. IEEE Transactions on Intelligent Transportation Systems 21, 9 (2020), 3848–3858.Google Scholar
- [12] . 2020. Spatiotemporal adaptive gated graph convolution network for urban traffic flow forecasting. In Proceedings of the 29th ACM International Conference on Information & Knowledge Management (2020).Google ScholarDigital Library
- [13] . 1996. Combining kohonen maps with ARIMA time series models to forecast traffic flow. Transportation Research Part C: Emerging Technologies 4, 5 (1996), 307–318Google Scholar
- [14] . 2013. Short-term traffic flow forecasting: An experimental comparison of time-series analysis and supervised learning. IEEE Transactions on Intelligent Transportation Systems 14, 2 (2013), 871–882.Google Scholar
- [15] . 2019. DeepUrbanEvent: A system for predicting citywide crowd dynamics at big events. In Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD’19). Association for Computing Machinery, New York, NY, 2114–2122.
DOI: Google ScholarDigital Library - [16] . 2019. Deep spatial-temporal 3D convolutional neural networks for traffic data forecasting. IEEE Transactions on Intelligent Transportation Systems 20, 10 (2019), 3913–3926.Google Scholar
- [17] . 2019. Cross-city transfer learning for deep spatio-temporal prediction. In Proceedings of the 28th International Joint Conference on Artificial Intelligence (IJCAI’19). AAAI Press, 1893–1899.Google ScholarDigital Library
- [18] . 2019. GSTNet: Global Spatial-Temporal Network for traffic flow prediction. In Proceedings of the 28th International Joint Conference on Artificial Intelligence, IJCAI-19. International Joint Conferences on Artificial Intelligence Organization. 228–2293.Google ScholarCross Ref
- [19] . 2020. Spatial-temporal synchronous graph convolutional networks: A new framework for spatial-temporal network data forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence (2020).Google ScholarCross Ref
- [20] . 1997. Support vector regression machines. In Proceedings of the Advances in Neural Information Processing Systems. 155–161.Google Scholar
- [21] . 2019. Graph wavenet for deep spatial-temporal graph modeling. In Proceedings of the International Joint Conference on Artificial Intelligence (2019).Google ScholarCross Ref
- [22] . 2018. PredRNN++: Towards a resolution of the deep-in-time dilemma in spatiotemporal predictive learning. In Proceedings of the International Conference on Machine Learning (2018).Google Scholar
- [23] . 2020. Multi-range attentive bicomponent graph convolutional network for traffic forecasting. Proceedings of the AAAI Conference on Artificial Intelligence 34, 4 (2020), 3529–3536.Google Scholar
- [24] . 2018. Spatio-temporal graph convolutional networks: A deep learning framework for traffic forecasting. In Proceedings of the International Joint Conference on Artificial Intelligence (2018).Google ScholarCross Ref
- [25] . 2017. Deep spatio-temporal residual networks for citywide crowd flows prediction. In Proceedings of the Association for the Advancement of Artificial Intelligence (2017).Google ScholarCross Ref
- [26] . 2019. Spatial-temporal graph attention networks: A deep learning approach for traffic forecasting. IEEE Access 7 (2019), 166246–166256.Google Scholar
- [27] . 2019. GMAN: A graph multi-attention network for traffic prediction. In Proceedings of the Association for the Advancement of Artificial Intelligence (2019).Google Scholar
- [28] . 1994. Time series analysis, Vol. 2. Princeton University Press Princeton, NJ.Google Scholar
- [29] . 2020. LSGCN: Long short-term traffic prediction with graph convolutional networks. In Proceedings of the 29th International Joint Conference on Artificial Intelligence (IJCAI’20). Article 326, 2355–2361.Google ScholarCross Ref
- [30] . 2021. Spatial-temporal fusion graph neural networks for traffic flow forecasting. In Proceedings of the 35th AAAI Conference on Artificial Intelligence, A Virtual Conference. 4189–4196.Google Scholar
- [31] . 2021. Spatial-temporal graph ode networks for traffic flow forecasting. In Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery and Data Mining. 364–373.Google ScholarDigital Library
- [32] . 2022. RadNet: Incident prediction in spatio-temporal road graph networks using traffic forecasting. IJCAI DSO Workshop (2022).Google Scholar
- [33] . 2022. From statistical-to machine learning-based network traffic prediction. Transactions on Emerging Telecommunications Technologies 33, 4 (2022), e4394.
DOI: Google ScholarDigital Library - [34] . 2017. Dynamic edgeconditioned filters in convolutional neural networks on graphs. In Proceedings of the Computer Vision and Pattern Recognition. 3693–3702.Google Scholar
- [35] . 1997. Long short-term memory. Neural Computation 9, 8 (1997), 1735–1780.Google Scholar
- [36] . 2017. Effective deep memory networks for distant supervised relation extraction. In Proceedings of the International Joint Conference on Artificial Intelligence. 19–25.Google ScholarCross Ref
Index Terms
- DMGF-Net: An Efficient Dynamic Multi-Graph Fusion Network for Traffic Prediction
Recommendations
Dynamic Graph Convolutional Recurrent Network for Traffic Prediction: Benchmark and Solution
Traffic prediction is the cornerstone of intelligent transportation system. Accurate traffic forecasting is essential for the applications of smart cities, i.e., intelligent traffic management and urban planning. Although various methods are proposed for ...
Traffic prediction for dynamic traffic engineering
Traffic engineering with traffic prediction is a promising approach to accommodate time-varying traffic without frequent route changes. In this approach, the routes are decided so as to avoid congestion on the basis of the predicted traffic. However, if ...
MFSTGN: a multi-scale spatial-temporal fusion graph network for traffic prediction
AbstractTraffic prediction is an important link in building a smart city, accurate prediction of future traffic conditions can provide a reference for people to travel and facilitate the work of relevant departments, and is vital for urban applications. ...
Comments