Model-enhanced spatial-temporal attention networks for traffic density prediction

Abstract

Traffic density is a crucial indicator for evaluating the level of service, as it directly reflects the degree of road congestion and driving comfort. However, accurately predicting real-time traffic density has been a significant challenge in Intelligent Transportation Systems (ITS) due to the nonlinear and spatial-temporal dynamic complexity of traffic density. In this paper, we propose a novel Model-enhanced Spatial-Temporal Attention Network (MSTAN), which constructs a spatial-temporal traffic kernel density model using the Kernel Density Estimation (KDE) method to process the spatiotemporal data and calculate the probabilities of various spatiotemporal events. These probabilities are input into the attention mechanism, enabling the model to recognize the inherent connection between dynamic and distant events. Through this fusion, the network can deeply learn and analyze the spatial-temporal properties of traffic features. Furthermore, this paper utilizes the attention mechanism to dynamically model spatial-temporal dependencies, capturing real-time traffic conditions and density, and constructs a spatial-temporal attention module for learning. To validate the performance of the proposed MSTAN model, experiments are conducted on two public datasets of California highways (PeMS04 and PeMS08). The experimental results demonstrate that the MSTAN model outperforms existing state-of-the-art baseline models in terms of prediction accuracy, thus proving the effectiveness of the model both theoretically and practically.