Dynamic multi-scale spatial-temporal graph convolutional network for traffic flow prediction

被引:7
作者
Gao, Ming [1 ,2 ]
Du, Zhuoran [1 ,2 ]
Qin, Hongmao [1 ,2 ]
Wang, Wei [3 ]
Jin, Guangyin [4 ]
Xie, Guotao [1 ,2 ]
机构
[1] Hunan Univ, Coll Mech & Vehicle Engn, Changsha 410082, Peoples R China
[2] Hunan Univ, Wuxi Intelligent Control Res Inst, Wuxi 214115, Peoples R China
[3] China Power Informat Technol Co Ltd, Beijing 100052, Peoples R China
[4] Natl Innovat Institude Def Technol, Beijing 100052, Peoples R China
关键词
Traffic flow prediction; Graph convolutional network; Multiscale feature extraction; Intelligent transportation systems; Spatial-temporal graph neural network;
D O I
10.1016/j.knosys.2024.112586
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper proposes a dynamic multi-scale spatial-temporal graph convolutional network (DS-STGCN) for traffic flow prediction. The network aims to comprehensively extract global and local dependencies in dynamic spatialtemporal data by inputting traffic network flow data to construct node feature graphs, topology graphs, and time slot feature graphs, capturing the complexity and dynamics of traffic flow. DS-STGCN interprets feature information of the traffic network from both spatial and temporal dimensions through dynamic multi-scale graph convolutional blocks. In the spatial dimension, these blocks use constraints at different levels to balance finegrained local features and extensive global features, revealing the intrinsic structure of traffic flow data. In the temporal dimension, these blocks jointly learn with temporal convolutional blocks to capture multifrequency time patterns and handle long sequence data, effectively extracting potential dependencies of time series. Furthermore, DS-STGCN effectively models the changing spatial-temporal relationships in road network flow by constructing dynamically adaptive updated adjacency tensors, generating dynamic graph structures to address the challenge of changing spatial-temporal relationships in the transportation system. Experimental results show that our method significantly outperforms other competing methods on five real traffic datasets (PEMS03, PEMS04, PEMS07, PEMS08 and METR-LA).
引用
收藏
页数:14
相关论文
共 48 条
[1]   A Long Short-Term Memory-based correlated traffic data prediction framework [J].
Afrin, Tanzina ;
Yodo, Nita .
KNOWLEDGE-BASED SYSTEMS, 2022, 237
[2]  
Bai L, 2020, ADV NEUR IN, V33
[3]   PLU-MCN: Perturbation learning enhanced U-shaped multi-graph convolutional network for traffic flow prediction [J].
Bao, Yinxin ;
Shen, Qinqin ;
Cao, Yang ;
Shi, Quan .
INFORMATION FUSION, 2024, 104
[4]   Spatial-Temporal Complex Graph Convolution Network for Traffic Flow Prediction [J].
Bao, Yinxin ;
Huang, Jiashuang ;
Shen, Qinqin ;
Cao, Yang ;
Ding, Weiping ;
Shi, Zhenquan ;
Shi, Quan .
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 121
[5]  
Bjorck J, 2018, ADV NEUR IN, V31
[6]  
Cao S., 2024, IEEE Trans. Veh. Technol., P1
[7]  
Chen C, 2001, TRANSPORT RES REC, P96
[8]   Short-Term Traffic Flow Prediction Method for Urban Road Sections Based on SpaceTime Analysis and GRU [J].
Dai, Guowen ;
Ma, Changxi ;
Xu, Xuecai .
IEEE ACCESS, 2019, 7 :143025-143035
[9]  
Defferrard M, 2016, ADV NEUR IN, V29
[10]   Using Kalman filter algorithm for short-term traffic flow prediction in a connected vehicle environment [J].
Emami, Azadeh ;
Sarvi, Majid ;
Bagloee, Saeed Asadi .
JOURNAL OF MODERN TRANSPORTATION, 2019, 27 (03) :222-232