LGTCN: A Spatial-Temporal Traffic Flow Prediction Model Based on Local-Global Feature Fusion Temporal Convolutional Network

被引:0
|
作者
Ye, Wei [1 ]
Kuang, Haoxuan [1 ]
Deng, Kunxiang [1 ]
Zhang, Dongran [2 ]
Li, Jun [1 ]
机构
[1] Sun Yat sen Univ, Sch Intelligent Syst Engn, Shenzhen 518107, Peoples R China
[2] China Mobile Internet Co Ltd, Guangzhou 510510, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2024年 / 14卷 / 19期
关键词
traffic flow prediction; spatial-temporal feature; local-global feature fusion; probabilistic sparse self-attention; temporal convolutional network;
D O I
10.3390/app14198847
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
High-precision traffic flow prediction facilitates intelligent traffic control and refined management decisions. Previous research has built a variety of exquisite models with good prediction results. However, they ignore the reality that traffic flows can propagate backwards on road networks when modeling spatial relationships, as well as associations between distant nodes. In addition, more effective model components for modeling temporal relationships remain to be developed. To address the above challenges, we propose a local-global features fusion temporal convolutional network (LGTCN) for spatio-temporal traffic flow prediction, which incorporates a bidirectional graph convolutional network, probabilistic sparse self-attention, and a multichannel temporal convolutional network. To extract the bidirectional propagation relationship of traffic flow on the road network, we improve the traditional graph convolutional network so that information can be propagated in multiple directions. In addition, in spatial global dimensions, we propose probabilistic sparse self-attention to effectively perceive global data correlations and reduce the computational complexity caused by the finite perspective graph. Furthermore, we develop a multichannel temporal convolutional network. It not only retains the temporal learning capability of temporal convolutional networks, but also corresponds each channel to a node, and it realizes the interaction of node features through output interoperation. Extensive experiments on four open access benchmark traffic flow datasets demonstrate the effectiveness of our model.
引用
收藏
页数:16
相关论文
共 50 条
  • [1] Local-Global Spatial-Temporal Graph Convolutional Network for Traffic Flow Forecasting
    Zong, Xinlu
    Chen, Zhen
    Yu, Fan
    Wei, Siwei
    ELECTRONICS, 2024, 13 (03)
  • [2] Local-global dynamic correlations based spatial-temporal convolutional network for traffic flow forecasting
    Zhang, Hong
    Gong, Lei
    Zhao, Tianxin
    Zhang, Xijun
    Wang, Hongyan
    High Technology Letters, 2024, 30 (04) : 370 - 379
  • [3] Local-global dynamic correlations based spatial-temporal convolutional network for traffic flow forecasting
    张红
    GONG Lei
    ZHAO Tianxin
    ZHANG Xijun
    WANG Hongyan
    High Technology Letters, 2024, 30 (04) : 370 - 379
  • [4] Spatial-Temporal Feature Fusion Network for Network Traffic Prediction
    Hong, Yu
    Zhou, Jianxin
    Zhou, Ning
    2023 3rd International Symposium on Computer Technology and Information Science, ISCTIS 2023, 2023, : 162 - 166
  • [5] A multi-feature spatial-temporal fusion network for traffic flow prediction
    Yan, Jiahe
    Li, Honghui
    Zhang, Dalin
    Bai, Yanhui
    Xu, Yi
    Han, Chengshan
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [6] STFGCN: Spatial-temporal fusion graph convolutional network for traffic prediction
    Li, Hao
    Liu, Jie
    Han, Shiyuan
    Zhou, Jin
    Zhang, Tong
    Chen, C. L. Philip
    EXPERT SYSTEMS WITH APPLICATIONS, 2024, 255
  • [7] Traffic Flow Prediction Based on Spatial-Temporal Attention Convolutional Neural Network
    Xia Y.
    Liu M.
    Xinan Jiaotong Daxue Xuebao/Journal of Southwest Jiaotong University, 2023, 58 (02): : 340 - 347
  • [8] Hierarchical Traffic Flow Prediction Based on Spatial-Temporal Graph Convolutional Network
    Wang, Hanqiu
    Zhang, Rongqing
    Cheng, Xiang
    Yang, Liuqing
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (09) : 16137 - 16147
  • [9] GSTNet: Global Spatial-Temporal Network for Traffic Flow Prediction
    Fang, Shen
    Zhang, Qi
    Meng, Gaofeng
    Xiang, Shiming
    Pan, Chunhong
    PROCEEDINGS OF THE TWENTY-EIGHTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, 2019, : 2286 - 2293
  • [10] Capturing Local and Global Spatial-Temporal Correlations of Spatial-Temporal Graph Data for Traffic Flow Prediction
    Cao, Shuqin
    Wu, Libing
    Zhang, Rui
    Li, Jianxin
    Wu, Dan
    2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2022,