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

被引:4
|
作者
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
相关论文
共 50 条
  • [1] A GT-LSTM Spatio-Temporal Approach for Winter Wheat Yield Prediction: From the Field Scale to County Scale
    Cheng, Enhui
    Wang, Fumin
    Peng, Dailiang
    Zhang, Bing
    Zhao, Bin
    Zhang, Wenjuan
    Hu, Jinkang
    Lou, Zihang
    Yang, Songlin
    Zhang, Hongchi
    Lv, Yulong
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [2] Spatio-Temporal AutoEncoder for Traffic Flow Prediction
    Liu, Mingzhe
    Zhu, Tongyu
    Ye, Junchen
    Meng, Qingxin
    Sun, Leilei
    Du, Bowen
    [J]. IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2023, 24 (05) : 5516 - 5526
  • [3] A Spatio-Temporal Graph Neural Network Approach for Traffic Flow Prediction
    Li, Yanbing
    Zhao, Wei
    Fan, Huilong
    [J]. MATHEMATICS, 2022, 10 (10)
  • [4] A Frequency-Aware Spatio-Temporal Network for Traffic Flow Prediction
    Peng, Shunfeng
    Shen, Yanyan
    Zhu, Yanmin
    Chen, Yuting
    [J]. DATABASE SYSTEMS FOR ADVANCED APPLICATIONS (DASFAA 2019), PT II, 2019, 11447 : 697 - 712
  • [5] A spatio-temporal sequence-to-sequence network for traffic flow prediction
    Cao, Shuqin
    Wu, Libing
    Wu, Jia
    Wu, Dan
    Li, Qingan
    [J]. INFORMATION SCIENCES, 2022, 610 : 185 - 203
  • [6] Multi-scale Spatio-temporal Attention Network for Traffic Flow Prediction
    Li, Minghao
    Li, Jinhong
    Ta, Xuxiang
    Bai, Yanbo
    Hao, Xinzhe
    [J]. ADVANCED INTELLIGENT COMPUTING TECHNOLOGY AND APPLICATIONS, PT II, ICIC 2024, 2024, 14876 : 294 - 305
  • [7] Federated Spatio-Temporal Traffic Flow Prediction Based on Graph Convolutional Network
    Wang, Hanqiu
    Zhang, Rongqing
    Cheng, Xiang
    Yang, Liuqing
    [J]. 2022 14TH INTERNATIONAL CONFERENCE ON WIRELESS COMMUNICATIONS AND SIGNAL PROCESSING, WCSP, 2022, : 221 - 225
  • [8] SPATIO-TEMPORAL GRAPH-TCN NEURAL NETWORK FOR TRAFFIC FLOW PREDICTION
    Ren, Hongjin
    Kang, Jinbiao
    Zhang, Ke
    [J]. 2022 19TH INTERNATIONAL COMPUTER CONFERENCE ON WAVELET ACTIVE MEDIA TECHNOLOGY AND INFORMATION PROCESSING (ICCWAMTIP), 2022,
  • [9] Spatio-Temporal Graph-TCN Neural Network for Traffic Flow Prediction
    Ren, Hongjin
    Kang, Jinbiao
    Zhang, Ke
    [J]. 2022 19th International Computer Conference on Wavelet Active Media Technology and Information Processing, ICCWAMTIP 2022, 2022,
  • [10] MSSTN: a multi-scale spatio-temporal network for traffic flow prediction
    Song, Yun
    Bai, Xinke
    Fan, Wendong
    Deng, Zelin
    Jiang, Cong
    [J]. INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS, 2024, 15 (07) : 2827 - 2841