Smart City Gnosys

Smart city article details

Title Transformer-Based Spatial-Temporal Graph Attention Network For Traffic Flow Prediction
ID_Doc 58803
Authors Yan F.; Chen Q.
Year 2023
Published Proceedings - 2023 International Conference on Cyber-Enabled Distributed Computing and Knowledge Discovery, CyberC 2023
DOI http://dx.doi.org/10.1109/CyberC58899.2023.00031
Abstract Traffic flow prediction, which plays an important role in intelligent traffic systems, has become a pressing problem to be addressed with the continuous development of smart cities. Currently, the fundamental obstacle lies in effectively modelling the complex spatial-temporal dependencies present in traffic flow data. Deep learning models such as Graph Neural Network based models and Transformer based models have shown promising results in this field. However, methods founded on a single model or framework have one significant limitation: Such methods cannot adequately represent the spatial and temporal features of traffic flow data, restricting the model's ability to learn the dynamics of urban transportation. In this paper, we propose a transformer-based spatial-temporal graph attention network model called TSTGAT for traffic flow prediction, which integrates Transformer and Graph Attention Network. Experiments on two real-world traffic datasets from the Caltrans Performance Measurement System (PeMS) demonstrate that the proposed TSTGAT model outperforms well-known baselines. © 2023 IEEE.
Author Keywords deep learning; graph neural network; Traffic flow prediction; transformer


Similar Articles


Id Similarity Authors Title Published
58593 View0.934Wang X.; Ma Y.; Wang Y.; Jin W.; Wang X.; Tang J.; Jia C.; Yu J.Traffic Flow Prediction Via Spatial Temporal Graph Neural NetworkThe Web Conference 2020 - Proceedings of the World Wide Web Conference, WWW 2020 (2020)
11029 View0.934Zhang H.; Liu J.; Tang Y.; Xiong G.Attention Based Graph Covolution Networks For Intelligent Traffic Flow AnalysisIEEE International Conference on Automation Science and Engineering, 2020-August (2020)
58579 View0.927Zhao J.Traffic Flow Prediction Based On Adjacency Graph And Attention Mechanism2024 5th International Seminar on Artificial Intelligence, Networking and Information Technology, AINIT 2024 (2024)
35590 View0.926Remmouche B.; Boukraa D.; Zakharova A.; Bouwmans T.; Taffar M.Long-Term Spatio-Temporal Graph Attention Network For Traffic ForecastingExpert Systems with Applications, 288 (2025)
52517 View0.925Huang X.; Pan Z.; Zhao G.Spatial-Temporal Interactive Graph Convolutional Networks For Traffic Forecasting2024 4th International Conference on Electronic Information Engineering and Computer Technology, EIECT 2024 (2024)
52568 View0.924Ren H.; Kang J.; Zhang K.Spatio-Temporal Graph-Tcn Neural Network For Traffic Flow Prediction2022 19th International Computer Conference on Wavelet Active Media Technology and Information Processing, ICCWAMTIP 2022 (2022)
11045 View0.923Yin S.; Wang J.; Cui Z.; Wang Y.Attention-Enabled Network-Level Traffic Speed Prediction2020 IEEE International Smart Cities Conference, ISC2 2020 (2020)
11043 View0.922He Q.; Xia D.; Li J.; Yang J.; Hu Y.; Li Y.; Li H.Attention-Based Spatiotemporal Adaptive Graph Diffusion Convolutional Network For Traffic Flow PredictionTransportation Research Record (2025)
58576 View0.921Zheng L.; Pu Y.; Sun W.Traffic Flow Prediction Algorithm Based On Attention Spatiotemporal Graph Convolution MechanismIEEE Advanced Information Technology, Electronic and Automation Control Conference (IAEAC) (2024)
44472 View0.92Nie X.; Peng J.; Wu Y.; Gupta B.B.; El-Latif A.A.A.Real-Time Traffic Speed Estimation For Smart Cities With Spatial Temporal Data: A Gated Graph Attention Network ApproachBig Data Research, 28 (2022)