Computer Science ›› 2021, Vol. 48 ›› Issue (11A): 159-165.doi: 10.11896/jsjkx.201200051

• Intelligent Computing • Previous Articles     Next Articles

Dual Autoregressive Components Traffic Prediction Based on Improved Graph WaveNet

LI Hao, WANG Fei, XIE Si-yu, KOU Yong-qi, ZHANG Lan, YANG Bing, KANG Yan   

  1. School of Software,Yunnan University,Kunming 650504,China
  • Online:2021-11-10 Published:2021-11-12
  • About author:LI Hao,born in 1970,Ph.D,professor. His main research interests include distributed computing,grid and cloud computing.
    KANG Yan,born in 1972,Ph.D,associate professor.Her main research interests include transfer learning,deep learning and integrated learning.
  • Supported by:
    National Natural Science Foundation of China(61762092),Open Fund Project of Key Laboratory of Software Engineering in Yunnan Province(2017SE204),Key Laboratory of Software Engineering of Yunnan Province(2020SE303),Major Science and Technology Projects in Yunnan Province(202002AB080001),Material Genetic Engineering-calculation Software Development of Integrated Calculation Function Module Based on Metcloud(2019CLJY06) and Gene Engineering of Rare and Precious Metal Materials in Yunnan Province(2019ZE001-1,202002AB080001).

Abstract: With the construction of smart cities,urban traffic flow forecasting is crucial in intelligent traffic early warning and traffic management decision-making.Due to the complex temporal and spatial correlation,it is a challenge to effectively predict traffic flow.Most of the existing traffic flow prediction methods use machine learning algorithms or deep learning models,and the methods have their own advantages and disadvantages.If the two advantages can be combined,the accuracy of traffic flow prediction will be further improved.Aiming at the traffic spatio-temporal data,a dual autoregressive component traffic prediction model based on improved Graph WaveNet is proposed.First,the three time convolution layers are effectively fused through the gated three-branch time convolution network,which further improves the ability of capture time correlation.Second,the autoregression component is introduced for the first time to effectively fuse the autoregression component with the gated three-branch time convolution network and the convolution layer,so that the model can fully reflect the linear and non-linear relationship between space-time data.Through experiments on two real public transportation data sets of METR-LA and PEMS-BAY,the proposed model is compared with other traffic flow prediction benchmark models.The results show that whether it is a short-term or long-term forecast,the model proposed in the paper is better than the benchmark model in all indicators.

Key words: Autoregressive component, Intelligent transportation, Spatio-temporal data, Time convolution layer, Traffic flow forecasting

CLC Number: 

  • TP181
