Spatio-Temporal Graph Neural Point Process for Traffic Congestion Event Prediction

Guangyin Jin,Lingbo Liu,Fuxian Li,Jincai Huang
DOI: https://doi.org/10.1609/aaai.v37i12.26669
2023-11-15
Abstract:Traffic congestion event prediction is an important yet challenging task in intelligent transportation systems. Many existing works about traffic prediction integrate various temporal encoders and graph convolution networks (GCNs), called spatio-temporal graph-based neural networks, which focus on predicting dense variables such as flow, speed and demand in time snapshots, but they can hardly forecast the traffic congestion events that are sparsely distributed on the continuous time axis. In recent years, neural point process (NPP) has emerged as an appropriate framework for event prediction in continuous time scenarios. However, most conventional works about NPP cannot model the complex spatio-temporal dependencies and congestion evolution patterns. To address these limitations, we propose a spatio-temporal graph neural point process framework, named STGNPP for traffic congestion event prediction. Specifically, we first design the spatio-temporal graph learning module to fully capture the long-range spatio-temporal dependencies from the historical traffic state data along with the road network. The extracted spatio-temporal hidden representation and congestion event information are then fed into a continuous gated recurrent unit to model the congestion evolution patterns. In particular, to fully exploit the periodic information, we also improve the intensity function calculation of the point process with a periodic gated mechanism. Finally, our model simultaneously predicts the occurrence time and duration of the next congestion. Extensive experiments on two real-world datasets demonstrate that our method achieves superior performance in comparison to existing state-of-the-art approaches.
Machine Learning,Artificial Intelligence,Social and Information Networks
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the important but challenging task of traffic congestion event prediction in intelligent transportation systems. Most of the existing traffic prediction works use Spatio - Temporal Graph - based Neural Networks. These methods mainly focus on predicting dense variables in time snapshots, such as flow, speed and demand, but perform poorly in predicting traffic congestion events sparsely distributed on the continuous time axis. In addition, although the Neural Point Process (NPP) performs well in event prediction in continuous - time scenarios, most traditional NPP methods have difficulty in modeling complex spatio - temporal dependencies and congestion evolution patterns. To solve these problems, the authors propose a new spatio - temporal graph neural point process framework, called STGNPP (Spatio - Temporal Graph Neural Point Process), which is specifically used for traffic congestion event prediction. Specifically, STGNPP improves the existing methods in the following ways: 1. **Spatio - temporal graph learning module**: A spatio - temporal graph learning module is designed to fully capture spatio - temporal dependencies over a long - time range from historical traffic state data. 2. **Continuous Gated Recurrent Unit**: The extracted spatio - temporal latent representation and congestion event information are input into a Continuous Gated Recurrent Unit (CGRU) to model the congestion evolution pattern. 3. **Periodic gate mechanism**: In order to make full use of periodic information, the calculation of the point process intensity function is improved and a periodic gate mechanism is introduced. 4. **Simultaneous prediction of occurrence time and duration**: Finally, the model can simultaneously predict the occurrence time and duration of the next congestion event. Through extensive experiments on two real - world datasets, the authors prove that the proposed STGNPP model significantly outperforms the existing state - of - the - art methods in prediction performance.