Dynamic Graph Convolution and Spatio-Temporal Self-Attention Network for Traffic Flow Prediction

As a typical spatio-temporal series prediction task, traffic flow prediction has found wide application in Intelligent Transportation Systems (ITS). Despite some progress, several unresolved issues persist. Many existing works calculate the dependencies between nodes based on stable long-term traffic data. However, the short-term dependencies are dynamically changing over time, and neglecting them would cause a decrease in predictive performance. In this paper, we propose a novel Dynamic Graph Convolution and Spatio-Temporal Self-Attention (DGSTA) network for traffic flow prediction. Specifically, considering the large amount of short-term and the dynamic dependencies between nodes, we design a new dynamic graph convolution module, which generates adjacency matrices for each time step in a day to dynamically capture the changing short-term dependencies. Additionally, we utilize a multi-head spatio-temporal self-attention module to respectively extract static spatial and temporal correlations between nodes. Furthermore, we design a sequential embedding to explicitly model the long-term correlation between nodes. Extensive experiments conducted on three real-world datasets demonstrate that DGSTA exhibits high competitiveness.