Abstract:Accurate traffic prediction is critical for enhancing the performance of intelligent transportation systems. The key challenge to this task is how to properly model the complex dynamics of traffic while respecting and exploiting both spatial and temporal heterogeneity in data. This paper proposes a novel framework called Meta Graph Transformer (MGT) to address this problem. The MGT framework is a generalization of the original transformer, which is used to model vector sequences in natural language processing. Specifically, MGT has an encoder-decoder architecture. The encoder is responsible for encoding historical traffic data into intermediate representations, while the decoder predicts future traffic states autoregressively. The main building blocks of MGT are three types of attention layers named Temporal Self-Attention (TSA), Spatial Self-Attention (SSA), and Temporal Encoder-Decoder Attention (TEDA), respectively. They all have a multi-head structure. TSAs and SSAs are employed by both the encoder and decoder to capture temporal and spatial correlations. TEDAs are employed by the decoder, allowing every position in the decoder to attend all positions in the input sequence temporally. By leveraging multiple graphs, SSA can conduct sparse spatial attention with various inductive biases. To facilitate the model’s awareness of temporal and spatial conditions, Spatial–Temporal Embeddings (STEs) are learned from external attributes, which are composed of temporal attributes (e.g. sequential order, time of day) and spatial attributes (e.g. Laplacian eigenmaps). These embeddings are then utilized by all the attention layers via meta-learning, hence endowing these layers with Spatial–Temporal Heterogeneity-Aware (STHA) properties. Experiments on three real-world traffic datasets demonstrate the superiority of our model over several state-of-the-art methods. Our code and data are available at ( http://github.com/lonicera-yx/MGT).

Spatial-Temporal Graph Multi-Gate Mixture-of-Expert Model for Traffic Prediction

A Spatial-Temporal Gated Hypergraph Convolution Network for Traffic Prediction

M STL: Multi-Range Multi-Level Spatial-Temporal Learning Model for Network Traffic Prediction

Multi-stage attention spatial-temporal graph networks for traffic prediction

M $^{2}$ STL: Multi-Range Multi-Level Spatial-Temporal Learning Model for Network Traffic Prediction

STG-Meta: Spatial-Temporal Graph Meta-Learning for Traffic Forecasting.

Spatial–temporal combination and multi-head flow-attention network for traffic flow prediction

MFSTGN: a Multi-Scale Spatial-Temporal Fusion Graph Network for Traffic Prediction

Multi-time Scale Spatial-Temporal Graph Neural Network for Traffic Flow Prediction

A Multitask Learning Model for Traffic Flow and Speed Forecasting

Spatial-temporal graph convolutional networks for traffic flow prediction considering multiple traffic parameters

STTG-TTE: Spatial–temporal Gated Multi-Modality Approach for Travel Time Estimation Based on Temporal Convolutional Networks

Spatial-temporal multi-factor fusion graph neural network for traffic prediction

Multistep traffic speed prediction from multiple time-scale spatiotemporal features using graph attention network

Trajectory-Based Spatiotemporal Multi-Task Multi-Graph Network for Traffic State Prediction

Temporal Multi-Graph Convolutional Network for Traffic Flow Prediction

ST-VGBiGRU: A Hybrid Model for Traffic Flow Prediction with Spatio-Temporal Multimodality.

Meta Graph Transformer: A Novel Framework for Spatial–Temporal Traffic Prediction

Multi-Task Synchronous Graph Neural Networks for Traffic Spatial-Temporal Prediction.

Robust Prediction of Traffic Flow Based on Multi-Task Graph Convolution Network

ST-Mamba: Spatial-Temporal Selective State Space Model for Traffic Flow Prediction