Dynamic spatial‐temporal network for traffic forecasting based on joint latent space representation
Qian Yu,Liang Ma,Pei Lai,Jin Guo
DOI: https://doi.org/10.1049/itr2.12517
IF: 2.7
2024-05-16
IET Intelligent Transport Systems
Abstract:In this work, we propose a deep neural network approach for traffic prediction, namely DSTN, which captures the dynamic nature in both temporal and spatial dimensions and periodicity of traffic data. In dynamic spatial modeling, unlike most existing attention‐based approaches, which generally model spatial dynamics by constructing a dynamic Laplace matrix based on an attention mechanism in a GCN or by using only an attention mechanism. We propose a joint latent space representation network for capturing spatial dynamics, which integrates GCN and attention mechanism in a novel way to compensate for the shortcomings of single information modeling and can capture global spatial dynamics adaptively. Multiple experiments on two real traffic datasets show that DSTN is superior to SOTA baseline in both road traffic and urban rail transit with good generalization capability. In the era of data‐driven transportation development, traffic forecasting is crucial. Established studies either ignore the inherent spatial structure of the traffic network or ignore the global spatial correlation and may not capture the spatial relationships adequately. In this work, a Dynamic Spatial‐Temporal Network (DSTN) based on Joint Latent Space Representation (JLSR) is proposed for traffic forecasting. Specifically, in the spatial dimension, a JLSR network is developed by integrating graph convolution and spatial attention operations to model complex spatial dependencies. Since it can adaptively fuse the representation information of local topological space and global dynamic space, a more comprehensive spatial dependency can be captured. In the temporal dimension, a Stacked Bidirectional Unidirectional Gated Recurrent Unit (SBUGRU) network is developed, which captures long‐term temporal dependencies through both forward and backward computations and superimposed recurrent layers. On these bases, DSTN is developed in an encoder‐decoder framework and periodicity is flexibly modeled by embedding branches. The performance of DSTN is validated on two types of real‐world traffic flow datasets, and it improves over baselines.
engineering, electrical & electronic,transportation science & technology