Computer Science ›› 2024, Vol. 51 ›› Issue (4): 151-157.doi: 10.11896/jsjkx.230100066

• Database & Big Data & Data Science • Previous Articles     Next Articles

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

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

CLC Number: 

  • 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.
[1] ZHANG Mingdao, ZHOU Xin, WU Xiaohong, QING Linbo, HE Xiaohai. Unified Fake News Detection Based on Semantic Expansion and HDGCN [J]. Computer Science, 2024, 51(4): 299-306.
[2] LI Ke, YANG Ling, ZHAO Yanbo, CHEN Yonglong, LUO Shouxi. EGCN-CeDML:A Distributed Machine Learning Framework for Vehicle Driving Behavior Prediction [J]. Computer Science, 2023, 50(9): 318-330.
[3] 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.
[4] CHENG Haiyang, ZHANG Jianxin, SUN Qisen, ZHANG Qiang, WEI Xiaopeng. Deep Cross-modal Information Fusion Network for Stock Trend Prediction [J]. Computer Science, 2023, 50(5): 128-136.
[5] YANG Ying, ZHANG Fan, LI Tianrui. Aspect-based Sentiment Analysis Based on Dual-channel Graph Convolutional Network with Sentiment Knowledge [J]. Computer Science, 2023, 50(5): 230-237.
[6] YIN Heng, ZHANG Fan, LI Tianrui. Short-time Traffic Flow Forecasting Based on Multi-adjacent Graph and Multi-head Attention Mechanism [J]. Computer Science, 2023, 50(4): 40-46.
[7] WANG Yali, ZHANG Fan, YU Zeng, LI Tianrui. Aspect-level Sentiment Classification Based on Interactive Attention and Graph Convolutional Network [J]. Computer Science, 2023, 50(4): 196-203.
[8] ZHOU Mingqiang, DAI Kailang, WU Quanwang, ZHU Qingsheng. Attention-aware Multi-channel Graph Convolutional Rating Prediction Model for Heterogeneous Information Networks [J]. Computer Science, 2023, 50(3): 129-138.
[9] LI Shuai, XU Bin, HAN Yike, LIAO Tongxin. SS-GCN:Aspect-based Sentiment Analysis Model with Affective Enhancement and Syntactic Enhancement [J]. Computer Science, 2023, 50(3): 3-11.
[10] CAO Jinjuan, QIAN Zhong, LI Peifeng. End-to-End Event Factuality Identification with Joint Model [J]. Computer Science, 2023, 50(2): 292-299.
[11] KOU Jiaying, ZHAO Weidong, LIU Xianhui. Method of Document Level Relation Extraction Based on Fusion of Relational Transfer Information Using Double Graph [J]. Computer Science, 2023, 50(12): 229-235.
[12] ZHANG Longji, ZHAO Hui. Aspect-level Sentiment Analysis Integrating Syntactic Distance and Aspect-attention [J]. Computer Science, 2023, 50(12): 262-269.
[13] HE Minglong, ZHAO Kun, LI Weihua, LI Chuan. Antigenicity Prediction of Influenza A/H3N2 Based on Graph Convolutional Networks [J]. Computer Science, 2023, 50(11A): 230100113-6.
[14] JIN Bowen, WANG Qingmei, HU Chengzuo, WEI Jiacheng. Global Feature Enhanced for Session-based Recommendation [J]. Computer Science, 2023, 50(11A): 220800205-8.
[15] YANG Xianming, ZHAN Xianchun, CHEN Hengliang, DING Haiyan. Diagnosis Prediction Based on Graph Convolutional Network and Attention Mechanism [J]. Computer Science, 2023, 50(11A): 221100232-6.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!