# Spatial-Temporal Feature Extraction and Evaluation Network for Citywide Traffic Condition Prediction

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

## Key facts

- Cited by 2 papers in the map (data through 14 Oct 2026).
- Builds on 13 of its 71 references that are in the map.
- First posted on arXiv 22 Jul 2022.

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

- [Semi-Supervised Classification with Graph Convolutional Networks](https://mlsanity.com/paper/1609.02907/) (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)
- [GMAN: A Graph Multi-Attention Network for Traffic Prediction](https://mlsanity.com/paper/1911.08415/) (2019)
- [T-GCN: A Temporal Graph ConvolutionalNetwork for Traffic Prediction](https://mlsanity.com/paper/1811.05320/) (2018)
- [Spatial-Temporal Transformer Networks for Traffic Flow Forecasting](https://mlsanity.com/paper/2001.02908/) (2020)
- [Revisiting Spatial-Temporal Similarity: A Deep Learning Framework for Traffic Prediction](https://mlsanity.com/paper/1803.01254/) (2018)
- [A3T-GCN: Attention Temporal Graph Convolutional Network for Traffic Forecasting](https://mlsanity.com/paper/2006.11583/) (2020)
- [Deep-learning Architecture for Short-term Passenger Flow Forecasting in Urban Rail Transit](https://mlsanity.com/paper/1912.12563/) (2019)
- [Relational Fusion Networks: Graph Convolutional Networks for Road Networks](https://mlsanity.com/paper/2006.09030/) (2020)
- [Wireless Traffic Prediction with Scalable Gaussian Process: Framework, Algorithms, and Verification](https://mlsanity.com/paper/1902.04763/) (2019)
- [Traffic4cast -- Large-scale Traffic Prediction using 3DResNet and Sparse-UNet](https://mlsanity.com/paper/2111.05990/) (2021)
- [MAF-GNN: Multi-adaptive Spatiotemporal-flow Graph Neural Network for Traffic Speed Forecasting](https://mlsanity.com/paper/2108.03594/) (2021)

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

- [HUTFormer: Hierarchical U-Net Transformer for Long-Term Traffic Forecasting](https://mlsanity.com/paper/2307.14596/) (2023)
- [SocialTraj: Two-Stage Socially-Aware Trajectory Prediction for Autonomous Driving via Conditional Diffusion Model](https://mlsanity.com/paper/2509.17850/) (2025)

## Abstract (arXiv)

Traffic prediction plays an important role in the realization of traffic control and scheduling tasks in intelligent transportation systems. With the diversification of data sources, reasonably using rich traffic data to model the complex spatial-temporal dependence and nonlinear characteristics in traffic flow are the key challenge for intelligent transportation system. In addition, clearly evaluating the importance of spatial-temporal features extracted from different data becomes a challenge. A Double Layer - Spatial Temporal Feature Extraction and Evaluation (DL-STFEE) model is proposed. The lower layer of DL-STFEE is spatial-temporal feature extraction layer. The spatial and temporal features in traffic data are extracted by multi-graph graph convolution and attention mechanism, and different combinations of spatial and temporal features are generated. The upper layer of DL-STFEE is the spatial-temporal feature evaluation layer. Through the attention score matrix generated by the high-dimensional self-attention mechanism, the spatial-temporal features combinations are fused and evaluated, so as to get the impact of different combinations on prediction effect. Three sets of experiments are performed on actual traffic datasets to show that DL-STFEE can effectively capture the spatial-temporal features and evaluate the importance of different spatial-temporal feature combinations.