首页|Local-global dynamic correlations based spatial-temporal convolutional network for traffic flow forecasting

Local-global dynamic correlations based spatial-temporal convolutional network for traffic flow forecasting

扫码查看
Local-global dynamic correlations based spatial-temporal convolutional network for traffic flow forecasting
Traffic flow forecasting plays a crucial role and is the key technology to realize dynamic traffic guidance and active traffic control in intelligent traffic systems(ITS).Aiming at the complex local and global spatial-temporal dynamic characteristics of traffic flow,this paper proposes a new traffic flow forecasting model spatial-temporal attention graph neural network(STA-GNN)by combining at-tention mechanism(AM)and spatial-temporal convolutional network.The model learns the hidden dynamic local spatial correlations of the traffic network by combining the dynamic adjacency matrix constructed by the graph learning layer with the graph convolutional network(GCN).The local tem-poral correlations of traffic flow at different scales are extracted by stacking multiple convolutional kernels in temporal convolutional network(TCN).And the global spatial-temporal dependencies of long-time sequences of traffic flow are captured by the spatial-temporal attention mechanism(STAtt),which enhances the global spatial-temporal modeling and the representational ability of model.The experimental results on two datasets,METR-LA and PEMS-BAY,show the proposed STA-GNN model outperforms the common baseline models in forecasting accuracy.

traffic flow forecastinggraph convolutional network(GCN)temporal convolu-tional network(TCN)attention mechanism(AM)

张红、GONG Lei、ZHAO Tianxin、ZHANG Xijun、WANG Hongyan

展开 >

College of Computer and Communication,Lanzhou University of Technology,Lanzhou 730050,P.R.China

traffic flow forecasting graph convolutional network(GCN) temporal convolu-tional network(TCN) attention mechanism(AM)

2024

高技术通讯(英文版)
中国科学技术信息研究所(ISTIC)

高技术通讯(英文版)

影响因子:0.058
ISSN:1006-6748
年,卷(期):2024.30(4)