计算机科学 ›› 2024, Vol. 51 ›› Issue (4): 151-157.doi: 10.11896/jsjkx.230100066

• 数据库&大数据&数据科学 • 上一篇    下一篇

基于双路先验自适应图神经常微分方程的交通流预测

袁蓉, 彭莉兰, 李天瑞, 李崇寿   

  1. 西南交通大学计算机与人工智能学院 成都611756
    可持续城市交通智能化教育部工程研究中心 成都611756
  • 收稿日期:2023-01-16 修回日期:2023-05-29 出版日期:2024-04-15 发布日期:2024-04-10
  • 通讯作者: 李崇寿(lics@swjtu.edu.cn)
  • 作者简介:(yrong@my.swjtu.edu.cn)
  • 基金资助:
    国家自然科学基金(62202395,62176221);四川省自然科学基金(2022NSFSC0930);中央高校基本科研业务费专项资金(2682022CX067)

Traffic Flow Prediction Model Based on Dual Prior-adaptive Graph Neural ODE Network

YUAN Rong, PENG Lilan, LI Tianrui, LI Chongshou   

  1. School of Computing and Artificial Intelligence,Southwest Jiaotong University,Chengdu 611756,China
    Engineering Research Center of Sustainable Urban Intelligent Transportation,Ministry of Education,Chengdu 611756,China
  • Received:2023-01-16 Revised:2023-05-29 Online:2024-04-15 Published:2024-04-10
  • Supported by:
    National Natural Science Foundation of China(62202395,62176221),Natural Science Foundation of Sichuan Province,China(2022NSFSC0930) and Fundamental Research Funds for the Central Universities of Ministry of Education of China(2682022CX067).

摘要: 准确的交通流量预测是智能交通系统不可或缺的组成部分。近年来,图神经网络在交通流预测任务中取得了较好的预测结果。然而,图神经网络的信息传递是不连续的潜在状态传播,且随着网络层数的增加存在过平滑的问题,这限制了模型捕获远距离节点的空间依赖关系的能力。同时,在表示道路网络的空间关系时,现有方法大多仅使用先验知识构建的预定义图或仅使用路网状况构建的自适应图,忽略了两类图结合的方式。针对上述问题,提出了一种基于双路先验自适应图神经常微分方程的交通流预测模型。利用时间卷积网络捕获序列的时间相关性,使用先验自适应图融合模块表示道路网络的空间关系,并通过基于张量乘法的神经常微分方程以连续的方式传播复杂的时空特征。最后,在美国加利福尼亚州4个公开的高速公路流量数据集上进行对比实验,结果表明所提模型的预测效果优于现有的10种对比方法。

关键词: 交通预测, 先验自适应图, 图卷积神经网络, 神经常微分方程, 张量乘法

Abstract: Accurate traffic flow prediction is an indispensable part of intelligent transportation system.In recent years,graph neural networks have generated effective results in traffic flow prediction tasks.However,the information transfer of graph neural network is discontinuous latent state propagation,and there is an over-smoothing problem as the number of network layers increases,which limits the ability of the model to capture the spatial dependencies of distant nodes.At the same time,when representing the spatial relationship of the road network,most of the existing methods only use the predefined graph constructed by prior knowledge or the adaptive graph constructed only by the road network conditions,ignoring the combination of those two graphs.Aiming at solving the above problems,this paper proposes a traffic flow prediction model based on a dual prior adaptive graph neural ordinary differential equation.Temporal convolutional network are utilized to capture the temporal correlation of sequences,a priori adaptive graph fusion module is used to represent the road network,and complex spatio-temporal features are propagated in a continuous manner through tensor multiplication-based nerual ODEs.Finally,experiments are carried out on four public data sets of highway traffic in California,USA.Experimental results show that the prediction performance of the model is better than that of the existing ten methods.

Key words: Traffic forecasting, Prior adaptive graph, Graph convolutional network, Neural ordinary differential equations, Tensor multiplication

中图分类号: 

  • TP181
