References & Citations
Computer Science > Machine Learning
Title: MAF-GNN: Multi-adaptive Spatiotemporal-flow Graph Neural Network for Traffic Speed Forecasting
(Submitted on 8 Aug 2021 (v1), last revised 9 Mar 2022 (this version, v2))
Abstract: Traffic forecasting is a core element of intelligent traffic monitoring system. Approaches based on graph neural networks have been widely used in this task to effectively capture spatial and temporal dependencies of road networks. However, these approaches can not effectively define the complicated network topology. Besides, their cascade network structures have limitations in transmitting distinct features in the time and space dimensions. In this paper, we propose a Multi-adaptive Spatiotemporal-flow Graph Neural Network (MAF-GNN) for traffic speed forecasting. MAF-GNN introduces an effective Multi-adaptive Adjacency Matrices Mechanism to capture multiple latent spatial dependencies between traffic nodes. Additionally, we propose Spatiotemporal-flow Modules aiming to further enhance feature propagation in both time and space dimensions. MAF-GNN achieves better performance than other models on two real-world datasets of public traffic network, METR-LA and PeMS-Bay, demonstrating the effectiveness of the proposed approach.
Submission history
From: Yaobin Xu [view email][v1] Sun, 8 Aug 2021 09:06:43 GMT (1175kb,D)
[v2] Wed, 9 Mar 2022 13:04:28 GMT (0kb,I)
Link back to: arXiv, form interface, contact.