AdpSTGCN: Adaptive spatial-temporal graph convolutional network for traffic forecasting

被引:3
|
作者
Zhang, Xudong [1 ]
Chen, Xuewen [1 ]
Tang, Haina [1 ]
Wu, Yulei [2 ]
Shen, Hanji [3 ]
Li, Jun [3 ]
机构
[1] Univ Chinese Acad Sci, Sch Artificial Intelligence, Beijing 100049, Peoples R China
[2] Univ Bristol, Sch Elect Elect & Mech Engn, Bristol BS8 1UB, England
[3] Chinese Acad Sci, Comp Network Informat Ctr, Beijing 100083, Peoples R China
关键词
Traffic forecasting; Graph structure learning; Adaptive graph convolution; Spatial-temporal graph modeling;
D O I
10.1016/j.knosys.2024.112295
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Traffic flow forecasting plays a crucial role in applications such as intelligent transportation systems. Despite significant research in this field, the current methods have limitations that hinder the realization of highly accurate predictions. Existing GCN-based approaches typically rely on a definite graph structure derived from a physical topology or learned from node features, which is insufficient for building intricate spatial relationships among nodes. To address this challenge, we propose an adaptive spatial-temporal graph convolutional network for traffic forecasting. Our approach exploits a multi-head attention mechanism to construct multi-view feature graphs. We then introduce an adaptive graph convolution method to dynamically aggregate and propagate information from both the topology graph and multi-view feature graphs, which are capable of capturing complex spatial correlations across diverse proximity ranges. Furthermore, we designed a cascaded structural framework that combines temporal information with node features using gated dilated causal convolution to ensure the integrated modeling of spatial-temporal dynamics in traffic flow. Experiments on real-world datasets demonstrate that our proposed method outperforms the current mainstream methods, achieving better performance in traffic flow forecasting. The code is available at https://github.com/dhxdla/AdpSTGCN.git.
引用
收藏
页数:11
相关论文
共 50 条
  • [41] Dynamic Spatial-Temporal Heterogeneous Graph Convolutional Network for Traffic Prediction
    Jin, Hengqing
    Pu, Lipeng
    Lecture Notes in Electrical Engineering, 2024, 1253 LNEE : 60 - 68
  • [42] DSTGCN: Dynamic Spatial-Temporal Graph Convolutional Network for Traffic Prediction
    Hu, Jia
    Lin, Xianghong
    Wang, Chu
    IEEE SENSORS JOURNAL, 2022, 22 (13) : 13116 - 13124
  • [43] KST-GCN: A Knowledge-Driven Spatial-Temporal Graph Convolutional Network for Traffic Forecasting
    Zhu, Jiawei
    Han, Xing
    Deng, Hanhan
    Tao, Chao
    Zhao, Ling
    Wang, Pu
    Lin, Tao
    Li, Haifeng
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (09) : 15055 - 15065
  • [44] STAGCN: Spatial-Temporal Attention Graph Convolution Network for Traffic Forecasting
    Gu, Yafeng
    Deng, Li
    MATHEMATICS, 2022, 10 (09)
  • [45] Decoupled Dynamic Spatial-Temporal Graph Neural Network for Traffic Forecasting
    Shao, Zezhi
    Zhang, Zhao
    Wei, Wei
    Wang, Fei
    Xu, Yongjun
    Cao, Xin
    Jensen, Christian S.
    PROCEEDINGS OF THE VLDB ENDOWMENT, 2022, 15 (11): : 2733 - 2746
  • [46] Graph learning-based spatial-temporal graph convolutional neural networks for traffic forecasting
    Hu, Na
    Zhang, Dafang
    Xie, Kun
    Liang, Wei
    Hsieh, Meng-Yen
    CONNECTION SCIENCE, 2022, 34 (01) : 429 - 448
  • [47] Efficient Adaptive Spatial-Temporal Attention Network for Traffic Flow Forecasting
    Su, Hongyang
    Wang, Xiaolong
    Chen, Qingcai
    Qin, Yang
    MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES: RESEARCH TRACK, ECML PKDD 2023, PT V, 2023, 14173 : 205 - 220
  • [48] Mask Adaptive Spatial-Temporal Recurrent Neural Network for Traffic Forecasting
    Hu, Xingbang
    Zhang, Shuo
    Zhang, Wenbo
    Huang, Hejiao
    ADVANCES IN KNOWLEDGE DISCOVERY AND DATA MINING, PT V, PAKDD 2024, 2024, 14649 : 259 - 270
  • [49] HIT-GCN: Spatial-Temporal Graph Convolutional Network Embedded with Heterogeneous Information of Road Network for Traffic Forecasting
    Xiong, Haitao
    Shen, Guojiang
    Lan, Xiang
    Yuan, Haopeng
    Kong, Xiangjie
    ELECTRONICS, 2023, 12 (06)
  • [50] Multi-Hierarchical Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting
    Li, Zilong
    Ren, Qianqian
    Chen, Long
    Sui, Xiaohong
    Li, Jinbao
    2022 26TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2022, : 4913 - 4919