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).

Sparse Mobile Crowdsensing for Cost-Effective Traffic State Estimation with Spatio-Temporal Transformer Graph Neural Network

A Spatial-Temporal Gated Hypergraph Convolution Network for Traffic Prediction

Short-Term Speed Forecasting of Large-Scale Urban Road Network Based on Transformer

Spatial-Temporal Attention Model for Traffic State Estimation with Sparse Internet of Vehicles

Learning Multiaspect Traffic Couplings by Multirelational Graph Attention Networks for Traffic Prediction

Spatial-Temporal Generative AI for Traffic Flow Estimation with Sparse Data of Connected Vehicles

Spatial-Temporal Data Inference With Graph Attention Neural Networks in Sparse Mobile Crowdsensing

Multicomponent Spatial-Temporal Graph Attention Convolution Networks for Traffic Prediction with Spatially Sparse Data

Learning dynamic and hierarchical traffic spatiotemporal features with Transformer

Real-Time Traffic Speed Estimation for Smart Cities with Spatial Temporal Data: A Gated Graph Attention Network Approach

Graph Spatial-Temporal Transformer Network for Traffic Prediction

A Spatiotemporal Graph Neural Network with Graph Adaptive and Attention Mechanisms for Traffic Flow Prediction

Adaptive spatial-temporal graph attention networks for traffic flow forecasting

A Graph-Based Temporal Attention Framework for Multi-Sensor Traffic Flow Forecasting

Adaptive Graph Spatial-Temporal Transformer Network for Traffic Flow Forecasting

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

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

STGformer: Efficient Spatiotemporal Graph Transformer for Traffic Forecasting

Empowering Spatial Knowledge Graph for Mobile Traffic Prediction.

Temporal Metrics Based Aggregated Graph Convolution Network for traffic forecasting

Spatio-Temporal Parallel Transformer based model for Traffic Prediction