Residential Collegefalse
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; Chen, Bidong1,2
2025-01
Source PublicationEngineering Applications of Artificial Intelligence
ISSN0952-1976
Volume139Pages: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.

KeywordTraffic Flow Prediction Spatio-temporal Correlation Local–global Spatio-temporal Feature Graph Convolutional Network Spatio-temporal Attention
DOI10.1016/j.engappai.2024.109575
URLView the original
Indexed BySCIE
Language英語English
WOS Research AreaAutomation & Control Systems ; Computer Science ; Engineering
WOS SubjectAutomation & Control Systems ; Computer Science, Artificial Intelligence ; Engineering, Multidisciplinary ; Engineering, Electrical & Electronic
WOS IDWOS:001356363400001
PublisherPERGAMON-ELSEVIER SCIENCE LTD, THE BOULEVARD, LANGFORD LANE, KIDLINGTON, OXFORD OX5 1GB, ENGLAND
Scopus ID2-s2.0-85208536854
Fulltext Access
Citation statistics
Document TypeJournal article
CollectionTHE 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 AuthorKe, Wei
Affiliation1.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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Chi, Haiyang]'s Articles
[Lu, Yuhuan]'s Articles
[Xie, Can]'s Articles
Baidu academic
Similar articles in Baidu academic
[Chi, Haiyang]'s Articles
[Lu, Yuhuan]'s Articles
[Xie, Can]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Chi, Haiyang]'s Articles
[Lu, Yuhuan]'s Articles
[Xie, Can]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.