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 条
  • [21] Spatial-Temporal Dilated and Graph Convolutional Network for traffic prediction
    Yang, Guoliang
    Wen, Junlin
    Yu, Dinglin
    Zhang, Shuo
    2020 CHINESE AUTOMATION CONGRESS (CAC 2020), 2020, : 802 - 806
  • [22] Traffic Flow Forecasting with Spatial-Temporal Graph Diffusion Network
    Zhang, Xiyue
    Huang, Chao
    Xu, Yong
    Xia, Lianghao
    Dai, Peng
    Bo, Liefeng
    Zhang, Junbo
    Zheng, Yu
    35th AAAI Conference on Artificial Intelligence, AAAI 2021, 2021, 17A : 15008 - 15015
  • [23] Spatial-Temporal Bipartite Graph Attention Network for Traffic Forecasting
    Lakma, Dimuthu
    Perera, Kushani
    Borovica-Gajic, Renata
    Karunasekera, Shanika
    ADVANCES IN KNOWLEDGE DISCOVERY AND DATA MINING, PT II, PAKDD 2024, 2024, 14646 : 68 - 80
  • [24] Hybrid spatial-temporal graph neural network for traffic forecasting
    Wang, Peng
    Feng, Longxi
    Zhu, Yijie
    Wu, Haopeng
    INFORMATION FUSION, 2025, 118
  • [25] Traffic forecasting with graph spatial-temporal position recurrent network
    Chen, Yibi
    Li, Kenli
    Yeo, Chai Kiat
    Li, Keqin
    NEURAL NETWORKS, 2023, 162 : 340 - 349
  • [26] Traffic Flow Forecasting with Spatial-Temporal Graph Diffusion Network
    Zhang, Xiyue
    Huang, Chao
    Xu, Yong
    Xia, Lianghao
    Dai, Peng
    Bo, Liefeng
    Zhang, Junbo
    Zheng, Yu
    THIRTY-FIFTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THIRTY-THIRD CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE AND THE ELEVENTH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2021, 35 : 15008 - 15015
  • [27] Hybrid Spatial-Temporal Graph Convolutional Network for Long-Term Traffic Flow Forecasting
    Wu, Zihao
    Lou, Ping
    2023 IEEE 8TH INTERNATIONAL CONFERENCE ON BIG DATA ANALYTICS, ICBDA, 2023, : 224 - 229
  • [28] Dynamic Spatial-Temporal Graph Attention Graph Convolutional Network for Short-Term Traffic Flow Forecasting
    Tang, Cong
    Sun, Jingru
    Sun, Yichuang
    2020 IEEE INTERNATIONAL SYMPOSIUM ON CIRCUITS AND SYSTEMS (ISCAS), 2020,
  • [29] Adaptive spatial-temporal graph attention networks for traffic flow forecasting
    Kong, Xiangyuan
    Zhang, Jian
    Wei, Xiang
    Xing, Weiwei
    Lu, Wei
    APPLIED INTELLIGENCE, 2022, 52 (04) : 4300 - 4316
  • [30] Spatial-temporal Graph Transformer Network for Spatial-temporal Forecasting
    Dao, Minh-Son
    Zetsu, Koji
    Hoang, Duy-Tang
    Proceedings - 2024 IEEE International Conference on Big Data, BigData 2024, 2024, : 1276 - 1281