GT-LSTM: A spatio-temporal ensemble network for traffic flow prediction

被引:31
作者
Luo, Yong [1 ,2 ]
Zheng, Jianying [1 ,2 ,3 ]
Wang, Xiang [1 ,2 ]
Tao, Yanyun [1 ,2 ]
Jiang, Xingxing [1 ,2 ]
机构
[1] Soochow Univ, Sch Rail Transportat, Suzhou 215131, Peoples R China
[2] Intelligent Urban Rail Engn Res Ctr Jiangsu Prov, Suzhou 215131, Peoples R China
[3] Suzhou City Univ, Sch Smart Mfg & Intelligent Transportat, Suzhou 215104, Peoples R China
基金
中国国家自然科学基金;
关键词
Traffic flow prediction; Deep learning; Ensemble network; Temporal convolutional network; Long short-term memory; NEURAL-NETWORK; CONVOLUTIONAL NETWORK; GRAPH;
D O I
10.1016/j.neunet.2023.12.016
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Traffic flow prediction plays an instrumental role in modern intelligent transportation systems. Numerous existing studies utilize inter-embedded fusion routes to extract the intrinsic patterns of traffic flow with a single temporal learning approach, which relies heavily on constructing graphs and has low training efficiency. Different from existing studies, this paper proposes a spatio-temporal ensemble network that aims to leverage the strengths of different sequential capturing approaches to obtain the intrinsic dependencies of traffic flow. Specifically, we propose a novel model named graph temporal convolutional long short-term memory network (GTLSTM), which mainly consists of features splicing and patterns capturing. In features splicing, the spatial dependencies of traffic flow are captured by employing self-adaptive graph convolutional network (GCN), and a non-inter-embedded approach is designed to integrate the spatial and temporal states. Further, the aggregated spatio-temporal states are fed into patterns capturing, which can effectively exploit the advantages of temporal convolutional network (TCN) and bidirectional long short-term memory network (Bi-LSTM) to extract the intrinsic patterns of traffic flow. Extensive experiments conducted on four real-world datasets demonstrate that the proposed network obtains excellent performance in both forecasting accuracy and training efficiency.
引用
收藏
页码:251 / 262
页数:12
相关论文
共 37 条
[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]   DGInet: Dynamic graph and interaction-aware convolutional network for vehicle trajectory prediction [J].
An, Jiyao ;
Liu, Wei ;
Liu, Qingqin ;
Guo, Liang ;
Ren, Ping ;
Li, Tao .
NEURAL NETWORKS, 2022, 151 :336-348
[3]   IGAGCN: Information geometry and attention-based spatiotemporal graph convolutional networks for traffic flow prediction [J].
An, Jiyao ;
Guo, Liang ;
Liu, Wei ;
Fu, Zhiqiang ;
Ren, Ping ;
Liu, Xinzhi ;
Li, Tao .
NEURAL NETWORKS, 2021, 143 :355-367
[4]  
Bai L, 2020, ADV NEUR IN, V33
[5]  
Bai SJ, 2018, Arxiv, DOI [arXiv:1803.01271, DOI 10.48550/ARXIV.1803.01271, 10.48550/arXiv.1803.01271]
[6]  
Berndt D., 1994, Proceedings of the AAAI
[7]   A Hybrid Prediction Method for Realistic Network Traffic With Temporal Convolutional Network and LSTM [J].
Bi, Jing ;
Zhang, Xiang ;
Yuan, Haitao ;
Zhang, Jia ;
Zhou, MengChu .
IEEE TRANSACTIONS ON AUTOMATION SCIENCE AND ENGINEERING, 2022, 19 (03) :1869-1879
[8]   A graph CNN-LSTM neural network for short and long-term traffic forecasting based on trajectory data [J].
Bogaerts, Toon ;
Masegosa, Antonio D. ;
Angarita-Zapata, Juan S. ;
Onieva, Enrique ;
Hellinckx, Peter .
TRANSPORTATION RESEARCH PART C-EMERGING TECHNOLOGIES, 2020, 112 :62-77
[9]   Traffic forecasting with graph spatial-temporal position recurrent network [J].
Chen, Yibi ;
Li, Kenli ;
Yeo, Chai Kiat ;
Li, Keqin .
NEURAL NETWORKS, 2023, 162 :340-349
[10]   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