A Spatio-Temporal Graph Neural Network Approach for Traffic Flow Prediction

被引:8
|
作者
Li, Yanbing [1 ]
Zhao, Wei [2 ]
Fan, Huilong [2 ]
机构
[1] Xinjiang Univ, Coll Informat Sci & Engn, Sch Cyber Sci & Engn, Urumqi 830046, Peoples R China
[2] Cent South Univ, Sch Comp Sci & Engn, Changsha 410075, Peoples R China
基金
中国国家自然科学基金;
关键词
traffic flow; deep learning; graph neural network; forecasting; DEEP; MODEL;
D O I
10.3390/math10101754
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
The accuracy of short-term traffic flow prediction is one of the important issues in the construction of smart cities, and it is an effective way to solve the problem of traffic congestion. Most previous studies could not effectively mine the potential relationship between the temporal and spatial dimensions of traffic data flow. Due to the large variability in the traffic flow data of road conditions, we analyzed it with "dynamic", using a dynamic-aware graph neural network model for the hidden relationships between space-time in the deep learning segment. In this paper, we propose a dynamic perceptual graph neural network model for the temporal and spatial hidden relationships of deep learning segments. This model mixes temporal features and spatial features with graphs and expresses them. The temporal features and spatial features are connected to each other to learn potential relationships, so as to more accurately predict the traffic speed in the future time period, we performed experiments on real data sets and compared with some baseline models. The experiments show that the method proposed in this paper has certain advantages.
引用
收藏
页数:14
相关论文
共 50 条
  • [21] STANN: A Spatio-Temporal Attentive Neural Network for Traffic Prediction
    He, Zhixiang
    Chow, Chi-Yin
    Zhang, Jia-Dong
    IEEE ACCESS, 2019, 7 : 4795 - 4806
  • [22] Deep spatio-temporal neural network based on interactive attention for traffic flow prediction
    Hui Zeng
    Zhiying Peng
    XiaoHui Huang
    Yixue Yang
    Rong Hu
    Applied Intelligence, 2022, 52 : 10285 - 10296
  • [23] Deep spatio-temporal neural network based on interactive attention for traffic flow prediction
    Zeng, Hui
    Peng, Zhiying
    Huang, XiaoHui
    Yang, Yixue
    Hu, Rong
    APPLIED INTELLIGENCE, 2022, 52 (09) : 10285 - 10296
  • [24] Deep spatio-temporal graph convolutional network for traffic accident prediction
    Yu, Le
    Du, Bowen
    Hu, Xiao
    Sun, Leilei
    Han, Liangzhe
    Lv, Weifeng
    NEUROCOMPUTING, 2021, 423 (423) : 135 - 147
  • [25] Traffic Prediction Model Based on Spatio-temporal Graph Attention Network
    Chen, Jing
    Wang, Linkai
    Wang, Wei
    Song, Ruizhuo
    2022 4TH INTERNATIONAL CONFERENCE ON CONTROL AND ROBOTICS, ICCR, 2022, : 428 - 432
  • [26] Spatio-Temporal Pivotal Graph Neural Networks for Traffic Flow Forecasting
    Kong, Weiyang
    Guo, Ziyu
    Liu, Yubao
    THIRTY-EIGHTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, VOL 38 NO 8, 2024, : 8627 - 8635
  • [27] STLGRU: Spatio-Temporal Lightweight Graph GRU for Traffic Flow Prediction
    Bhaumik, Kishor Kumar
    Niloy, Fahim Faisal
    Mahmud, Saif
    Woo, Simon S.
    ADVANCES IN KNOWLEDGE DISCOVERY AND DATA MINING, PT VI, PAKDD 2024, 2024, 14650 : 288 - 299
  • [28] Dynamic Spatio-temporal traffic flow prediction based on multi fusion graph attention network
    Cheng, Manru
    Jiang, Guo-Ping
    Song, Yurong
    Yang, Chen
    2022 41ST CHINESE CONTROL CONFERENCE (CCC), 2022, : 7285 - 7291
  • [29] Spatio-temporal causal graph attention network for traffic flow prediction in intelligent transportation systems
    Zhao, Wei
    Zhang, Shiqi
    Wang, Bei
    Zhou, Bing
    PeerJ Computer Science, 2023, 9
  • [30] Spatio-Temporal Residual Graph Convolutional Network for Short-Term Traffic Flow Prediction
    Zhang, Qingyong
    Tan, Meifang
    Li, Changwu
    Xia, Huiwen
    Chang, Wanfeng
    Li, Minglong
    IEEE ACCESS, 2023, 11 : 84187 - 84199