[1]YANG B,KANG Y,ZHANG Y,et al.Spatio-Temporal Expand-and-Squeeze Networks for Crowd Flow Prediction in Metropolis[J].IET Intelligent Transport Systems,2020,14(5).
[2]MEDRANO R D,AZNARTE J L.A Spatio-Temporal Spot-Forecasting Framework for Urban Traffic Prediction[J].arXiv:2003.13977,2020.
[3]AHMED M S,COOKA R.Analysis of freeway traffic time-series data by using box-jenkins techniques[J].Journal of the Transportation Research Record,1979,773(722):1-9.
[4]HODGE V J,KRISHNAN R,AUSTINJ,et al.Short-term prediction of traffic flow using a binary neural network[J].Neural Computing & Applications,2014,25(12):1639-1655.
[5]SUN H,ZHANG C,RAN B.Interval prediction for traffic time series using local linear predictor[C]//The 7th International IEEE Conference on Intelligent Transportation Systems.IEEE,2004.
[6]WILLIAMS B M,HOEL L A.Modeling and Forecasting Vehicular Traffic Flow as a Seasonal ARIMA Process:Theoretical Basis and Empirical Results[J].Journal of Transportation Engineering,2003,129(6):664-672.
[7]OKUTANI I,STEPHANEDES Y J.Dynamic prediction of traffic volume through Kalman filtering theory[J].Transportation Research Part B Methodological,1984,18(1):1-11.
[8]KE J,YANG H,ZHENG H,et al.Hexagon-Based ConvolutionalNeural Network for Supply-Demand Forecasting of Ride-Sourcing Services[J].IEEE Transactions on Intelligent Transportation Systems,2019,20(11):4160-4173.
[9]HOCHREITER S,SCHMIDHUBER J.Long Short-Term Me-mory[J].Neural Computation,1997,9(8):1735-1780.
[10]CHO K,MERRIENBOER B V ,BAHDANAU D,et al.On the Properties of Neural Machine Translation:Encoder-Decoder Approaches[J].arXiv:1409.1259,2014.
[11]SONG C,LIN Y,GUO S,et al.Spatial-Temporal Synchronous Graph Convolutional Networks:A New Framework for Spatial-Temporal Network Data Forecasting[J].Proceedings of the AAAI Conference on Artificial Intelligence,2020,34(1):914-921.
[12]LI Y,YU R,SHAHABI C,et al.Diffusion Convolutional Recurrent Neural Network:Data-Driven Traffic Forecasting[J].arXiv:1707.01926,2017.
[13]WU Z,PAN S,LONG G,et al.Graph Wave-Net for Deep Spatial-Temporal Graph Modeling[C]//IJCAI-19.2019:1907-1913.
[14]HUANG S,WANG D,WU X,et al.DSANet:Dual Self-Attention Network for Multivariate Time Series Forecasting[C]//the 28th ACM International Conference.ACM,2019.
[15]WANG J Y,LI C,XIONG Z,et al.Survey of dataGcentric smart city[J].Journal of Computer Research and Development,2014(2):239-259.
[16]DRUCKER H,BURGES C J,KAUFMAN L,et al.Support vector regression machines[C]//Advances in Neural Information Processing Systems.1997:155-161.
[17]SUN S,ZHANG C,YU G.A bayesian network approach totraffic flow forecasting[J].IEEE Transactions on Intelligent Transportation Systems,2006,7(1):124-132.
[18]SHI X,CHEN Z,WANG H,et al.Convolutional LSTM Net-work:A Machine Learning Approach for Precipitation Nowcasting[J].arXiv:1506.04214,2015.
[19]ZHANG J,YU Z,QI D.Deep Spatio-Temporal Residual Net-works for Citywide Crowd Flows Prediction[C]//AAAI'17.2017:1655-1661.
[20]YU B,YIN H,ZHU Z.Spatio-Temporal Graph Convolutional Networks:A Deep Learning Framework for Traffic Forecasting[J].arXiv:1709.04875,2017.
[21]GUO S,LIN Y,FENG N,et al.Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2019:922-929.
[22]ZHENG C,FAN X,WANG C,et al.Gman:A graph multi-attention network for traffic prediction[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2020:1234-1241.
[23]KIPF T N,WELLING M.Semi-Supervised Classification withGraph Convolutional Networks[J].arXiv:1609.02907,2016.
[24]YU F,KOLTUN V.Multi-Scale Context Aggregation by Dilated Convolutions[C]//ICLR.2016.
[25]DAUPHIN Y A N,FAN A,AULI M,et al.Language modeling with gated convolutional networks[C]//ICML.2017:933-941.
[26]ORESHKIN B N,AMINI A,COYLE L,et al.FC-GAGA:Fully Connected Gated Graph Architecture for Spatio-Temporal Traffic Forecasting[J].arXiv:2007.15531,2020.
[27]WU Z,PAN S,LONG G,et al.Connecting the Dots:Multiva-riate Time Series Forecasting with Graph Neural Networks[C]//KDD'20.ACM,2020.
[1] LI Rong-fan, ZHONG Ting, WU Jin, ZHOU Fan, KUANG Ping. Spatio-Temporal Attention-based Kriging for Land Deformation Data Interpolation [J]. Computer Science, 2022, 49(8): 33-39.
[2] WANG Ru-bin, LI Rui-yuan, HE Hua-jun, LIU Tong, LI Tian-rui. Distributed Distance Join Algorithm for Massive Spatial Data [J]. Computer Science, 2022, 49(1): 95-100.
[3] NIU Kang-li, CHEN Yu-zhang, ZHANG Gong-ping, TAN Qian-cheng, WANG Yi-chong, LUO Mei-qi. Vehicle Flow Measuring of UVA Based on Deep Learning [J]. Computer Science, 2021, 48(6A): 275-280.
[4] ZHOU Xin, LIU Shuo-di, PAN Wei, CHEN Yuan-yuan. Vehicle Color Recognition in Natural Traffic Scene [J]. Computer Science, 2021, 48(6A): 15-20.
[5] ZENG Wei-liang, HAN Yu, HE Jin-yuan, WU Miao-sen, SUN Wei-jun. Simulation Analysis on Dynamic Ridesharing Efficiency of Shared Autonomous Taxi [J]. Computer Science, 2021, 48(2): 257-263.
[6] LYU Ming-qi, HONG Zhao-xiong, CHEN Tie-ming. Traffic Flow Forecasting Method Combining Spatio-Temporal Correlations and Social Events [J]. Computer Science, 2021, 48(2): 264-270.
[7] YOU Lan, HAN Xue-wei, HE Zheng-wei, XIAO Si-yu, HE Du, PAN Xiao-meng. Improved Sequence-to-Sequence Model for Short-term Vessel Trajectory Prediction Using AIS Data Streams [J]. Computer Science, 2020, 47(9): 169-174.
[8] LI Lin, ZHAO Kai-yue, ZHAO Xiao-yong, WEI Shuai-qin and ZHANG Bing. Contaminated and Shielded Number Plate Recognition Based on Convolutional Neural Network [J]. Computer Science, 2020, 47(6A): 213-219.
[9] SUN Hao,CHEN Chun-lin,LIU Qiong,ZHAO Jia-bao. Traffic Signal Control Method Based on Deep Reinforcement Learning [J]. Computer Science, 2020, 47(2): 169-174.
[10] SUN Tian-xu, ZHAO Yun-long, LIAN Zuo-wei, SUN Yi, CAI Yue-xiao. Mobility Pattern Mining for People Flow Based on Spatio-Temporal Data [J]. Computer Science, 2020, 47(10): 91-96.
[11] LIU Chang-yun,YANG Yu-di,ZHOU Li-hua,ZHAO Li-hong. Discovering Popular Social Location with Time Label [J]. Computer Science, 2019, 46(7): 186-194.
[12] GUO Sheng-nan, LIN You-fang, JIN Wen-wei, WAN Huai-yu. Citywide Crowd Flows Prediction Based on Spatio-Temporal Recurrent Convolutional Networks [J]. Computer Science, 2019, 46(6A): 385-391.
[13] DAI Liang,MEI Yang,QIAO Chao,MENG Yun,LV Jin-ming. Survey on Short-term Traffic Flow Forecasting Based on Deep Learning [J]. Computer Science, 2019, 46(3): 39-47.
[14] LAI Jian-hui. Traffic Signal Control Based on Double Deep Q-learning Network with Dueling Architecture [J]. Computer Science, 2019, 46(11A): 117-121.
[15] DUAN Na and WANG Lei. Image Retrieval of Global and Personalized ROI Adjustment of Features [J]. Computer Science, 2016, 43(Z11): 205-207.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!