Global spatio‐temporal dynamic capturing network‐based traffic flow prediction

Abstract

AbstractCapturing the complex spatio‐temporal relationships of traffic roads is essential to accurately predict traffic flow data. Traditional models typically collect spatial and temporal relationships and increase the complexity of the model by considering connected and unconnected roads. However, global road networks are dynamic and hidden connectivity relationships generally undergo variations over time. A deterministic single‐connection correlation inevitably limits the learning capability of the model. In this paper, the authors propose a global spatio‐temporal dynamic capturing network (GSTDCN) for traffic flow prediction. First, the global encoding module based on the attention mechanism is set up to describe the dynamic spatio‐temporal relationships. It is shown that GSTDCN can learn the hidden node information by spatial correlation at different times. Meanwhile, an effective temporal prediction module is constructed, which facilitates the data augmentation and improves the prediction results of GSTDCN. The model is experimented on four public transportation datasets, and the results show that the GSTDCN outperforms the state‐of‐the‐art baseline.