A Novel Graph Convolutional Gated Recurrent Unit Framework for Network-Based Traffic Prediction

被引:4
作者
Hussain, Basharat [1 ,2 ]
Afzal, Muhammad Khalil [1 ]
Anjum, Sheraz [3 ]
Rao, Imran [3 ]
Kim, Byung-Seo [4 ]
机构
[1] COMSATS Univ Islamabad, Dept Comp Sci, Wah Campus, Wah Cantt 47040, Pakistan
[2] Riphah Int Univ, Fac Comp, Islamabad 44000, Pakistan
[3] Blue Brackets Technol Pvt Ltd, Islamabad 44000, Pakistan
[4] Hongik Univ, Dept Software & Commun Engn, Sejong 30016, South Korea
基金
新加坡国家研究基金会;
关键词
Intelligent transportation system; traffic flow prediction; traffic network graph; gated recurrent unit; hyperparameters optimization; deep learning; FLOW;
D O I
10.1109/ACCESS.2023.3333938
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
A Smart City is characterized mainly as an efficient, technologically advanced, green, and socially informed city. An intelligent transportation system (ITS) is a subset area of smart cities that enhances the safety and mobility of road vehicles. It essentially makes travel more convenient, time-efficient and improves the citizens' quality of life. Accurate and real-time traffic prediction enables law enforcement agencies with well-informed about traffic congestion. However, accurate traffic prediction has been considered a challenging issue. Traffic prediction has restrictions on road network topology and the patterns of dynamic change in time-series data. We propose a novel deep learning framework GCST-GRU, called graph convolutional Spatio-temporal gated recurrent unit, to determine the next traffic state from traffic data. The proposed model learns complex topological structures by capturing a) spatial dependencies from data by using the graph convolution operator, and b) temporal dependencies by using the GRU neural network. Experimental results demonstrate that our framework can obtain complex Spatio-temporal correlations efficiently from the traffic network and perform better than state-of-the-art baseline models on a real-world traffic dataset. The graphical visualization by using convolution operation over the neural network shows that the model outperforms the reachability of the 3-hops neighbor effect in the traffic data graph. Additionally, the training time of the proposed framework is better than the existing state-of-the-art deep learning studies.
引用
收藏
页码:130102 / 130118
页数:17
相关论文
共 36 条
[1]   Exploiting dynamic spatio-temporal graph convolutional neural networks for citywide traffic flows prediction [J].
Ali, Ahmad ;
Zhu, Yanmin ;
Zakarya, Muhammad .
NEURAL NETWORKS, 2022, 145 :233-247
[2]  
Chen C, 2019, AAAI CONF ARTIF INTE, P485
[3]  
Cho KYHY, 2014, Arxiv, DOI [arXiv:1406.1078, 10.48550/arXiv.1406.1078.]
[4]   Traffic Graph Convolutional Recurrent Neural Network: A Deep Learning Framework for Network-Scale Traffic Learning and Forecasting [J].
Cui, Zhiyong ;
Henrickson, Kristian ;
Ke, Ruimin ;
Wang, Yinhai .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2020, 21 (11) :4883-4894
[5]  
Graves A, 2012, STUD COMPUT INTELL, V385, P1, DOI [10.1162/neco.1997.9.1.1, 10.1007/978-3-642-24797-2]
[6]   Optimized Graph Convolution Recurrent Neural Network for Traffic Prediction [J].
Guo, Kan ;
Hu, Yongli ;
Qian, Zhen ;
Liu, Hao ;
Zhang, Ke ;
Sun, Yanfeng ;
Gao, Junbin ;
Yin, Baocai .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2021, 22 (02) :1138-1149
[7]  
Henaff M, 2015, Arxiv, DOI arXiv:1506.05163
[8]   Diffusion Convolutional Recurrent Neural Network with Rank Influence Learning for Traffic Forecasting [J].
Huang, Yujun ;
Weng, Yunpeng ;
Yu, Shuai ;
Chen, Xu .
2019 18TH IEEE INTERNATIONAL CONFERENCE ON TRUST, SECURITY AND PRIVACY IN COMPUTING AND COMMUNICATIONS/13TH IEEE INTERNATIONAL CONFERENCE ON BIG DATA SCIENCE AND ENGINEERING (TRUSTCOM/BIGDATASE 2019), 2019, :678-685
[9]   Intelligent Traffic Flow Prediction Using Optimized GRU Model [J].
Hussain, Basharat ;
Afzal, Muhammad Khalil ;
Ahmad, Shafiq ;
Mostafa, Almetwally M. .
IEEE ACCESS, 2021, 9 :100736-100746
[10]   Rainfall-integrated traffic speed prediction using deep learning method [J].
Jia, Yuhan ;
Wu, Jianping ;
Ben-Akiva, Moshe ;
Seshadri, Ravi ;
Du, Yiman .
IET INTELLIGENT TRANSPORT SYSTEMS, 2017, 11 (09) :531-536