1. Multi-level spatial-temporal fusion neural network for traffic flow prediction.
- Author
-
Peng, Zhiying, Yang, Yixue, and Zhao, Hao
- Subjects
- *
CONVOLUTIONAL neural networks , *ARTIFICIAL neural networks , *TRAFFIC flow , *TRAFFIC estimation , *TRAFFIC engineering , *INTELLIGENT transportation systems - 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%. [ABSTRACT FROM AUTHOR]
- Published
- 2024
- Full Text
- View/download PDF