JOURNAL ARTICLE

STANN: A Spatio–Temporal Attentive Neural Network for Traffic Prediction

Zhixiang HeChi-Yin ChowJia-Dong Zhang

Year: 2018 Journal:   IEEE Access Vol: 7 Pages: 4795-4806   Publisher: Institute of Electrical and Electronics Engineers

Abstract

Recently, traffic prediction based on deep learning methods has attracted much attention. However, there still exist two major challenges, namely, dynamic spatio-temporal dependences among network-wide links and long-term traffic prediction for the next few hours. To address these two challenges, this paper proposes a spatio-temporal attentive neural network (STANN) for the network-wide and long-term traffic prediction. The STANN captures the spatial–temporal dependences based on the encoder–decoder architecture with the attention mechanisms. In the encoder, the STANN learns the spatio-temporal dependences from historical traffic series using a recurrent neural network (RNN) with long short-term memory (LSTM) units, in which a new spatial attention model is developed to consider the contribution of each link to the network-wide prediction. In the decoder, the STANN exploits another RNN with LSTM units and a temporal attention model to select the relevant and important historical spatio-temporal dependences from the encoder for long-term traffic prediction. Finally, we conduct extensive experiments to evaluate STANN on three real-world traffic datasets. The experimental results show that the STANN is significantly better than other state-of-the-art models.

Keywords:
Computer science Recurrent neural network Encoder Exploit Artificial intelligence Term (time) Artificial neural network Deep learning Temporal database Machine learning Data mining

Metrics

63
Cited By
5.90
FWCI (Field Weighted Citation Impact)
39
Refs
0.96
Citation Normalized Percentile
Is in top 1%
Is in top 10%

Citation History

Topics

Traffic Prediction and Management Techniques
Physical Sciences →  Engineering →  Building and Construction
Traffic control and management
Physical Sciences →  Engineering →  Control and Systems Engineering
Transportation Planning and Optimization
Social Sciences →  Social Sciences →  Transportation

Related Documents

JOURNAL ARTICLE

STANN: Spatio-Temporal Attention-based Neural Network for Epidemic Prediction

Zhuanghu LvJing LiDafeng LiuYue PengBenyun Shi

Journal:   IEEE/WIC/ACM International Conference on Web Intelligence Year: 2021 Pages: 314-319
JOURNAL ARTICLE

Spatio-Temporal Wireless Traffic Prediction With Recurrent Neural Network

Chen QiuYanyan ZhangZhiyong FengPing ZhangShuguang Cui

Journal:   IEEE Wireless Communications Letters Year: 2018 Vol: 7 (4)Pages: 554-557
JOURNAL ARTICLE

A macro–micro spatio-temporal neural network for traffic prediction

Siyuan FengShuqing WeiJunbo ZhangYexin LiJintao KeGaode ChenYu ZhengHai Yang

Journal:   Transportation Research Part C Emerging Technologies Year: 2023 Vol: 156 Pages: 104331-104331
JOURNAL ARTICLE

Spatio-Temporal Graph-TCN Neural Network for Traffic Flow Prediction

Hongjin RenJinbiao KangKe Zhang

Journal:   2022 19th International Computer Conference on Wavelet Active Media Technology and Information Processing (ICCWAMTIP) Year: 2022 Pages: 1-4
© 2026 ScienceGate Book Chapters — All rights reserved.