Papers › Network Level Spatial Temporal Traffic State Forecasting with Hierarchical Attention...

Network Level Spatial Temporal Traffic State Forecasting with Hierarchical Attention LSTM (HierAttnLSTM)

15 Jan 2022arXiv:2201.05760archive 2025-07-28

Tianya Terry Zhang

Traffic state data, such as speed, volume and travel time collected from ubiquitous traffic monitoring sensors require advanced network level analytics for forecasting and identifying significant traffic patterns. This paper leverages diverse traffic state datasets from the Caltrans Performance Measurement System (PeMS) hosted on the open benchmark and achieved promising performance compared to well recognized spatial-temporal models. Drawing inspiration from the success of hierarchical architectures in various Artificial Intelligence (AI) tasks, we integrate cell and hidden states from low-level to high-level Long Short-Term Memory (LSTM) networks with an attention pooling mechanism, similar to human perception systems. The developed hierarchical structure is designed to account for dependencies across different time scales, capturing the spatial-temporal correlations of network-level traffic states, enabling the prediction of traffic states for all corridors rather than a single link or route. The efficiency of designed attention-based LSTM is analyzed by ablation study. Comparative results with baseline LSTM models demonstrate that the Hierarchical Attention LSTM (HierAttnLSTM) model not only provides higher prediction accuracy but also effectively forecasts unusual congestion patterns. Data and code are made publicly available to support reproducible scientific research.

PaperPDFCode

Code

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Traffic Prediction

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Traffic Prediction PeMSD4 Hierarchical-Attention-LSTM (HierAttnLSTM) 12 steps MAE 9.168 #1 of 13 Archive leaderboard report
Traffic Prediction PeMSD4 Hierarchical-Attention-LSTM (HierAttnLSTM) 12 steps RMSE 22.844 #1 of 13 Archive leaderboard report
Traffic Prediction PeMSD8 Hierarchical-Attention-LSTM (HierAttnLSTM) 12 steps MAE 9.215 #1 of 13 Archive leaderboard report
Traffic Prediction PeMSD8 Hierarchical-Attention-LSTM (HierAttnLSTM) 12 steps RMSE 22.320 #1 of 13 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Methods

Introduced by this paper: Attention Pooling

Attention PoolingHierarchical Feature FusionLSTM

1 archive method tag without a method page not shown.

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections