Traffic Flow Prediction Based on Dynamic Spatial Enhancement

Shumei Bao,Cheng Chen,Boyan Huo,Xiaoyi Lv,Chen
DOI: https://doi.org/10.1117/12.3039888
2024-01-01
Abstract:Traffic flow prediction is a spatiotemporal prediction task. The biggest difference between traffic flow and ordinary time series prediction problems is that it is restricted by the topology of the road network. Therefore, it is necessary to accurately mine the spatial information in the road network. This paper introduces a spatial block designed to obtain spatial correlation, incorporating both Graph Convolutional Networks (GCN) and a novel spatial transformer module for handling spatial properties. By leveraging spatial transformers, the model can extract dynamic spatial correlations among individual nodes. The dynamic spatial information and static spatial information are integrated through the gating mechanism. In addition, this chapter also adds the Informer module based on GRU to mine long-term sequence information to improve the forgetting problem of using the GRU module. The final experimental results were verified on the London M25 highway data set and compared with other benchmark methods and found to achieve good results.
What problem does this paper attempt to address?