Unified Spatio-Temporal Modeling for Traffic Forecasting using Graph Neural Network

by   Amit Roy, et al.

Research in deep learning models to forecast traffic intensities has gained great attention in recent years due to their capability to capture the complex spatio-temporal relationships within the traffic data. However, most state-of-the-art approaches have designed spatial-only (e.g. Graph Neural Networks) and temporal-only (e.g. Recurrent Neural Networks) modules to separately extract spatial and temporal features. However, we argue that it is less effective to extract the complex spatio-temporal relationship with such factorized modules. Besides, most existing works predict the traffic intensity of a particular time interval only based on the traffic data of the previous one hour of that day. And thereby ignores the repetitive daily/weekly pattern that may exist in the last hour of data. Therefore, we propose a Unified Spatio-Temporal Graph Convolution Network (USTGCN) for traffic forecasting that performs both spatial and temporal aggregation through direct information propagation across different timestamp nodes with the help of spectral graph convolution on a spatio-temporal graph. Furthermore, it captures historical daily patterns in previous days and current-day patterns in current-day traffic data. Finally, we validate our work's effectiveness through experimental analysis, which shows that our model USTGCN can outperform state-of-the-art performances in three popular benchmark datasets from the Performance Measurement System (PeMS). Moreover, the training time is reduced significantly with our proposed USTGCN model.


page 1

page 4


SST-GNN: Simplified Spatio-temporal Traffic forecasting model using Graph Neural Network

To capture spatial relationships and temporal dynamics in traffic data, ...

Space Meets Time: Local Spacetime Neural Network For Traffic Flow Forecasting

Traffic flow forecasting is a crucial task in urban computing. The chall...

Deep Learning Methods for Daily Wildfire Danger Forecasting

Wildfire forecasting is of paramount importance for disaster risk reduct...

Uncertainty-based Traffic Accident Anticipation with Spatio-Temporal Relational Learning

Traffic accident anticipation aims to predict accidents from dashcam vid...

RadNet: Incident Prediction in Spatio-Temporal Road Graph Networks Using Traffic Forecasting

Efficient and accurate incident prediction in spatio-temporal systems is...

DGC: Training Dynamic Graphs with Spatio-Temporal Non-Uniformity using Graph Partitioning by Chunks

Dynamic Graph Neural Network (DGNN) has shown a strong capability of lea...

Tuna-AI: tuna biomass estimation with Machine Learning models trained on oceanography and echosounder FAD data

Echo-sounder data registered by buoys attached to drifting FADs provide ...

Please sign up or login with your details

Forgot password? Click here to reset