TF-MVGNN: an accurate traffic forecasting framework based on spatial–temporal graph neural network through exploiting multiple-view graph construction and learning
Haoyuan Cheng,Yufeng Wang,Jianhua Ma,Qun Jin
DOI: https://doi.org/10.1007/s00521-024-10508-4
2024-12-05
Neural Computing and Applications
Abstract:In complex and volatile transportation traffic situations, accurately forecasting traffic flow can greatly facilitate to manage and plan real-time traffic. However, due to complex road map and traffic conditions, forecasting temporal traffic is a highly challenging task. In literature, most existing works rely on various spatial–temporal models solely built on the topological structure of road networks, to capture both spatial and temporal dependencies of traffic flow at multiple sensing sites. However, besides the road topological relationship, the dynamics of real-world traffic is indispensably affected by the complex but unknown inter-dependency of multiple sites. Based on the above consideration, this paper proposed a novel traffic forecast framework based on multiple-view graph construction and learning, and spatiotemporal graph neural network (STGNN). Specifically, our work's contributions are following. First, considering transfer entropy (TE)-based relationship graph can characterize the causal influence from one sensing sites to another, the traditional undirected topological graph is transformed into a directed graph through multiplying with the constructed TE relationship graph, which can capture both the topological relationship, and latent casual influence from one site to another, and greatly facilitate to accurately forecast traffic. Meanwhile, a learnable graph matrix is trained to complement the TE-enhanced causality graph. Second, the proposed STGNN-based forecasting framework smoothly integrates recurrent neural network and GNN that works on directional and learnable graphs, so-called DL-GNN. In detail, the aggregation process in DL-GNN is formulated as three parts: aggregation on indegree and outdegree directions of the causality graph, and on learnable graph to fully extract the sites' representations. The comprehensive experiments on real traffic data demonstrate that our proposal outperforms other state-of-the-art STGNN-based traffic forecasting schemes.
computer science, artificial intelligence