Traffic Graph Convolutional Recurrent Neural Network: A Deep Learning Framework for Network-Scale Traffic Learning and Forecasting

Abstract

Traffic forecasting is a particularly challenging application of spatiotemporal forecasting, due to the time-varying traffic patterns and the complicated spatial dependencies on road networks. To address this challenge, we learn the traffic network as a graph and propose a novel deep learning framework, Traffic Graph Convolutional Long Short-Term Memory Neural Network (TGC-LSTM), to learn the interactions between roadways in the traffic network and forecast the network-wide traffic state. We define the traffic graph convolution based on the physical network topology. The relationship between the proposed traffic graph convolution and the spectral graph convolution is also discussed. An L1-norm on graph convolution weights and an L2-norm on graph convolution features are added to the model's loss function to enhance the interpretability of the proposed model. Experimental results show that the proposed model outperforms baseline methods on two real-world traffic state datasets. The visualization of the graph convolution weights indicates that the proposed framework can recognize the most influential road segments in real-world traffic networks.

References in corpus (6)

• The Emerging Field of Signal Processing on Graphs: Extending High-Dimensional Data Analysis to Networks and Other Irregular Domains
• Spectral Networks and Locally Connected Networks on Graphs
• Deep Convolutional Networks on Graph-Structured Data
• Deep Learning for Short-Term Traffic Flow Prediction
• Deep Bidirectional and Unidirectional LSTM Recurrent Neural Network for Network-wide Traffic Speed Prediction
• 3D Graph Convolutional Networks with Temporal Graphs: A Spatial Information Free Framework For Traffic Forecasting

Cited by in corpus (8)

• Graph Neural Networks: A Review of Methods and Applications
• Multi-graph convolutional network for short-term passenger flow forecasting in urban rail transit
• Graph Neural Ordinary Differential Equations
• Physical-Virtual Collaboration Modeling for Intra-and Inter-Station Metro Ridership Prediction
• Graph Markov Network for Traffic Forecasting with Missing Data
• Graph Neural Processes: Towards Bayesian Graph Neural Networks
• Mind the Gap when Conditioning Amortised Inference in Sequential Latent-Variable Models
• Transfer Learning from an Auxiliary Discriminative Task for Unsupervised Anomaly Detection