Deep spatial-temporal information fusion dynamic graph convolutional network for traffic flow prediction

Abstract

Abstract Predicting traffic flow is vital for advancing intelligent transportation systems, but achieving precise forecasts remains challenging. To deeply explore the complex and diverse dynamic spatio-temporal correlation of traffic data, this study proposes a multi-information fusion spatio-temporal dynamic graph convolution model for traffic flow prediction. In the model construction, firstly, a dynamic graph generation module is designed to fully capture the complex spatial correlation of the traffic network and model the changing spatio-temporal interaction; secondly, the spatial self-attention mechanism is used to dynamically focus on the spatial relationship between different nodes, and combined with the dynamic graph convolution network to extract the spatial dynamic correlation features of the road network to study deep spatiotemporal information; Finally, existing models rarely consider other traffic information related to traffic flow, and a spatiotemporal information fusion module is designed to enhance the ability of the model to perceive information, thereby improving the performance of traffic flow prediction models. Experiments are conducted on four real traffic datasets using three performance evaluation metrics and 18 baseline models. The results show that the model has higher prediction accuracy.