Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting
Open accessAbstract
Forecasting the traffic flows is a critical issue for researchers and practitioners in the field of transportation. However, it is very challenging since the traffic flows usually show high nonlinearities and complex patterns. Most existing traffic flow prediction methods, lacking abilities of modeling the dynamic spatial-temporal correlations of traffic data, thus cannot yield satisfactory prediction results. In this paper, we propose a novel attention based spatial-temporal graph convolutional network (ASTGCN) model to solve traffic flow forecasting problem. ASTGCN mainly consists of three independent components to respectively model three temporal properties of traffic flows, i.e., recent, daily-periodic and weekly-periodic dependencies. More specifically, each component contains two major parts: 1) the spatial-temporal attention mechanism to effectively capture the dynamic spatialtemporal correlations in traffic data; 2) the spatial-temporal convolution which simultaneously employs graph convolutions to capture the spatial patterns and common standard convolutions to describe the temporal features. The output of the three components are weighted fused to generate the final prediction results. Experiments on two real-world datasets from the Caltrans Performance Measurement System (PeMS) demonstrate that the proposed ASTGCN model outperforms the state-of-the-art baselines.
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Cites
Deep Residual Learning for Image Recognition
He, K., Zhang, X., Ren, S., & Sun, J. (2016) · 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr.2016.90 · Cited by 235,474 · see citing works
Hochreiter, S., & Schmidhuber, J. (1997) · Neural Computation · doi:10.1162/neco.1997.9.8.1735 · Cited by 102,318 · see citing works
Semi-supervised classification with graph convolutional networks
Kipf, T. N., & Welling, M. (2016) · arXiv · doi:10.48550/arxiv.1609.02907 · Cited by 24,853 · see citing works
Veličković, P., Cucurull, G., Casanova, A., Romero, A., Liò, P., & Bengio, Y. (2017) · arXiv · doi:10.48550/arxiv.1710.10903 · Cited by 12,107
Graph Attention NetworksPreprint
Veličković, P., Casanova, A., Liò, P., Cucurull, G., Romero, A., & Bengio, Y. (2017) · arXiv (Cornell University) · doi:10.17863/cam.48429 · Cited by 12,099 · see citing works
Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling
Chung, J., Gulcehre, C., Cho, K., & Bengio, Y. (2014) · arXiv · doi:10.48550/arxiv.1412.3555 · Cited by 11,008 · see citing works
Show, Attend and Tell: Neural Image Caption Generation with Visual Attention
Xu, K., Ba, J., Kiros, R., Cho, K., Courville, A., Salakhutdinov, R., Zemel, R., & Bengio, Y. (2015) · arXiv · doi:10.48550/arxiv.1502.03044 · Cited by 8,248 · see citing works
Advances In Deep Learning On Graphs (Gsp'18 Workshop)Preprint
Defferrard, M., Bresson, X., & Vandergheynst, P. (2016) · arXiv (Cornell University) · doi:10.48550/arxiv.1606.09375 · Cited by 6,912 · see citing works
Advances In Deep Learning On Graphs (Gsp'18 Workshop)
Defferrard, M. (2016) · Figshare · doi:10.5281/zenodo.1286817 · Cited by 4,977 · see citing works
Shuman, D. I., Narang, S. K., Frossard, P., Ortega, A., & Vandergheynst, P. (2013) · IEEE Signal Processing Magazine · doi:10.1109/msp.2012.2235192 · Cited by 4,421 · see citing works
Spectral Networks and Locally Connected Networks on Graphs
Bruna, J., Zaremba, W., Szlam, A., & LeCun, Y. (2013) · arXiv · doi:10.48550/arxiv.1312.6203 · Cited by 3,983 · see citing works
Learning Convolutional Neural Networks for GraphsPreprint
Niepert, M., Ahmed, M., & Kutzkov, K. (2016) · International Conference on Machine Learning · doi:10.48550/arxiv.1605.05273 · Cited by 2,311 · see citing works
Williams, B. M., & Hoel, L. A. (2003) · Journal of Transportation Engineering · doi:10.1061/(asce)0733-947x(2003)129:6(664 · Cited by 2,184 · see citing works
Data-Driven Intelligent Transportation Systems: A Survey
Zhang, J., Wang, F.-Y., Wang, K., Lin, W.-H., Xu, X., & Chen, C. (2011) · IEEE Transactions on Intelligent Transportation Systems · doi:10.1109/tits.2011.2158001 · Cited by 1,817 · see citing works
Dynamic Edge-Conditioned Filters in Convolutional Neural Networks on Graphs
Simonovsky, M., & Nikos Komodakis, A. (2017) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr.2017.11 · Cited by 1,346 · see citing works
Deep Convolutional Networks on Graph-Structured Data
Henaff, M., Bruna, J., & LeCun, Y. (2015) · arXiv · doi:10.48550/arxiv.1506.05163 · Cited by 1,255 · see citing works
Deep Multi-View Spatial-Temporal Network for Taxi Demand Prediction
Yao, H., Wu, F. Q., Ke, J., Tang, X., Jia, Y., Lu, S., Gong, P., Ye, J., & Li, Z. (2018) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v32i1.11836 · Cited by 1,222 · see citing works
Freeway Performance Measurement System: Mining Loop Detector Data
Chen, C., Petty, K. F., Skabardonis, A., Varaiya, P. P., & Jia, Z. (2001) · Transportation Research Record Journal of the Transportation Research Board · doi:10.3141/1748-12 · Cited by 759 · see citing works
GeoMAN: Multi-level Attention Networks for Geo-sensory Time Series Prediction
Liang, Y., Ke, S., Zhang, J., Yi, X., & Zheng, Y. (2018) · International Joint Conference on Artificial Intelligence · doi:10.24963/ijcai.2018/476 · Cited by 539 · see citing works
Predicting citywide crowd flows using deep spatio-temporal residual networks
Zhang, J., Zheng, Y., Qi, D., Li, R., Yi, X., & Li, T. (2018) · Artificial Intelligence · doi:10.1016/j.artint.2018.03.002 · Cited by 524 · see citing works
Supervised Weighting-Online Learning Algorithm for Short-Term Traffic Flow Prediction
Jeong, Y., Byon, Y.-J., de Castro-Neto, M. M., & Easa, S. M. (2013) · IEEE Transactions on Intelligent Transportation Systems · doi:10.1109/tits.2013.2267735 · Cited by 334 · see citing works
Short-Term Traffic and Travel Time Prediction Models
van Lint, H., & van Hinsbergen, C. (2012) · Transportation research circular · Cited by 245 · see citing works
Vector Autoregressive Models for Multivariate Time Series
Zivot, E. W., & Wang, J. (2003) · Springer eBooks · doi:10.1007/978-0-387-21763-5_11 · Cited by 239 · see citing works
Spatio-Temporal Graph Convolution for Skeleton Based Action Recognition
Li, C., Cui, Z., Zheng, W., Xu, C., & Yang, J. J. (2018) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v32i1.11776 · Cited by 193 · see citing works
Vector Autoregressive Models for Multivariate Time Series
(2007) · Springer eBooks · doi:10.1007/978-0-387-32348-0_11 · Cited by 180 · see citing works
Effective Deep Memory Networks for Distant Supervised Relation Extraction
Feng, X., Guo, J., Qin, B., Liu, T., & Liu, Y. (2017) · International Joint Conference on Artificial Intelligence · doi:10.24963/ijcai.2017/559 · Cited by 126 · see citing works
Modeling Spatial-Temporal Dynamics for Traffic Prediction.Preprint
Yao, H., Tang, X., Wei, H., Zheng, G., Yu, Y., & Li, Z. (2018) · arXiv (Cornell University) · Cited by 96 · see citing works
Predicting Citywide Crowd Flows Using Deep Spatio-Temporal Residual NetworksPreprint
Zhang, J., Zheng, Y., Qi, D., et al. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1701.02543 · Cited by 31 · see citing works
Spatio-Temporal Graph Convolution for Skeleton Based Action RecognitionPreprint
Li, C., Cui, Z., Zheng, W., et al. (2018) · arXiv (Cornell University) · doi:10.48550/arxiv.1802.09834 · Cited by 31 · see citing works
Chen, T., Wang, Z., Li, L., Li, D., Li, Z., Chang, X., Bian, P., Liang, G., Wang, Q., & Xie, T. (2025) · Schloss Dagstuhl – Leibniz-Zentrum für Informatik · doi:10.4230/lipics.ecoop.2025.6
Flagged first, then by citations
Research works printed but not linked: 3
- Defferrard, M.; Bresson, X.; and Vandergheynst, P. 2016. Con- volutional neural networks on graphs with fast localized spectral filtering. In Advances in Neural Information Processing Systems, 3844-3852.
- Hochreiter, S., and Schmidhuber, J. 1997. Long short-term mem- ory. Neural Computation 9(8):1735-1780.
- Yu, B.; Yin, H.; and Zhu, Z. 2018. Spatio-Temporal Graph Convo- lutional Networks: A Deep Learning Framework for Traffic Fore- casting. In International Joint Conference on Artificial Intelli- gence, 3634-3640.
Cited by
Sorted by citations
A Comprehensive Survey on Graph Neural Networks
Wu, Z., Pan, S., Chen, F., Long, G., Zhang, C., & Yu, P. S. (2021) · IEEE Transactions on Neural Networks and Learning Systems · doi:10.1109/tnnls.2020.2978386 · Cited by 11,000 · see citing works
Graph neural networks: A review of methods and applications
Zhou, J., Cui, G., Hu, S., Zhang, Z., Yang, C., Liu, Z., Wang, L., Li, C., & Sun, M. (2020) · AI Open · doi:10.1016/j.aiopen.2021.01.001 · Cited by 6,880 · see citing works
Short-Term Traffic Flow Prediction Based on Graph Convolutional Networks and Federated Learning
Xia, M., Jin, D., & Chen, J. (2023) · IEEE Transactions on Intelligent Transportation Systems · doi:10.1109/tits.2022.3179391 · Cited by 142 · see citing works
Yang, J., Zhang, L., Chen, C., Li, Y., Li, R., Wang, G., Jiang, S., & Zeng, Z. (2020) · Information Sciences · doi:10.1016/j.ins.2020.05.090 · Cited by 123 · see citing works
Fine-Grained Urban Flow Prediction
Liang, Y., Ouyang, K., Sun, J., Wang, Y., Zhang, J., Zheng, Y., Rosenblum, D., & Zimmermann, R. (2021) · Proceedings of the Web Conference 2021 · doi:10.1145/3442381.3449792 · Cited by 89 · see citing works
Ren, R., Wang, Y., Liu, F., Li, Z., & Xie, G. (2023) · 2023 IEEE/ACM 31st International Symposium on Quality of Service (IWQoS) · doi:10.1109/iwqos57198.2023.10188773 · Cited by 16 · see citing works
Informer: irregular traffic detection for containerized microservices RPC in the real world
Chen, J., Huang, H., & Chen, H. (2019) · Proceedings of the 4th ACM/IEEE Symposium on Edge Computing · doi:10.1145/3318216.3363375 · Cited by 6 · see citing works
- Authors
- Guo, S., Lin, Y., Feng, N., Song, C., & Wan, H.
- Affiliations
- Beijing Jiaotong University (CN)
- Venue
- Proceedings of the AAAI Conference on Artificial Intelligence
- Published
- 2019-07-17
- Volume
- 33
- Issue
- 01
- Pages
- 922-929
- ISSN
- 2159-5399 (ISSN-L)
- 2374-3468
- OpenAlex
- W2903871660
- Cited by
- 3,123
- References
- 30
- Citations by year
- 2026: 373 · 2025: 710 · 2024: 700 · 2023: 552 · 2022: 402 · 2021: 261 · 2020: 118 · 2019: 7
- FWCI
- 332.36
- Type
- conference paper
- Language
- en
- Open access
- diamond
- doi.org
- Licence
- www.aaai.org/ (unspecified)
- Locations
- Proceedings of the AAAI Conference on Artificial Intelligence (publishedVersion)
- AAAI Publications (The Association for the Advancement of Artificial Intelligence (AAAI)) (submittedVersion)
- Funders
- National Natural Science Foundation of China (61603028)
- Topics
- Physical Sciences › Engineering › Building and Construction › Traffic Prediction and Management Techniques
- Physical Sciences › Engineering › Control and Systems Engineering › Traffic control and management
- Physical Sciences › Computer Science › Artificial Intelligence › Advanced Graph Neural Networks
- Keywords
- traffic flow prediction, spatio-temporal graph convolutional network, attention mechanism, spatio-temporal correlation, feature fusion, PEMS, traffic prediction
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review