ML Research Wiki / Benchmarks / Traffic Prediction / PEMS-BAY

PEMS-BAY

Traffic Prediction Benchmark

Performance Over Time

📊 Showing 11 results | 📏 Metric: MAE @ 12 step

Top Performing Models

Rank	Model	Paper	MAE @ 12 step	Date	Code
1	STBayesian	Traffic signal prediction on transportation networks using spatio-temporal correlations on graphs	4.44	2021-04-27	📦 semink/LargeScale-DLM 📦 semink/lsdlm
2	DCRNN	Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting	2.07	2017-07-06	📦 dmlc/dgl 📦 benedekrozemberczki/pytorch_geometric_temporal 📦 liyaguang/DCRNN
3	Graph Wave-Net	Graph WaveNet for Deep Spatial-Temporal Graph Modeling	1.95	2019-05-31	📦 nnzhan/Graph-WaveNet 📦 sshleifer/Graph-WaveNet 📦 zachysun/taxi_traffic_benchmark
4	GMAN	GMAN: A Graph Multi-Attention Network for Traffic Prediction	1.92	2019-11-11	📦 benedekrozemberczki/pytorch_geometric_temporal 📦 zhengchuanpan/GMAN 📦 VincLee8188/Gman_Ptorch_version
5	STAEformer	STAEformer: Spatio-Temporal Adaptive Embedding Makes Vanilla Transformer SOTA for Traffic Forecasting	1.91	2023-08-21	📦 xdzhelheim/staeformer
6	MegaCRN	Spatio-Temporal Meta-Graph Learning for Traffic Forecasting	1.88	2022-11-27	📦 deepkashiwa20/megacrn
7	D2STGNN	Decoupled Dynamic Spatial-Temporal Graph Neural Network for Traffic Forecasting	1.85	2022-06-18	📦 zezhishao/d2stgnn
8	STEP 📚	Pre-training Enhanced Spatial-temporal Graph Neural Network for Multivariate Time Series Forecasting	1.79	2022-06-18	📦 zezhishao/step 📦 2023-MindSpore-4/Code16
9	STD-MAE 📚	Spatial-Temporal-Decoupled Masked Pre-training for Spatiotemporal Forecasting	1.77	2023-12-01	📦 jimmy-7664/std-mae
10	TITAN	A Time Series is Worth Five Experts: Heterogeneous Mixture of Experts for Traffic Flow Prediction	1.69	2024-09-26	📦 sqlcow/TITAN

All Papers (11)

Traffic signal prediction on transportation networks using spatio-temporal correlations on graphs

2021

STBayesian

semink/LargeScale-DLM semink/lsdlm

Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting

2017

DCRNN

dmlc/dgl benedekrozemberczki/pytorch_geometric_temporal

Graph WaveNet for Deep Spatial-Temporal Graph Modeling

2019

Graph Wave-Net

nnzhan/Graph-WaveNet sshleifer/Graph-WaveNet

GMAN: A Graph Multi-Attention Network for Traffic Prediction

2019

GMAN

benedekrozemberczki/pytorch_geometric_temporal zhengchuanpan/GMAN

STAEformer: Spatio-Temporal Adaptive Embedding Makes Vanilla Transformer SOTA for Traffic Forecasting

2023

STAEformer

xdzhelheim/staeformer

Spatio-Temporal Meta-Graph Learning for Traffic Forecasting

2022

MegaCRN

deepkashiwa20/megacrn

Decoupled Dynamic Spatial-Temporal Graph Neural Network for Traffic Forecasting

2022

D2STGNN

zezhishao/d2stgnn

Pre-training Enhanced Spatial-temporal Graph Neural Network for Multivariate Time Series Forecasting

2022

STEP

zezhishao/step 2023-MindSpore-4/Code16

Spatial-Temporal-Decoupled Masked Pre-training for Spatiotemporal Forecasting

2023

STD-MAE

jimmy-7664/std-mae

A Time Series is Worth Five Experts: Heterogeneous Mixture of Experts for Traffic Flow Prediction

2024

TITAN

sqlcow/TITAN

T-Graphormer: Using Transformers for Spatiotemporal Forecasting

2025

T-Graphormer

rdh1115/T-Graphormer

PEMS-BAY

Performance Over Time

Edit Benchmark Results

Edit Result

Top Performing Models

All Papers (11)

Traffic signal prediction on transportation networks using spatio-temporal correlations on graphs

Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting

Graph WaveNet for Deep Spatial-Temporal Graph Modeling

GMAN: A Graph Multi-Attention Network for Traffic Prediction

STAEformer: Spatio-Temporal Adaptive Embedding Makes Vanilla Transformer SOTA for Traffic Forecasting

Spatio-Temporal Meta-Graph Learning for Traffic Forecasting

Decoupled Dynamic Spatial-Temporal Graph Neural Network for Traffic Forecasting

Pre-training Enhanced Spatial-temporal Graph Neural Network for Multivariate Time Series Forecasting

Spatial-Temporal-Decoupled Masked Pre-training for Spatiotemporal Forecasting

A Time Series is Worth Five Experts: Heterogeneous Mixture of Experts for Traffic Flow Prediction

T-Graphormer: Using Transformers for Spatiotemporal Forecasting

Model	Paper	MAE @ 12 step	Date
STBayesian	Traffic signal prediction on transportation netwo…	4.44	2021-04-27
DCRNN	Diffusion Convolutional Recurrent Neural Network:…	2.07	2017-07-06
Graph Wave-Net	Graph WaveNet for Deep Spatial-Temporal Graph Mod…	1.95	2019-05-31
GMAN	GMAN: A Graph Multi-Attention Network for Traffic…	1.92	2019-11-11
STAEformer	STAEformer: Spatio-Temporal Adaptive Embedding Ma…	1.91	2023-08-21
MegaCRN	Spatio-Temporal Meta-Graph Learning for Traffic F…	1.88	2022-11-27
D2STGNN	Decoupled Dynamic Spatial-Temporal Graph Neural N…	1.85	2022-06-18
STEP	Pre-training Enhanced Spatial-temporal Graph Neur…	1.79	2022-06-18
STD-MAE	Spatial-Temporal-Decoupled Masked Pre-training fo…	1.77	2023-12-01
TITAN	A Time Series is Worth Five Experts: Heterogeneou…	1.69	2024-09-26
T-Graphormer	T-Graphormer: Using Transformers for Spatiotempor…	1.63	2025-01-22