Spatial-Temporal Dynamic Graph Diffusion Convolutional Network for Traffic Flow Forecasting

Yang Li, Yaying Zhang

Abstract: Traffic forecasting plays a crucial role in intelligent transportation systems and finds application in various domains. Accurate traffic forecasting remains challenging due to the time-varying correlations within the data and the heterogeneous correlations between regions. Although various dynamic spatial-temporal graph models have been proposed to address these challenges in recent years, most of them are burdened by high computation costs and not intuitive to understand. In this paper, we propose a spatial-temporal graph model, Spatial-Temporal Dynamic Graph Diffusion Convolutional Network (SDGDN) that provides an effective and efficient approach to traffic forecasting. From the perspective of traffic flow transition probabilities, SDGDN learns dynamic graph structures to capture the time-varying traffic transition relationships. Besides dynamic graph structures, static node features are employed in diffusion convolution to better capture heterogeneous regional features. Furthermore, we utilize temporal encoding and also generate varying graphs in each stacked layer to enhance the forecasting performance. Experiments results on five real-world datasets demonstrate that SDGDN outperforms most baseline models in terms of both performance and computation efficiency.