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 条
  • [31] TPST: A Traffic Flow Prediction Model Based on Spatial-Temporal Identity
    Hou, Yuchen
    Cao, Buqing
    Liu, Jianxun
    Li, Changyun
    Shi, Min
    CONCURRENCY AND COMPUTATION-PRACTICE & EXPERIENCE, 2025, 37 (4-5):
  • [32] Transfer Learning With Spatial-Temporal Graph Convolutional Network for Traffic Prediction
    Yao, Zhixiu
    Xia, Shichao
    Li, Yun
    Wu, Guangfu
    Zuo, Linli
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2023, 24 (08) : 8592 - 8605
  • [33] Spatial-Temporal Dynamic Graph Convolutional Neural Network for Traffic Prediction
    Xiao, Wenjuan
    Wang, Xiaoming
    IEEE ACCESS, 2023, 11 : 97920 - 97929
  • [34] Spatial-Temporal Tensor Graph Convolutional Network for Traffic Speed Prediction
    Xu, Xuran
    Zhang, Tong
    Xu, Chunyan
    Cui, Zhen
    Yang, Jian
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2023, 24 (01) : 92 - 103
  • [35] Dynamic Spatial-Temporal Heterogeneous Graph Convolutional Network for Traffic Prediction
    Jin, Hengqing
    Pu, Lipeng
    Lecture Notes in Electrical Engineering, 2024, 1253 LNEE : 60 - 68
  • [36] DSTGCN: Dynamic Spatial-Temporal Graph Convolutional Network for Traffic Prediction
    Hu, Jia
    Lin, Xianghong
    Wang, Chu
    IEEE SENSORS JOURNAL, 2022, 22 (13) : 13116 - 13124
  • [37] Multi-level spatial-temporal fusion neural network for traffic flow prediction
    Peng, Zhiying
    Yang, Yixue
    Zhao, Hao
    CLUSTER COMPUTING-THE JOURNAL OF NETWORKS SOFTWARE TOOLS AND APPLICATIONS, 2024, 27 (05): : 6689 - 6702
  • [38] Attention-based global and local spatial-temporal graph convolutional network for vehicle emission prediction
    Fei, Xihong
    Ling, Qiang
    NEUROCOMPUTING, 2023, 521 : 41 - 55
  • [39] Capturing spatial-temporal correlations with Attention based Graph Convolutional Network for network traffic prediction
    Guo, Yingya
    Peng, Yufei
    Hao, Run
    Tang, Xiang
    JOURNAL OF NETWORK AND COMPUTER APPLICATIONS, 2023, 220
  • [40] Network traffic prediction based on feature fusion spatio-temporal graph convolutional network
    Key Laboratory of Universal Wireless Communications, Ministry of Education, Beijing University of Posts and Telecommunications, Beijing
    100876, China
    不详
    100876, China
    Proc SPIE Int Soc Opt Eng,