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 条
  • [31] Adaptive spatial-temporal graph attention networks for traffic flow forecasting
    Xiangyuan Kong
    Jian Zhang
    Xiang Wei
    Weiwei Xing
    Wei Lu
    Applied Intelligence, 2022, 52 : 4300 - 4316
  • [32] Spatial-Temporal Synchronous Graph Convolutional Networks: A New Framework for Spatial-Temporal Network Data Forecasting
    Song, Chao
    Lin, Youfang
    Guo, Shengnan
    Wan, Huaiyu
    THIRTY-FOURTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THE THIRTY-SECOND INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE AND THE TENTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2020, 34 : 914 - 921
  • [33] Spatial-Temporal Convolutional Graph Attention Networks for Citywide Traffic Flow Forecasting
    Zhang, Xiyue
    Huang, Chao
    Xu, Yong
    Xia, Lianghao
    CIKM '20: PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, 2020, : 1853 - 1862
  • [34] Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting
    Guo, Shengnan
    Lin, Youfang
    Feng, Ning
    Song, Chao
    Wan, Huaiyu
    THIRTY-THIRD AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FIRST INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE / NINTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2019, : 922 - 929
  • [35] Dynamic spatial-temporal graph convolutional recurrent networks for traffic flow forecasting
    Xia, Zhichao
    Zhang, Yong
    Yang, Jielong
    Xie, Linbo
    EXPERT SYSTEMS WITH APPLICATIONS, 2024, 240
  • [36] Dynamic Spatial-Temporal Perception Graph Convolutional Networks for Traffic Flow Forecasting
    Cao, Jingsi
    Liu, Weibin
    Xing, Weiwei
    PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2024, PT II, 2025, 15032 : 347 - 360
  • [37] Spatial-Temporal Dynamic Graph Convolutional Neural Network for Traffic Prediction
    Xiao, Wenjuan
    Wang, Xiaoming
    IEEE ACCESS, 2023, 11 : 97920 - 97929
  • [38] 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
  • [39] 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
  • [40] 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