Residential College | false |
Status | 已發表Published |
Spatio-temporal attention based collaborative local–global learning for traffic flow prediction | |
Chi, Haiyang1,2; Lu, Yuhuan3,4; Xie, Can1,2; Ke, Wei1,2![]() | |
2025-01 | |
Source Publication | Engineering Applications of Artificial Intelligence
![]() |
ISSN | 0952-1976 |
Volume | 139Pages:109575 |
Abstract | Traffic flow prediction is crucial for intelligent transportation systems (ITS), providing valuable insights for traffic control, route planning, and operation management. Existing work often separately models the spatial and temporal dependencies and primarily relies on predefined graphs to represent spatio-temporal dependencies, neglecting the traffic dynamics caused by unexpected events and the global relationships among road segments. Unlike previous models that primarily focus on local feature extraction, we propose a novel collaborative local–global learning model (LOGO) that employs spatio-temporal attention (STA) and graph convolutional networks (GCN). Specifically, LOGO simultaneously extracts hidden traffic features from both local and global perspectives. In local feature extraction, a novel STA is devised to directly attend to spatio-temporal coupling interdependencies instead of separately modeling temporal and spatial dependencies, and to capture in-depth spatio-temporal traffic context with an adaptive graph focusing on the dynamics in traffic flow. In global feature extraction, a global correlation matrix is constructed and GCNs are utilized to propagate messages on the obtained matrix to achieve interactions between both adjacent and similar road segments. Finally, the obtained local and global features are concatenated and fed into a gated aggregation to forecast future traffic flow. Extensive experiments on four real-world traffic datasets sourced from the Caltrans Performance Measurement System (PEMS03, PEMS04, PEMS07, and PEMS08) demonstrate the effectiveness of our proposed model. LOGO achieves the best performance over 18 state-of-the-art baselines and the best prediction performance with the highest improvement of 6.06% on the PEMS07 dataset. Additionally, two real-world case studies further substantiate the robustness and interpretability of LOGO. |
Keyword | Traffic Flow Prediction Spatio-temporal Correlation Local–global Spatio-temporal Feature Graph Convolutional Network Spatio-temporal Attention |
DOI | 10.1016/j.engappai.2024.109575 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Automation & Control Systems ; Computer Science ; Engineering |
WOS Subject | Automation & Control Systems ; Computer Science, Artificial Intelligence ; Engineering, Multidisciplinary ; Engineering, Electrical & Electronic |
WOS ID | WOS:001356363400001 |
Publisher | PERGAMON-ELSEVIER SCIENCE LTD, THE BOULEVARD, LANGFORD LANE, KIDLINGTON, OXFORD OX5 1GB, ENGLAND |
Scopus ID | 2-s2.0-85208536854 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) Faculty of Science and Technology DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Ke, Wei |
Affiliation | 1.Faculty of Applied Sciences, Macao Polytechnic University, Macao 2.Engineering Research Centre of Applied Technology on Machine Translation and Artificial Intelligence, Ministry of Education, Macao Polytechnic University, Macao 3.The State Key Laboratory of Internet of Things for Smart City, University of Macau, Macao 4.Department of Computer and Information Science, University of Macau, Macao |
Recommended Citation GB/T 7714 | Chi, Haiyang,Lu, Yuhuan,Xie, Can,et al. Spatio-temporal attention based collaborative local–global learning for traffic flow prediction[J]. Engineering Applications of Artificial Intelligence, 2025, 139, 109575. |
APA | Chi, Haiyang., Lu, Yuhuan., Xie, Can., Ke, Wei., & Chen, Bidong (2025). Spatio-temporal attention based collaborative local–global learning for traffic flow prediction. Engineering Applications of Artificial Intelligence, 139, 109575. |
MLA | Chi, Haiyang,et al."Spatio-temporal attention based collaborative local–global learning for traffic flow prediction".Engineering Applications of Artificial Intelligence 139(2025):109575. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment