Abstract
Traffic flow prediction is the upstream problem of path planning, intelligent transportation system, and other tasks. Many studies have been carried out on the traffic flow prediction of the spatio-temporal network, but the effects of spatio-temporal flexibility (historical data of the same type of time intervals in the same location will change flexibly) and spatio-temporal correlation (different road conditions have different effects at different times) have not been considered at the same time. We propose the Deep Spatio-temporal Adaptive 3D Convolution Neural Network (ST-A3DNet), which is a new scheme to solve both spatio-temporal correlation and flexibility, and consider spatio-temporal complexity (complex external factors, such as weather and holidays). Different from other traffic forecasting models, ST-A3DNet captures the spatio-temporal relationship at the same time through the Adaptive 3D convolution module, assigns different weights flexibly according to the influence of historical data, and obtains the impact of external factors on the flow through the ex-mask module. Considering the holidays and weather conditions, we train our model for experiments in Xi’an and Chengdu. We evaluate the ST-A3DNet and the results show that we have better results than the other 11 baselines.
- [1] . 2019. Stg2seq: Spatial-temporal graph to sequence model for multi-step passenger demand forecasting. In Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence (IJCAI’19). International Joint Conferences on Artificial Intelligence Organization, 1981–1987. https://doi.org/10.24963/ijcai.2019/274 Google ScholarDigital Library
- [2] Lei Bai, Lina Yao, Can Li, Xianzhi Wang, and Can Wang. 2020. Adaptive graph convolutional recurrent network for traffic forecasting. In Advances in Neural Information Processing Systems, H. Larochelle, M. Ranzato, R. Hadsell, M. F. Balcan, and H. Lin (Eds.), Vol. 33. Curran Associates, Inc., 17804–17815. https://proceedings.neurips.cc/paper/2020/file/ce1aad92b939420fc17005e5461e6f48-Paper.pdf.Google Scholar
- [3] . 2020. Multi-range attentive bicomponent graph convolutional network for traffic forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence, Vol. 34. 3529–3536.Google ScholarCross Ref
- [4] . 2019. GSTNet: Global spatial-temporal network for traffic flow prediction.. In Proceedings of the 28th International Joint Conferenceon ArtificialIntelligence. 2286–2293. Google ScholarDigital Library
- [5] . 2016. Using LSTM and GRU neural network methods for traffic flow prediction. In Proceedings of the 2016 31st Youth Academic Annual Conference of Chinese Association of Automation. IEEE, 324–328.Google ScholarCross Ref
- [6] . 2019. Multi-modal graph interaction for multi-graph convolution network in urban spatiotemporal forecasting. arXiv:1905.11395. Retrieved from https://arxiv.org/abs/1905.11395.Google Scholar
- [7] . 2019. Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence, Vol. 33. 922–929. Google ScholarDigital Library
- [8] . 2019. Deep spatial–temporal 3D convolutional neural networks for traffic data forecasting. IEEE Transactions on Intelligent Transportation Systems 20, 10 (2019), 3913–3926.Google ScholarCross Ref
- [9] . 2018. Squeeze-and-excitation networks. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. 7132–7141.Google ScholarCross Ref
- [10] . 2015. Machine learning: Trends, perspectives, and prospects. Science 349, 6245 (2015), 255–260.Google ScholarCross Ref
- [11] . 2015. Deep learning. Nature 521, 7553 (2015), 436–444.Google ScholarCross Ref
- [12] . 2017. Diffusion convolutional recurrent neural network: Data-driven traffic forecasting. In International Conference on Learning Representations. https://openreview.net/forum?id=SJiHXGWAZ.Google Scholar
- [13] . 2019. Learning heterogeneous spatial-temporal representation for bike-sharing demand prediction. In Proceedings of the AAAI Conference on Artificial Intelligence, Vol. 33. 1004–1011. Google ScholarDigital Library
- [14] . 2020. Dynamic spatial-temporal representation learning for traffic flow prediction. IEEE Transactions on Intelligent Transportation Systems 22, 11 (2020), 7169–7183.Google Scholar
- [15] . 2017. Short-term traffic flow prediction with Conv-LSTM. In Proceedings of the 2017 9th International Conference on Wireless Communications and Signal Processing. IEEE, 1–6.Google ScholarCross Ref
- [16] . 2017. Learning traffic as images: A deep convolutional neural network for large-scale transportation network speed prediction. Sensors 17, 4 (2017), 818.Google ScholarCross Ref
- [17] Mengzhang Li and Zhanxing Zhu. 2021. Spatial-temporal fusion graph neural networks for traffic flow forecasting. Proceedings of the AAAI Conference on Artificial Intelligence 35, 5 (May 2021), 4189–4196. https://ojs.aaai.org/index.php/AAAI/article/view/16542.Google Scholar
- [18] . 2015. Predicting traffic volumes and estimating the effects of shocks in massive transportation systems. Proceedings of the National Academy of Sciences 112, 18 (2015), 5643–5648.Google ScholarCross Ref
- [19] . 1997. Traffic flow forecasting: Comparison of modeling approaches. Journal of Transportation Engineering 123, 4 (1997), 261–266.Google ScholarCross Ref
- [20] . 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, Vol. 34. 914–921.Google ScholarCross Ref
- [21] . 2014. Prediction of human emergency behavior and their mobility following large-scale disaster. In Proceedings of the 20th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. 5–14. Google ScholarDigital Library
- [22] . 2015. Learning spatiotemporal features with 3d convolutional networks. In Proceedings of the IEEE International Conference on Computer Vision. 4489–4497. Google ScholarDigital Library
- [23] . 1996. Combining Kohonen maps with ARIMA time series models to forecast traffic flow. Transportation Research Part C: Emerging Technologies 4, 5 (1996), 307–318.Google ScholarCross Ref
- [24] . 2018. Cross-city transfer learning for deep spatio-temporal prediction. In Proceedings of the 28th International Joint Conference on Artificial Intelligence, Macao, China (IJCAI’19). AAAI Press, 1893–1899. Google ScholarDigital Library
- [25] . 2003. Modeling and forecasting vehicular traffic flow as a seasonal ARIMA process: Theoretical basis and empirical results. Journal of Transportation Engineering 129, 6 (2003), 664–672.Google ScholarCross Ref
- [26] . 2004. Travel-time prediction with support vector regression. IEEE Transactions on Intelligent Transportation Systems 5, 4 (2004), 276–281. Google ScholarDigital Library
- [27] . 2015. Convolutional LSTM network: A machine learning approach for precipitation nowcasting. In Proceedings of the Advances in Neural Information Processing Systems. 802–810. Google ScholarDigital Library
- [28] Bing Yu, Haoteng Yin, and Zhanxing Zhu. 2018. Spatio-temporal graph convolutional networks: A deep learning framework for traffic forecasting. In Proceedings of the 27th International Joint Conference on Artificial Intelligence Stockholm, Sweden (IJCAI’18). AAAI Press, 3634–3640. Google ScholarDigital Library
- [29] . 2019. 3D CNN-based accurate prediction for large-scale traffic flow. In Proceedings of the 2019 4th International Conference on Intelligent Transportation Engineering. IEEE, 99–103.Google ScholarCross Ref
- [30] . 2018. Consensus ensemble system for traffic flow prediction. IEEE Transactions on Intelligent Transportation Systems 19, 12 (2018), 3903–3914.Google ScholarDigital Library
- [31] . 2018. Gaan: Gated attention networks for learning on large and spatiotemporal graphs. Uncertainty in Artificial Intelligence (2018), 339–349.Google Scholar
- [32] . 2016. Deep spatio-temporal residual networks for citywide crowd flows prediction. In Proceedings of the Thirty-First AAAI Conference on Artificial Intelligence, San Francisco, California, USA (AAAI’17). AAAI Press, 1655–1661. Google ScholarDigital Library
- [33] . 2016. DNN-based prediction model for spatio-temporal data. In Proceedings of the 24th ACM SIGSPATIAL International Conference on Advances in Geographic Information Systems. 1–4. Google ScholarDigital Library
- [34] . 2018. Predicting citywide crowd flows using deep spatio-temporal residual networks. Artificial Intelligence 259, 0004–3702 (2018), 147–166.Google ScholarCross Ref
- [35] . 2019. Flow prediction in spatio-temporal networks based on multitask deep learning. IEEE Transactions on Knowledge and Data Engineering 32, 3 (2019), 468–478.Google ScholarCross Ref
- [36] . 2019. T-gcn: A temporal graph convolutional network for traffic prediction. IEEE Transactions on Intelligent Transportation Systems 21, 9 (2019), 3848–3858.Google ScholarCross Ref
- [37] . 2020. Short-term traffic flow forecasting method with MB-LSTM hybrid network. IEEE Transactions on Intelligent Transportation Systems (2020), 1–11.Google Scholar
- [38] . 2020. Gman: A graph multi-attention network for traffic prediction. In Proceedings of the AAAI Conference on Artificial Intelligence, Vol. 34. 1234–1241.Google ScholarCross Ref
Index Terms
- Deep Spatio-temporal Adaptive 3D Convolutional Neural Networks for Traffic Flow Prediction
Recommendations
Citywide Traffic Flow Prediction Based on Multiple Gated Spatio-temporal Convolutional Neural Networks
Traffic flow prediction is crucial for public safety and traffic management, and remains a big challenge because of many complicated factors, e.g., multiple spatio-temporal dependencies, holidays, and weather. Some work leveraged 2D convolutional neural ...
Deep spatio-temporal 3D dilated dense neural network for traffic flow prediction
AbstractTraffic flow prediction is increasingly vital for the administration of metropolitan areas. Many research on spatio-temporal networks have been explored but the impacts of both spatial and temporal flexibility, complex spatial correlation has not ...
Highlights- The autocorrelation analysis’s early fusion method is proposed for traffic flow prediction.
- A 3DCNN-based multi-scale densenet network is proposed to improve the prediction performance.
- SCR block is developed to recalibrate the ...
Deep spatio-temporal 3D densenet with multiscale ConvLSTM-Resnet network for citywide traffic flow forecasting
AbstractReliable traffic flow forecasting is paramount in Intelligent Transportation Systems (ITS) as it can effectively improve traffic efficiency and social security. Its vital challenge is to effectively integrate various factors (such as ...
Comments