Computer Science ›› 2023, Vol. 50 ›› Issue (6A): 220600266-7.doi: 10.11896/jsjkx.220600266

• Big Data & Data Science • Previous Articles     Next Articles

City Traffic Flow Prediction Method Based on Dynamic Spatio-Temporal Neural Network

MENG Xiangfu, XU Ruihang   

  1. School of Electronic and Information Engineering,Liaoning Technical University,Huludao,Liaoning 125105,China
  • Online:2023-06-10 Published:2023-06-12
  • About author:MENG Xiangfu,born in 1981,Ph.D,professor,is a senior member of China Computer Federation.His main research interests include big data analysis and query,spatio-temporal data mi-ning,and machine learning algorithms.
  • Supported by:
    National Natural Science Foundation of China(61772249) and Research Project of Liaoning Education Department(LJKZ0355).

Abstract: Traffic flow forecasting is of great importance to urban road planning,traffic safety issues and building smart cities.However,most existing traffic prediction models cannot capture the dynamic spatio-temporal correlation of traffic data well enough to obtain satisfactory prediction results.To address this problem,a dynamic spatio-temporal neural network-based city traffic flow prediction method is proposed to solve the traffic flow prediction problem.First,by modelling the nearest cycle dependence,daily cycle dependence and weekly cycle dependence of the traffic data,a 3D convolutional neural network is used on each component to extract the high-dimensional features of urban traffic.Then,an improved residual structure is used to capture the correlation between remote area pairs and the prediction area,and a fusion of spatial attention and temporal attention mechanisms is used to capture the dynamic correlation between traffic flows in different time periods in different areas.Finally,the outputs of the three components are weighted and fused using a parameter matrix-based approach to obtain the prediction results.Experiments on two publicly available datasets,TaxiBJ and BikeNYC,show that the proposed model outperforms the mainstream traffic forecasting models.

Key words: Temporal correlation, Traffic forecast, Traffic fluidity, Attention mechanism, Fusion mechanism

CLC Number: 

  • TP183
