Spatial-Temporal Graph ODE Networks for Traffic Flow Forecasting

Abstract

Spatial-temporal forecasting has attracted tremendous attention in a wide range of applications, and traffic flow prediction is a canonical and typical example. The complex and long-range spatial-temporal correlations of traffic flow bring it to a most intractable challenge. Existing works typically utilize shallow graph convolution networks (GNNs) and temporal extracting modules to model spatial and temporal dependencies respectively. However, the representation ability of such models is limited due to: (1) shallow GNNs are incapable to capture long-range spatial correlations, (2) only spatial connections are considered and a mass of semantic connections are ignored, which are of great importance for a comprehensive understanding of traffic networks. To this end, we propose Spatial-Temporal Graph Ordinary Differential Equation Networks (STGODE). Specifically, we capture spatial-temporal dynamics through a tensor-based ordinary differential equation (ODE), as a result, deeper networks can be constructed and spatial-temporal features are utilized synchronously. To understand the network more comprehensively, semantical adjacency matrix is considered in our model, and a well-design temporal dialated convolution structure is used to capture long term temporal dependencies. We evaluate our model on multiple real-world traffic datasets and superior performance is achieved over state-of-the-art baselines.

References in corpus (5)

• Semi-Supervised Classification with Graph Convolutional Networks
• Deeper Insights into Graph Convolutional Networks for Semi-Supervised Learning
• Deep Spatio-Temporal Residual Networks for Citywide Crowd Flows Prediction
• Graph WaveNet for Deep Spatial-Temporal Graph Modeling
• Theoretically Improving Graph Neural Networks via Anonymous Walk Graph Kernels

Cited by in corpus (5)

• Spatio-Temporal Graph Neural Point Process for Traffic Congestion Event Prediction
• Modeling Network-level Traffic Flow Transitions on Sparse Data
• Inductive Graph Alignment Prompt: Bridging the Gap between Graph Pre-training and Inductive Fine-tuning From Spectral Perspective
• A Data-driven Region Generation Framework for Spatiotemporal Transportation Service Management
• FRIGATE: Frugal Spatio-temporal Forecasting on Road Networks