[1]CHEN Q,SONG X,YAMADA H,et al.Learning deep representation from big and heterogeneous data for traffic accident inference[C]//Proceedings of the 30th AAAI Conference on Artificial Intelligence.2016.
[2]GONG Y,LI Z,ZHANG J,et al.Potential passenger flow prediction:A novel study for urban transportation development[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2020:4020-4027.
[3]JAYARAJAH K,TAN A,MISRA A.Understanding the interdependency of land use and mobility for urban planning[C]//Proceedings of the 2018 ACM International Joint Conference and 2018 International Symposium on Pervasive and Ubiquitous Computing and Wearable Computers.2018:1079-1087.
[4]YU B,YIN H,ZHU Z.Spatio-temporal graph convolutional net-works:a deep learning framework for traffic forecasting[C]//Proceedings of the 27th International Joint Conference on Artificial Intelligence.2018:3634-3640.
[5]LI Y,YU R,SHAHABI C,et al.Diffusion convolutional recurrent neural network:Data-driven traffic forecasting[J].arXiv:1707.01926,2017.
[6]WU Z,PAN S,LONG G,et al.Graph WaveNet for Deep Spatial-Temporal Graph Modeling[C]//The 28th International Joint Conference on Artificial Intelligence(IJCAI).2019.
[7]BAI S,KOLTER J Z,KOLTUN V.An empirical evaluation of generic convolutional and recurrent networks for sequence mo-deling[J].arXiv:1803.01271,2018.
[8]WILLIAMS B M,DURVASULA P K,BROWN D E.Urbanfreeway traffic flow prediction:application of seasonal autoregressive integrated moving average and exponential smoothing models[J].Transportation Research Record,1998,1644(1):132-141.
[9]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.
[10]LIPPI M,BERTINI M,FRASCONI P.Short-term traffic flow forecasting:An experimental comparison of time-seriesanalysis and supervised learning[J].IEEE Transactions on Intelligent Transportation Systems,2013,14(2):871-882.
[11]HABTEMICHAEL F G,CETIN M.Short-term traffic flow rate forecasting based on identifying similar traffic patterns[J].Transportation Research Part C:Emerging Technologies,2016,66:61-78.
[12]ZHAO Z,CHEN W,WU X,et al.LSTM network:a deep lear-ning approach for short-term traffic forecast[J].IET Intelligent Transport Systems,2017,11(2):68-75.
[13]FU R,ZHANG Z,LI L.Using LSTM and GRU neural network methods for traffic flow prediction[C]//Proceedings of the 31st Youth Academic Annual Conference of Chinese Association of Automation(YAC).IEEE,2016:324-328.
[14]SUTSKEVER I,VINYALS O,LE Q V.Sequence to Sequence Learning with Neural Networks[J].arXiv:1409.3215,2014.
[15]ZHANG J,ZHENG Y,QI D.Deep spatio-temporal residual networks for citywide crowd flows prediction[C]//Proceedings of the 31th AAAI Conference on Artificial Intelligence.2017.
[16]JIANG W,ZHANG L.Geospatial data to images:A deep-lear-ning framework for traffic forecasting[J].Tsinghua Science and Technology,2018,24(1):52-64.
[17]YU H,WU Z,WANG S,et al.Spatiotemporal recurrent convolutional networks for traffic prediction in transportation networks[J].Sensors,2017,17(7):1501.
[18]REN J H,ZHU Y,MENG X F,et al.Predicting Citywide Traffic Flow Using Dynamic Spatial-temporal Neural Networks[J].Journal of Chinese Computer Systems,2023,44(3):529-535.
[19]ZHAO L,SONG Y,ZHANG C,et al.T-gcn:A temporal graph convolutional network for traffic prediction[J].IEEE Transactions on Intelligent Transportation Systems,2019,21(9):3848-3858.
[20]YU B,LEE Y,SOHN K.Forecasting road traffic speeds by considering area-wide spatio-temporal dependencies based on a graph convolutional neural network(GCN)[J].Transportation Research Part C:Emerging Technologies,2020,114:189-204.
[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,33(1):922-929.
[22]SONG C,LIN Y,GUO S,et al.Spatial-temporal synchronous graph convolutional networks:A new framework for spatial-temporal network data forecasting[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2020,34(1):914-921.
[23]LI M,ZHU Z.Spatial-temporal fusion graph neural networksfor traffic flow forecasting[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2021,35(5):4189-4196.
[24]CHEN R T Q,RUBANOVA Y,BETTENCOURT J,et al.Neural ordinary differential equations[J].arXiv:1806.07366,2018.
[25]FANG Z,LONG Q,SONG G,et al.Spatial-temporal graph ode networks for traffic flow forecasting[C]//Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery & Data Mining.2021:364-373.
[26]BERNDT D J,CLIFFORD J.Using dynamic time warping tofind patterns in time series[C]//KDD workshop.1994,10(16):359-370.
[27]LI Q,HAN Z,WU X M.Deeper insights into graph convolutional networks for semi-supervised learning[C]//Proceedings of the 32th AAAI Conference on Artificial Intelligence.2018.
[28]XU K,LIC,TIAN Y,et al.Representation learning on graphs with jumping knowledge networks[C]//International Confe-rence on Machine Learning.PMLR,2018:5453-5462.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!