# SST-GNN: Simplified Spatio-temporal Traffic forecasting model using Graph Neural Network

arXiv: https://arxiv.org/abs/2104.00055 · HTML: https://mlsanity.com/paper/2104.00055/

## Key facts

- Cited by 9 papers in the map (data through 14 Oct 2026).
- Builds on 6 of its 14 references that are in the map.
- First posted on arXiv 31 Mar 2021.

## Builds on (most cited first, 6 of 6)

- [Attention Is All You Need](https://mlsanity.com/paper/1706.03762/) (2017)
- [Semi-Supervised Classification with Graph Convolutional Networks](https://mlsanity.com/paper/1609.02907/) (2016)
- [Inductive Representation Learning on Large Graphs](https://mlsanity.com/paper/1706.02216/) (2017)
- [Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering](https://mlsanity.com/paper/1606.09375/) (2016)
- [Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting](https://mlsanity.com/paper/1709.04875/) (2017)
- [Graph WaveNet for Deep Spatial-Temporal Graph Modeling](https://mlsanity.com/paper/1906.00121/) (2019)

## Built on by (most cited first, 9 of 9)

- [FPTN: Fast Pure Transformer Network for Traffic Flow Forecasting](https://mlsanity.com/paper/2303.07685/) (2023)
- [Graph Neural Networks in Intelligent Transportation Systems: Advances, Applications and Trends](https://mlsanity.com/paper/2401.00713/) (2024)
- [Urban Incident Prediction with Graph Neural Networks: Integrating Government Ratings and Crowdsourced Reports](https://mlsanity.com/paper/2506.08740/) (2025)
- [Deep Learning-Derived Optimal Aviation Strategies to Control Pandemics](https://mlsanity.com/paper/2210.10888/) (2022)
- [STLGRU: Spatio-Temporal Lightweight Graph GRU for Traffic Flow Prediction](https://mlsanity.com/paper/2212.04548/) (2022)
- [Interpretable Water Level Forecaster with Spatiotemporal Causal Attention Mechanisms](https://mlsanity.com/paper/2303.00515/) (2023)
- [Futures Quantitative Investment with Heterogeneous Continual Graph Neural Network](https://mlsanity.com/paper/2303.16532/) (2023)
- [Adaptive Graph Pruning with Sudden-Events Evaluation for Traffic Prediction using Online Semi-Decentralized ST-GNNs](https://mlsanity.com/paper/2512.17352/) (2025)
- [Efficient Prompt Learning for Traffic Forecasting](https://mlsanity.com/paper/2605.08273/) (2026)

## Abstract (arXiv)

To capture spatial relationships and temporal dynamics in traffic data, spatio-temporal models for traffic forecasting have drawn significant attention in recent years. Most of the recent works employed graph neural networks(GNN) with multiple layers to capture the spatial dependency. However, road junctions with different hop-distance can carry distinct traffic information which should be exploited separately but existing multi-layer GNNs are incompetent to discriminate between their impact. Again, to capture the temporal interrelationship, recurrent neural networks are common in state-of-the-art approaches that often fail to capture long-range dependencies. Furthermore, traffic data shows repeated patterns in a daily or weekly period which should be addressed explicitly. To address these limitations, we have designed a Simplified Spatio-temporal Traffic forecasting GNN(SST-GNN) that effectively encodes the spatial dependency by separately aggregating different neighborhood representations rather than with multiple layers and capture the temporal dependency with a simple yet effective weighted spatio-temporal aggregation mechanism. We capture the periodic traffic patterns by using a novel position encoding scheme with historical and current data in two different models. With extensive experimental analysis, we have shown that our model has significantly outperformed the state-of-the-art models on three real-world traffic datasets from the Performance Measurement System (PeMS).
