计算机科学 ›› 2021, Vol. 48 ›› Issue (11A): 159-165.doi: 10.11896/jsjkx.201200051

• 智能计算 • 上一篇    下一篇

一种基于改进图波网的双重自回归分量交通预测模型

李浩, 王飞, 谢思宇, 寇勇奇, 张兰, 杨兵, 康雁   

  1. 云南大学软件学院 昆明650504
  • 出版日期:2021-11-10 发布日期:2021-11-12
  • 通讯作者: 康雁(562530855@qq.com)
  • 作者简介:lihao707@ynu.edu.cn
  • 基金资助:
    国家自然科学基金(61762092);云南省软件工程重点实验室开放基金项目(2017SE204);云南省软件工程重点实验室(2020SE303);云南省重大科技专项(202002AB080001);《材料基因工程-基于Metcloud的集成计算功能模块计算软件开发》(2019CLJY06);云南省稀贵金属材料基因工程(2019ZE001-1,202002AB080001)

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).

摘要: 随着智慧城市的建设,城市交通流量预测在智能交通预警和交通管理决策方面至关重要。由于复杂的时空相关性,有效地对交通流量进行预测成为了一项挑战。现有的对交通流量进行预测的方法大多采用机器学习算法或深度学习模型,而它们各有优缺点,若能够将两者优点结合起来,将进一步提高交通流量预测的精度。文中针对交通时空数据,提出了一种基于改进图波网(Graph WaveNet)的双重自回归分量交通预测模型。首先,通过门控3分支时间卷积网络有效融合3个时间卷积层,从而进一步提升了捕获时间相关性的能力;其次,首次引入自回归分量,将自回归分量和门控三分支时间卷积网络、图卷积层有效融合,使模型能够充分反映时空数据之间的线性和非线性关系。在METR-LA和PEMS-BAY两个真实的公共交通数据集上进行实验,并将所提模型与其他交通流量预测基准模型进行比较。结果表明,不管是短时间还是长时间的预测,文中所提模型在各个指标上都优于基准模型。

关键词: 交通流量预测, 时间卷积层, 时空数据, 智能交通, 自回归分量

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

中图分类号: 

  • 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] 黎嵘繁, 钟婷, 吴劲, 周帆, 匡平.
基于时空注意力克里金的边坡形变数据插值方法
Spatio-Temporal Attention-based Kriging for Land Deformation Data Interpolation
计算机科学, 2022, 49(8): 33-39. https://doi.org/10.11896/jsjkx.210600161
[2] 王如斌, 李瑞远, 何华均, 刘通, 李天瑞.
面向海量空间数据的分布式距离连接算法
Distributed Distance Join Algorithm for Massive Spatial Data
计算机科学, 2022, 49(1): 95-100. https://doi.org/10.11896/jsjkx.210100060
[3] 宋龙泽, 万怀宇, 郭晟楠, 林友芳.
面向出租车空载时间预测的多任务时空图卷积网络
Multi-task Spatial-Temporal Graph Convolutional Network for Taxi Idle Time Prediction
计算机科学, 2021, 48(7): 112-117. https://doi.org/10.11896/jsjkx.201000089
[4] 周欣, 刘硕迪, 潘薇, 陈媛媛.
自然交通场景中的车辆颜色识别
Vehicle Color Recognition in Natural Traffic Scene
计算机科学, 2021, 48(6A): 15-20. https://doi.org/10.11896/jsjkx.200800078
[5] 牛康力, 谌雨章, 张龚平, 谭前程, 王绎冲, 罗美琪.
基于深度学习的无人机航拍车流量监测
Vehicle Flow Measuring of UVA Based on Deep Learning
计算机科学, 2021, 48(6A): 275-280. https://doi.org/10.11896/jsjkx.200900149
[6] 曾伟良, 韩宇, 何锦源, 吴淼森, 孙为军.
自动驾驶出租车动态合乘效益仿真分析
Simulation Analysis on Dynamic Ridesharing Efficiency of Shared Autonomous Taxi
计算机科学, 2021, 48(2): 257-263. https://doi.org/10.11896/jsjkx.200400008
[7] 游兰, 韩雪薇, 何正伟, 肖丝雨, 何渡, 潘筱萌.
基于改进Seq2Seq的短时AIS轨迹序列预测模型
Improved Sequence-to-Sequence Model for Short-term Vessel Trajectory Prediction Using AIS Data Streams
计算机科学, 2020, 47(9): 169-174. https://doi.org/10.11896/jsjkx.190800060
[8] 李林, 赵凯月, 赵晓永, 魏帅琴, 张兵.
基于卷积神经网络的污损遮挡号牌分类
Contaminated and Shielded Number Plate Recognition Based on Convolutional Neural Network
计算机科学, 2020, 47(6A): 213-219. https://doi.org/10.11896/JsJkx.191100089
[9] 熊亭, 戚湧, 张伟斌.
基于DCGRU-RF模型的路网短时交通流预测
Short-term Traffic Flow Prediction Based on DCGRU-RF Model for Road Network
计算机科学, 2020, 47(5): 84-89. https://doi.org/10.11896/jsjkx.190100213
[10] 孙浩,陈春林,刘琼,赵佳宝.
基于深度强化学习的交通信号控制方法
Traffic Signal Control Method Based on Deep Reinforcement Learning
计算机科学, 2020, 47(2): 169-174. https://doi.org/10.11896/jsjkx.190600154
[11] 张德干, 杨鹏, 张捷, 高瑾馨, 张婷.
基于量子粒子群优化策略的车联网交通流量预测方法
New Method of Traffic Flow Forecasting of Connected Vehicles Based on Quantum Particle Swarm Optimization Strategy
计算机科学, 2020, 47(11A): 327-333. https://doi.org/10.11896/jsjkx.191200126
[12] 徐鹤, 吴昊, 李鹏.
面向物联网的时空数据处理算法设计
Design of Temporal-spatial Data Processing Algorithm for IoT
计算机科学, 2020, 47(11): 310-315. https://doi.org/10.11896/jsjkx.200400045
[13] 孙天旭, 赵蕴龙, 练作为, 孙毅, 蔡月啸.
基于时空数据的城市人流移动模式挖掘
Mobility Pattern Mining for People Flow Based on Spatio-Temporal Data
计算机科学, 2020, 47(10): 91-96. https://doi.org/10.11896/jsjkx.200100001
[14] 谷晓会,章国安.
SDN在车载网中的应用综述
Survey of SDN Applications in Vehicular Networks
计算机科学, 2020, 47(1): 237-244. https://doi.org/10.11896/jsjkx.190100178
[15] 刘长赟,杨宇迪,周丽华,赵丽红.
带有时间标签的流行社交位置发现
Discovering Popular Social Location with Time Label
计算机科学, 2019, 46(7): 186-194. https://doi.org/10.11896/j.issn.1002-137X.2019.07.029
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!