Multi-level spatial-temporal fusion neural network for traffic flow prediction

Zhiying Peng,Yixue Yang,Hao Zhao
DOI: https://doi.org/10.1007/s10586-024-04296-8
2024-03-07
Cluster Computing
Abstract:In urban system management and public safety, accurate traffic flow forecasting is pivotal for real-world applications such as traffic control, resource-sharing scheduling platforms, and intelligent transportation systems. The challenge lies in effectively capturing complex spatio-temporal correlations. To address this, we introduce ST-DSTN, a novel spatial-temporal fusion attention-based deep neural network that significantly advances the current state-of-the-art. ST-DSTN utilizes three distinct temporal branches to capture temporal dependencies and a convolutional neural network to encapsulate spatial dependencies. Our innovative contribution includes a spatio-temporal fusion attention module, effectively modeling the dynamic spatio-temporal dependencies of traffic data, thus improving upon existing methods. Additionally, we propose a hierarchical fusion method that permeates the entire model, capturing complex correlations at various levels through multi-level fusion. Our comprehensive experiments on the BikeNYC dataset demonstrate ST-DSTN's superior performance, supported by numerical results that substantiate its improvement over current methods, surpassing the DeepSTN+ model by 4.31%.
computer science, information systems, theory & methods
What problem does this paper attempt to address?