Home

Awesome

This list can be considered outdated. For a more up-to-date list, check: https://github.com/lixus7/Time-Series-Works-Conferences

Traffic Prediction

Traffic prediction is the task of predicting future traffic measurements (e.g. volume, speed, etc.) in a road network (graph), using historical data (timeseries).

Things are usually better defined through exclusions, so here are similar things that I do not include:

The papers are haphazardly selected.

Summary

A tabular summary of paper and publically available datasets. The paper is reverse chronologically sorted. NO GUARANTEE is made that this table is complete or accurate (please raise an issue if you spot any error).

papervenuepublished date# other datsetsMETR-LAPeMS-BAYPeMS-D7(M)PeMS-D7(L)PeMS-04PeMS-08LOOPSZ-taxiLos-loopPeMS-03PeMS-07PeMS-I-405PeMS-04(S)TOTAL open
TOTAL38286333322111195
G-SWaNIoTDI9 May 2311114
SCPTArXiv9 May 2311114
MP-WaveNetArXiv9 May 23112
GTSICLR4 May 211112
FASTGNNTII29 Jan 2111
HetGATJAIHC23 Jan 21112
GST-GATIEEE Access6 Jan 21112
CLGRNarXiv4 Jan 21311
DKFNSIGSPATIAL3 Nov 20112
STGAMCISP-BMEI17 Oct 20112
ARNNNat. Commun11 Sept 2011
ST-TrafficNetELECGJ9 Sept 20112
M2J. AdHoc1 Sept 20112
H-STGCNKDD23 Aug 200
SGMNJ. TRC20 Aug 20112
GDRNNNTU16 Aug 20112
ISTD-GCNarXiv10 Aug 20112
GTSUCONN3 Aug 20112
FC-GAGAarXiv30 Jul 20112
STGATIEEE Access22 Jul 20112
STNNT-ITS16 Jul 200
AGCRNarXiv6 Jul 20112
GWNN-LSTMJ. Phys. Conf. Ser.20 Jun 2011
A3T-GCNarXiv20 Jun 20112
TSE-SCTrans-GIS1 Jun 20112
MTGNNarXiv24 May 20112
ST-MetaNet+TKDE19 May 20112
STGNNWWW20 Apr 20112
STSeq2SeqarXiv6 Apr 20112
DSTGNNarXiv12 Mar 2011
RSTAGIoT-J19 Feb 20112
GMANAAAI7 Feb 2011
MRA-BGCNAAAI7 Feb 20112
STSGCNAAAI7 Feb 2011114
SLCNNAAAI7 Feb 201113
DDP-GCNarXiv7 Feb 200
R-SSMICLR13 Jan 2011
GWNV2arXiv11 Dec 19112
DeepGLONeurIPS8 Dec 19111
STGRATarXiv29 Nov 19112
TGC-LSTMT-ITS28 Nov 1911
DCRNN-RILTrustCom/BigDataSE31 Oct 19112
L-VGAEarXiv18 Oct 1911
T-GCNT-ITS22 Aug 19112
GWNIJCAI10 Aug 19112
ST-MetaNetKDD25 Jul 1911
MRes-RGNN-GAAAI17 Jul 19112
CDSAarXiv23 May 1911
STDGIICLR12 Apr 1911
ST-UNetarXiv13 Mar 191113
3D-TGCNarXiv3 Mar 191113
ASTGCNAAAI27 Jan 19112
PSNT-ITS17 Aug 1810
GaANUAI6 Aug 18211
Seq2Seq HybridKDD19 Jul 180
STGCNIJCAI13 Jul 18112
DCRNNICLR30 Apr 18112
SBU-LSTMUrbComp14 Aug 1711
GRUYAC5 Jan 1711

Performance

METR-LA MAE@60 mins

PeMS-BAY MAE@60 mins

NOTES: The experimental setttings may vary. But the common setting is:

However, there are many caveats:

Dataset

Publically available datasets and where to find them.

Baidu, code: 'umqd'

The following datasets are not publically available:

Also relevant:

Libraries

Paper

The papers are sorted alphabetically based on model name. The citations are based on Google scholar citation.

You can find the bibtex in traffic_prediction.bib (not complete yet)

