Multi-scale attention graph convolutional recurrent network for traffic forecasting

被引:2
作者
Xiong, Liyan [1 ]
Hu, Zhuyi [1 ]
Yuan, Xinhua [1 ]
Ding, Weihua [1 ]
Huang, Xiaohui [1 ]
Lan, Yuanchun [1 ]
机构
[1] East China Jiaotong Univ, Informat Engn, Shuanggang East St, Nanchang 330013, Jiangxi, Peoples R China
来源
CLUSTER COMPUTING-THE JOURNAL OF NETWORKS SOFTWARE TOOLS AND APPLICATIONS | 2024年 / 27卷 / 03期
基金
中国国家自然科学基金;
关键词
Traffic flow prediction; Graph convolutional network; Attention mechanism; Adaptive node; Multi-scale;
D O I
10.1007/s10586-023-04140-5
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In the backdrop of an ever-expanding urban transportation road network, the dramatic changes in traffic flow make traffic flow forecasting become a challenge. Which encompass intricate spatial correlations and intricate non-linear temporal relationships. Given a complex time series of traffic flow, accurately predicting traffic flow under such circumstances becomes an arduous task, particularly for long-term sequences. To tackle this conundrum, we put together a traffic prediction framework called the multi-scale attention-based graph convolutional recurrent network (MAGRN). Based on GCN and RNN, this framework skilfully extracts dynamic spatial and temporal features from node attributes. By incorporating multi-scale feature extraction and two attention mechanisms within the attention structure, MAGRN adeptly captures features at varying granularities. This enables MAGRN to diligently capture both the temporal interdependencies and spatial correlations inherent in traffic sequences, thereby minimizing errors. Experimental validation on three real-world traffic datasets unequivocally attests to the superior performance of the MAGRN model.
引用
收藏
页码:3277 / 3291
页数:15
相关论文
共 36 条
[1]  
Ajit A., 2020, 2020 INT C EMERGING, DOI [10.1109/ic-ETITE47903.2020.049, DOI 10.1109/IC-ETITE47903.2020.049, 10.1109/ic-etite47903.2020.049]
[2]  
Bai L., 2019, ARXIV
[3]  
Bai L, 2020, ADV NEUR IN, V33
[4]   Passenger Demand Forecasting with Multi-Task Convolutional Recurrent Neural Networks [J].
Bai, Lei ;
Yao, Lina ;
Kanhere, Sala S. ;
Yang, Zheng ;
Chu, Jing ;
Wang, Xianzhi .
ADVANCES IN KNOWLEDGE DISCOVERY AND DATA MINING, PAKDD 2019, PT II, 2019, 11440 :29-42
[5]  
Chen WQ, 2020, AAAI CONF ARTIF INTE, V34, P3529
[6]  
Chung J., 2014, Empirical evaluation of gated recurrent neural networks on sequence modeling
[7]  
Cushing M. J.., 2006, J. Forensic Econ., V19, P139
[8]  
Geng X, 2019, AAAI CONF ARTIF INTE, P3656
[9]  
Guo SN, 2019, AAAI CONF ARTIF INTE, P922
[10]  
Haichao H., 2021, J E CHINA JIAOTONG U, DOI [10.16749/j.cnki.jecjtu.20210706.018, DOI 10.16749/J.CNKI.JECJTU.20210706.018]