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

GeoTransformer: Enhancing Urban Forecasting with Dependency Retrieval and Geospatial Attention

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

Forecaster: A Graph Transformer for Forecasting Spatial and Time-Dependent Data

uTransformer: unified spatial-temporal transformer with external factors for traffic flow forecasting

A Novel Multiscale Transformer Network Framework for Natural Gas Consumption Forecasting

Spatial-Temporal Graph Sandwich Transformer for Traffic Flow Forecasting

Earthformer: Exploring Space-Time Transformers for Earth System Forecasting

Efficient Large-Scale Traffic Forecasting with Transformers: A Spatial Data Management Perspective

Spatiotemporal Fusion Transformer for large-scale traffic forecasting

A Lightweight and Accurate Spatial-Temporal Transformer for Traffic Forecasting

Deep transformer-based heterogeneous spatiotemporal graph learning for geographical traffic forecasting

Boosting Urban Traffic Speed Prediction via Integrating Implicit Spatial Correlations

FGITrans: Cross-City Transformer for Fine-grained Urban Flow Inference

High-Resolution Urban Flows Forecasting with Coarse-Grained Spatiotemporal Data

Towards Integrated and Fine-Grained Traffic Forecasting: A Spatio-Temporal Heterogeneous Graph Transformer Approach

Spatio-Temporal Transformer Network for Weather Forecasting

STGformer: Efficient Spatiotemporal Graph Transformer for Traffic Forecasting

UrbanDiT: A Foundation Model for Open-World Urban Spatio-Temporal Learning

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

MultiSPANS: A Multi-range Spatial-Temporal Transformer Network for Traffic Forecast via Structural Entropy Optimization

Transformer network with decoupled spatial–temporal embedding for traffic flow forecasting