modelcitationsvenuepublished datepapercodes
3D-TGCN12arXiv3 Mar 193D Graph Convolutional Networks with Temporal Graphs: A Spatial Information Free Framework For Traffic Forecasting
AGCRN3arXiv6 Jul 20Adaptive Graph Convolutional Recurrent Network for Traffic ForecastingPyTorch
ARNN0Nat. Commun11 Sep 20Autoreservoir computing for multistep ahead prediction based on the spatiotemporal information transformation
ASTGCN63AAAI27 Jan 19Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow ForecastingPytorch MXNet
CDSA2arXiv23 May 19CDSA: Cross-Dimensional Self-Attention for Multivariate, Geo-tagged Time Series Imputation
CLGRN0arXiv4 Jan 21Conditional Local Filters with Explainers for Spatio-Temporal Forecasting
DCRNN427ICLR30 Apr 18DIFFUSION CONVOLUTIONAL RECURRENT NEURAL NETWORK: DATA-DRIVEN TRAFFIC FORECASTINGtf PyTorch
DCRNN-RIL2TrustCom/BigDataSE31 Oct 19Diffusion Convolutional Recurrent Neural Network with Rank Influence Learning for Traffic Forecasting
DDP-GCN1arXiv7 Feb 20DDP-GCN: Multi-Graph Convolutional Network for Spatiotemporal Traffic Forecasting
DeepGLO22NeurIPS8 Dec 19Think Globally, Act Locally: A Deep Neural Network Approach to High-Dimensional Time Series Forecasting
DGCRN0ArXiv30 Apr 21Spatiotemporal Adaptive Gated Graph Convolution Network for Urban Traffic Flow ForecastingPyTorch
DKFN0SIGSPATIAL3 Nov 20Graph Convolutional Networks with Kalman Filtering for Traffic PredictionPyTorch
DSTGNN0arXiv12 Mar 20Dynamic Spatiotemporal Graph Neural Network with Tensor Network
FASTGNN0TII29 Jan 21FASTGNN: A Topological Information Protected Federated Learning Approach For Traffic Speed Forecasting
FC-GAGA0arXiv30 Jul 20FC-GAGA: Fully Connected Gated Graph Architecture for Spatio-Temporal Traffic Forecasting
FreqST2ICDM17 Nov 21FreqST: Exploiting Frequency Information in Spatiotemporal Modeling for Traffic Prediction
GaAN126UAI6 Aug 18GaAN: Gated Attention Networks for Learning on Large and Spatiotemporal GraphsMXNet
GDRNN0NTU16 Aug 20Deep learning approaches for traffic prediction
GMAN20AAAI7 Feb 20GMAN: A Graph Multi-Attention Network for Traffic Predictiontf
GRU308YAC5 Jan 17Using LSTM and GRU neural network methods for traffic flow predictionKeras
GST-GAT0IEEE Access6 Jan 21Modeling Global Spatial–Temporal Graph Attention Network for Traffic Prediction
G-SWaN0IoTDI9 May 23Because Every Sensor Is Unique, so Is Every Pair: Handling Dynamicity in Traffic ForecastingPyTorch
GTS0UCONN3 Aug 20End-to-End Structure-Aware Convolutional Networks on Graphs
GTS0ICLR4 May 21Discrete Graph Structure Learning for Forecasting Multiple Time SeriesPyTorch
GWN46IJCAI10 Aug 19Graph WaveNet for Deep Spatial-Temporal Graph ModelingPyTorch
GWNN-LSTM0J. Phys. Conf. Ser.20 Jun 20Graph Wavelet Long Short-Term Memory Neural Network: A Novel Spatial-Temporal Network for Traffic Prediction.
GWNV20arXiv11 Dec 19Incrementally Improving Graph WaveNet Performance on Traffic PredictionPyTorch
H-STGCN0KDD23 Aug 20Hybrid Spatio-Temporal Graph Convolutional Network: Improving Traffic Prediction with Navigation Data
HetGAT0JAIHC23 Jan 21HetGAT: a heterogeneous graph attention network for freeway traffic speed prediction
ISTD-GCN0arXiv10 Aug 20ISTD-GCN: Iterative Spatial-Temporal Diffusion Graph Convolutional Network for Traffic Speed Forecasting
L-VGAE0arXiv18 Oct 19Decoupling feature propagation from the design of graph auto-encoders
LSTM39TENCON22 Nov 16Traffic flow prediction with Long Short-Term Memory Networks (LSTMs)
M21J. AdHoc1 Sep 20A performance modeling and analysis of a novel vehicular traffic flow prediction system using a hybrid machine learning-based model
MP-WaveNet0ArXiv9 May 23Message Passing Neural Networks for Traffic Forecasting
MRA-BGCN7AAAI7 Feb 20Multi-Range Attentive Bicomponent Graph Convolutional Network for Traffic Forecasting
MRes-RGNN-G20AAAI17 Jul 19Gated Residual Recurrent Graph Neural Networks for Traffic Prediction
MTGNN7arXiv24 May 20Connecting the Dots: Multivariate Time Series Forecasting with Graph Neural Networks
PSN4T-ITS17 Aug 18Pattern Sensitive Prediction of Traffic Flow Based on Generative Adversarial Framework
R-SSM0ICLR13 Jan 20Relational State-Space Model for Stochastic Multi-Object Systems
RSTAG3IoT-J19 Feb 20Reinforced Spatiotemporal Attentive Graph Neural Networks for Traffic Forecasting
SAE1626T-ITS9 Sep 14Traffic flow prediction with big data: a deep learning approachKeras
SBU-LSTM157UrbComp14 Aug 17Deep Bidirectional and Unidirectional LSTM Recurrent Neural Network for Network-wide Traffic Speed Prediction
SCPT0ArXiv9 May 23Traffic Forecasting on New Roads Unseen in the Training Data Using Spatial Contrastive Pre-Training
Seq2Seq Hybrid48KDD19 Jul 18Deep Sequence Learning with Auxiliary Information for Traffic Predictiontf
SGMN1J. TRC20 Aug 20Graph Markov network for traffic forecasting with missing data
SLCNN1AAAI7 Feb 20Spatio-Temporal Graph Structure Learning for Traffic Forecasting
ST-MetaNet39KDD25 Jul 19Urban traffic prediction from spatio-temporal data using deep meta learningMXNet
ST-MetaNet+0TKDE19 May 20Spatio-Temporal Meta Learning for Urban Traffic Prediction
ST-TrafficNet2ELECGJ9 Sep 20ST-TrafficNet: A Spatial-Temporal Deep Learning Network for Traffic Forecasting
ST-UNet11arXiv13 Mar 19ST-UNet: A Spatio-Temporal U-Network for Graph-structured Time Series Modeling
STDGI3ICLR12 Apr 19Spatio-Temporal Deep Graph Infomax
STGAT0IEEE Access22 Jul 20STGAT: Spatial-Temporal Graph Attention Networks for Traffic Flow Forecasting
STGAT0IEEE Access22 Jul 20STGAT: Spatial-Temporal Graph Attention Networks for Traffic Flow Forecasting
STGCN322IJCAI13 Jul 18Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecastingtf PyTorch MXNet
STGNN4WWW20 Apr 20Traffic Flow Prediction via Spatial Temporal Graph Neural Network
STGRAT6arXiv29 Nov 19STGRAT: A Spatio-Temporal Graph Attention Network for Traffic Forecasting
STNN0T-ITS16 Jul 20STNN: A Spatio-Temporal Neural Network for Traffic Predictions
STSeq2Seq0arXiv6 Apr 20Forecast Network-Wide Traffic States for Multiple Steps Ahead: A Deep Learning Approach Considering Dynamic Non-Local Spatial Correlation and Non-Stationary Temporal Dependency
STGAM1CISP-BMEI17 Oct 20Spatial-Temporal Graph Attention Model on Traffic Forecasting
STSGCN5AAAI7 Feb 20Spatial-Temporal Synchronous Graph Convolutional Networks: A New Framework for Spatial-Temporal Network Data ForecastingMXNet
TGC-LSTM95T-ITS28 Nov 19Traffic Graph Convolutional Recurrent Neural Network: A Deep Learning Framework for Network-Scale Traffic Learning and Forecasting
TSE-SC0Trans-GIS1 Jun 20Traffic transformer: Capturing the continuity and periodicity of time series for traffic forecasting
TSSRGCN4ICDM17 Nov 21TSSRGCN: Temporal Spectral Spatial Retrieval Graph Convolutional Network for Traffic Flow Forecasting
0arXiv15 Jul 20On the Inclusion of Spatial Information for Spatio-Temporal Neural NetworksPyTorch
96NeuCom27 Nov 18LSTM-based traffic flow prediction with missing data

Things that would be in the table above if I have more time:

Other works

Other works that is not based on a static-spatial-graph of timeseries:

Other lists:

Acknowledgement