Traffic Speed Prediction Based on Dynamic Spatial-Temporal Attention Graph Convolutional Network

Abstract The prediction of traffic speed represents a crucial branch within the broader field of traffic forecasting. Its objective is to forecast the average speed or speed distribution on road segments or within a network for a designated future time period . The prediction results can provide scientific decision support for traffic management departments to effectively control and guide traffic flow. However, given the limitations of urban road network topology and its dynamic temporal shifts, traffic prediction has consistently been deemed a perpetual scientific challenge. To simultaneously encapsulate both spatial and temporal dependencies, we introduce an innovative Dynamic Spatio-Temporal Attention Network (DSTANet). This model integrates spatio-temporal attention with graph convolutional networks and temporal convolution to enhance its analytical capabilities. Specifically, the graph convolutional network is leveraged to learn the intricate topology for capturing spatial relations, while temporal convolution is employed to ascertain the dynamic fluctuations of traffic data, thereby acquiring temporal dependencies. Subsequently, we implement our model to forecast traffic patterns within the framework of road networks. Experimental results demonstrate that our model effectively captures the spatio-temporal dependencies among roadway junctions derived from traffic data, outperforming state-of-the-art benchmarks on real-world traffic datasets in terms of predictive accuracy.