Dynamic Global-Local Spatial-Temporal Network for Traffic Speed Prediction

Abstract

Predicting traffic speed accurately is a very challenging task of the intelligent traffic system (ITS), due to the complex and dynamic spatial-temporal dependencies from both temporal and spatial aspects. There not only exits short-term local neighboring fluctuation and long-term global trend in temporal aspect, but also local and global correlations in spatial aspect. Most existing work focus on the local spatial-temporal dependencies, ignoring the global dynamic spatial-temporal corrections, which is comparably critical for traffic speed prediction. To address this problem, we propose a novel Dynamic Global-Local Spatial-Temporal Network(DGLSTNet) for traffic speed prediction, which consists of multiple spatial-temporal module considering the local and global information simultaneously from both temporal and spatial perspective. Each temporal module applies stacked dilated convolution block to exploit multi-scale local temporal information. Moreover, we empoly a global temporal attention block to capture global dependencies of temporal domain in an attention mechanism. In each spatial module, we not only learn the local but also focus on dynamic global spatial information learned by dymamic graph learning block. Combining the feature results from local and global perspective, the capability and expressiveness of traffic predicting model is improved. Experiment results on two real-world traffic datasets have demonstrated that our proposed model can effectively capture the comprehensive spatial-temporal dependencies and can achieve state-of-the-art prediction performance compared with the existing works.