[1]BENABDALLAH B R,BEGHDAD B K.A Comparative Study of Road Traffic Forecasting Models[C]//International Confe-rence on Artificial Intelligence and Its Applications.Cham:Springer,2022:272-280.
[2]WU T F,XU H Z,LU J.Short term passenger flow forecastbased on arrival interval[J].Journal of Wuhan University of Technology(Traffic Science and Engineering),2020,44(6):1098-1102.
[3]ZIVOT E,WANG J.Vector autoregressive models for multivariate time series[J/OL]http://www.databaser.net/moniwiki/pds/TimeSeries/varModels.pdf.
[4]YANG H,LI X,QIANG W,et al.A network traffic forecasting method based on SA optimized ARIMA-BP neural network[J].Computer Networks,2021,193:108102.
[5]VAN L J W C,VAN H C.Short-term traffic and travel time prediction models[J].Artificial Intelligence Applications to Cri-tical Transportation Issues,2012,22(1):22-41.
[6]JEONG Y S,BYON Y J,CASTRO-NETO M M,et al.Super-vised weighting-online learning algorithm for short-term traffic flow prediction[J].IEEE Transactions on Intelligent Transportation Systems,2013,14(4):1700-1707.
[7]LIU X,SONG L,LIU S,et al.A review of deep-learning-based medical image segmentation methods[J].Sustainability,2021,13(3):1224.
[8]TANG J X,CHEN Y,ZHOU M Y,et al.A survey of studies on deep learning applications in POI recommendation[J].Computer Engineering,2022,48(1):12-23,42.
[9]ZHANG J,ZHENG Y,QI D,et al.Predicting citywide crowdflows using deep spatio-temporal residual networks[J].Artificial Intelligence,2018,259:147-166.
[10]YAO H,WU F,KE J,et al.Deep multi-view spatial-temporal network for taxi demand prediction[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2018.
[11]DAI G,HU X Y,GE Y M,et al.Attention based simplified deep residual network for citywide crowd flows prediction[J].Frontiers of Computer Science,2021,15(2):1-12.
[12]ZHANG Y,HU Y,XIN D R.A deep learning short-term traffic flow prediction method considering spatial-temporal association[J].Chinese Journal of Intelligent Science and Technology,2021,3(2):172-178.
[13]LI S J,ZHU S S,WANG P.Traffic prediction based on spatiotemporal correlation multitask neural network[J].Computer Applications and Software,2021,38(9):286-292.
[14]ABDULJABBAR R L,DIA H,TSAI P W,et al.Short-termtraffic forecasting:an LSTM network for spatial-temporal speed prediction[J].Future Transportation,2021,1(1):21-37.
[15]ZHANG X H,FENG A M.Short-term traffic flow prediction based on empirical mode decomposition and long short-term memory neural network[J].Journal of Computer Applications,2021,41(1):225-230.
[16]TSAI M J,CHEN H Y,CUI Z,et al.Multivariate Long And Short Term LSTM-Based Network for Traffic Forecasting Under Interference:Experiments During COVID-19[C]//2021 IEEE International Intelligent Transportation Systems Confe-rence(ITSC).IEEE,2021:2169-2174.
[17]NAQVI U,MAJID A,ABBAS S A.UTSA:Urdu text sentiment analysis using deep learning methods[J].IEEE Access,2021,9:114085-114094.
[18]CAI W,ZHAI B,LIU Y,et al.Quadratic polynomial guided fuzzy C-means and dual attention mechanism for medical image segmentation[J].Displays,2021,70:102106.
[19]XU K,BA J,KIROS R,et al.Show,attend and tell:Neuralimage caption generation with visual attention[C]//International Conference on Machine Learning.PMLR,2015:2048-2057.
[20]VELICKOVIC P,CUCURULL G,CASANOVA A,et al.Graph attention networks[J].arXiv:1710.10903,2017.
[21]LIU Q,LIU T,CAI Y,et al.Explanatory prediction of traffic congestion propagation mode:A self-attention based approach[J].Physica A:Statistical Mechanics and its Applications,2021,573:125940.
[22]WANG J J,CHEN Q K.Traffic Prediction Model Based on Spatiotemporal Attention Network[J].Journal of Chinese Compu-ter Systems,2021,42(2):303-307.
[23]REN J H,ZHU Y,et al.Predicting Citywide Traffic Flow Using Dynamic Spatial-temporal Neural Networks[J].Journal of Chinese Computer Systems,2023,44(3):529-535.
[24]LIN Z,FENG J,LU Z,et al.Deepstn+:Context-aware spatial-temporal neural network for crowd flow prediction in metropolis[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2019:1020-1027.
[25]WOO S Y,PARK J C,LEE J Y.CBAM:convolutional block attention module[C]//European Conference on Computer Vision(ECCV).2018:3-19.
[26]YANG D,CHEN K,YANG M,et al.Urban rail transit passenger flow forecast based on LSTM with enhanced long-term features[J].IET Intelligent Transport Systems,2019,13(10):1475-1482.
[27]PAN Z,LIANG Y,WANG W,et al.Urban traffic prediction from spatio-temporal data using deep meta learning[C]//25th ACM SIGKDD International Conference on Knowledge Disco-very & Data Mining,2019:1720-1730.
[1] WANG Jiahao, ZHONG Xin, LI Wenxiong, ZHAO Dexin. Human Activity Recognition with Meta-learning and Attention [J]. Computer Science, 2023, 50(8): 193-201.
[2] WANG Yu, WANG Zuchao, PAN Rui. Survey of DGA Domain Name Detection Based on Character Feature [J]. Computer Science, 2023, 50(8): 251-259.
[3] ZHANG Yian, YANG Ying, REN Gang, WANG Gang. Study on Multimodal Online Reviews Helpfulness Prediction Based on Attention Mechanism [J]. Computer Science, 2023, 50(8): 37-44.
[4] TENG Sihang, WANG Lie, LI Ya. Non-autoregressive Transformer Chinese Speech Recognition Incorporating Pronunciation- Character Representation Conversion [J]. Computer Science, 2023, 50(8): 111-117.
[5] YAN Mingqiang, YU Pengfei, LI Haiyan, LI Hongsong. Arbitrary Image Style Transfer with Consistent Semantic Style [J]. Computer Science, 2023, 50(7): 129-136.
[6] GAO Xiang, TANG Jiqiang, ZHU Junwu, LIANG Mingxuan, LI Yang. Study on Named Entity Recognition Method Based on Knowledge Graph Enhancement [J]. Computer Science, 2023, 50(6A): 220700153-6.
[7] ZHANG Tao, CHENG Yifei, SUN Xinxu. Graph Attention Networks Based on Causal Inference [J]. Computer Science, 2023, 50(6A): 220600230-9.
[8] CUI Lin, CUI Chenlu, LIU Zhengwei, XUE Kai. Speech Emotion Recognition Based on Improved MFCC and Parallel Hybrid Model [J]. Computer Science, 2023, 50(6A): 220800211-7.
[9] DUAN Jianyong, YANG Xiao, WANG Hao, HE Li, LI Xin. Document-level Relation Extraction of Graph Attention Convolutional Network Based onInter-sentence Information [J]. Computer Science, 2023, 50(6A): 220800189-6.
[10] YANG Xing, SONG Lingling, WANG Shihui. Remote Sensing Image Classification Based on Improved ResNeXt Network Structure [J]. Computer Science, 2023, 50(6A): 220100158-6.
[11] ZHANG Shunyao, LI Huawang, ZHANG Yonghe, WANG Xinyu, DING Guopeng. Image Retrieval Based on Independent Attention Mechanism [J]. Computer Science, 2023, 50(6A): 220300092-6.
[12] LIU Haowei, YAO Jingchi, LIU Bo, BI Xiuli, XIAO Bin. Two-stage Method for Restoration of Heritage Images Based on Muti-scale Attention Mechanism [J]. Computer Science, 2023, 50(6A): 220600129-8.
[13] LI Fan, JIA Dongli, YAO Yumin, TU Jun. Graph Neural Network Few Shot Image Classification Network Based on Residual and Self-attention Mechanism [J]. Computer Science, 2023, 50(6A): 220500104-5.
[14] SUN Kaiwei, WANG Zhihao, LIU Hu, RAN Xue. Maximum Overlap Single Target Tracking Algorithm Based on Attention Mechanism [J]. Computer Science, 2023, 50(6A): 220400023-5.
[15] WU Liuchen, ZHANG Hui, LIU Jiaxuan, ZHAO Chenyang. Defect Detection of Transmission Line Bolt Based on Region Attention Mechanism andMulti-scale Feature Fusion [J]. Computer Science, 2023, 50(6A): 220200096-7.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!