A Study on Urban Traffic Speed Prediction Using Spatiotemporal Attention-Based Convolutional Neural Networks

Urban traffic speed prediction requires effective modeling of road-network spatial dependencies and temporal dynamics. This study develops an attention-based spatiotemporal graph convolutional framework (AT-GCN-GRU) for multi-horizon traffic speed prediction. The model combines graph convolution for spatial feature extraction, GRU for temporal modeling, and a temporal attention mechanism for adaptively weighting historical hidden states. Experiments were conducted on the SZ-taxi dataset and further evaluated on METR-LA. On SZ-taxi, AT-GCN-GRU was compared with conventional and modern baselines, including T-GCN, ASTGCN, and AGCRN, over 15, 30, 45, and 60 min forecasting horizons. The proposed model remained competitive with the modern baselines and obtained lower mean RMSE than T-GCN across the four evaluated horizons, with relative reductions of approximately 1.4–2.9%. Additional ablation, missing-data robustness, attention-weight visualization, and computational-cost analyses were conducted. The results suggest that temporal attention can provide incremental improvements to the GCN-GRU framework with relatively limited additional computational overhead.

Authors

Institutions

Publication Details

Journal
Electronics
Published
2026-09-11
DOI
https://doi.org/10.3390/electronics15184128
Primary Topic
Traffic Prediction and Management Techniques
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
article

A Study on Urban Traffic Speed Prediction Using Spatiotemporal Attention-Based Convolutional Neural Networks

Hongqiang Xin, Zhuye Xu, Shaoyang Wang, Zhongdong Hua
Electronics
Traffic Prediction and Management Techniques
article

A Study on Urban Traffic Speed Prediction Using Spatiotemporal Attention-Based Convolutional Neural Networks

Hongqiang Xin, Zhuye Xu, Shaoyang Wang, Zhongdong Hua
article en

Abstract

Urban traffic speed prediction requires effective modeling of road-network spatial dependencies and temporal dynamics. This study develops an attention-based spatiotemporal graph convolutional framework (AT-GCN-GRU) for multi-horizon traffic speed prediction. The model combines graph convolution for spatial feature extraction, GRU for temporal modeling, and a temporal attention mechanism for adaptively weighting historical hidden states. Experiments were conducted on the SZ-taxi dataset and further evaluated on METR-LA. On SZ-taxi, AT-GCN-GRU was compared with conventional and modern baselines, including T-GCN, ASTGCN, and AGCRN, over 15, 30, 45, and 60 min forecasting horizons. The proposed model remained competitive with the modern baselines and obtained lower mean RMSE than T-GCN across the four evaluated horizons, with relative reductions of approximately 1.4–2.9%. Additional ablation, missing-data robustness, attention-weight visualization, and computational-cost analyses were conducted. The results suggest that temporal attention can provide incremental improvements to the GCN-GRU framework with relatively limited additional computational overhead.

ElectronicsVol. 15(18)
Lanzhou Jiaotong University (CN), Pingliang People's Hospital (CN), Changsha University of Science and Technology (CN)
Lanzhou Jiaotong University, Gansu Education Department, Science and Technology Department of Gansu Province
Sustainable cities and communities
Openalex Percentile: Top 14%
Traffic Prediction and Management Techniques
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

A Study on Urban Traffic Speed Prediction Using Spatiotemporal Attention-Based Convolutional Neural Networks — Hongqiang Xin, Zhuye Xu, et al. · Electronics (2026) | TGRS Research Map | TGRS