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 条
  • [1] Adaptive Multi-receptive Field Spatial-Temporal Graph Convolutional Network for Traffic Forecasting
    Wang, Xing
    Zhao, Juan
    Zhu, Lin
    Zhou, Xu
    Li, Zhao
    Feng, Junlan
    Deng, Chao
    Zhang, Yong
    2021 IEEE GLOBAL COMMUNICATIONS CONFERENCE (GLOBECOM), 2021,
  • [2] STLGCN: Spatial-Temporal Graph Convolutional Network for Long Term Traffic Forecasting
    Chen, Xuewen
    Peng, Peng
    Tang, Haina
    BIG DATA TECHNOLOGIES AND APPLICATIONS, EAI INTERNATIONAL CONFERENCE, BDTA 2023, 2024, 555 : 49 - 61
  • [3] Spatial-Temporal Dynamic Graph Convolutional Network With Interactive Learning for Traffic Forecasting
    Liu, Aoyu
    Zhang, Yaying
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, 25 (07) : 7645 - 7660
  • [4] Spatial-temporal hypergraph convolutional network for traffic forecasting
    Zhao, Zhenzhen
    Shen, Guojiang
    Zhou, Junjie
    Jin, Junchen
    Kong, Xiangjie
    PEERJ COMPUTER SCIENCE, 2023, 9
  • [5] Spatial-temporal hypergraph convolutional network for traffic forecasting
    Zhao Z.
    Shen G.
    Zhou J.
    Jin J.
    Kong X.
    PeerJ Computer Science, 2023, 9
  • [6] Local-Global Spatial-Temporal Graph Convolutional Network for Traffic Flow Forecasting
    Zong, Xinlu
    Chen, Zhen
    Yu, Fan
    Wei, Siwei
    ELECTRONICS, 2024, 13 (03)
  • [7] Adaptive Spatial-Temporal Convolution Network for Traffic Forecasting
    Li, Zhao
    Zhang, Yong
    Zhang, Zhao
    Wang, Xing
    Zhu, Lin
    KNOWLEDGE SCIENCE, ENGINEERING AND MANAGEMENT, PT II, 2022, 13369 : 287 - 299
  • [8] PGCN: Progressive Graph Convolutional Networks for Spatial-Temporal Traffic Forecasting
    Shin, Yuyol
    Yoon, Yoonjin
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, 25 (07) : 7633 - 7644
  • [9] 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
  • [10] Hybrid spatial-temporal graph neural network for traffic forecasting
    Wang, Peng
    Feng, Longxi
    Zhu, Yijie
    Wu, Haopeng
    INFORMATION FUSION, 2025, 118