Multi-spatio-temporal Fusion Graph Recurrent Network for Traffic Forecasting

卷:124

Article

关键词: NEURAL-NETWORKS;    PREDICTION;   

DOI  :  10.1016/j.engappai.2023.106615

来源: SCIE

【 摘 要 】

Accurate traffic forecasting is crucial for smart city development in the new era. However, the intricate spatial and temporal dependencies in traffic data present significant challenges for prediction accuracy. Existing methods often rely on predefined adjacency matrices to capture Spatio-temporal dependencies, which may not adapt well to the dynamic nature of road traffic. To address these challenges, we propose the Multi-Spatiotemporal Fusion Graph Recurrent Network (MSTFGRN). This innovative approach introduces a data-driven method for generating a weighted adjacency matrix, effectively capturing real-time spatial dependencies that are not adequately captured by predefined matrices. The MSTFGRN also incorporates a novel two-way Spatio-temporal fusion operation to learn hidden dependencies between parallel Spatio-temporal relations at different time points. Additionally, a global attention mechanism is integrated into the Spatio-temporal fusion module, enabling the simultaneous capture of global Spatio-temporal dependencies. Through extensive trials on publicly available highway traffic datasets, our method demonstrates state-of-the-art performance compared to alternative baselines.

【 授权许可】

Free