Dual-channel spatial-temporal difference graph neural network for PM2.5 forecasting

被引:0
作者
Ouyang, Xiaocao [1 ,2 ]
Yang, Yan [1 ,2 ]
Zhang, Yiling [1 ,2 ]
Zhou, Wei [1 ,2 ]
Guo, Dongyu [1 ]
机构
[1] Southwest Jiaotong Univ, Sch Comp & Artificial Intelligence, Chengdu 611756, Peoples R China
[2] Mfg Ind Chains Collaborat & Informat Support Tech, Chengdu 611756, Peoples R China
基金
中国国家自然科学基金;
关键词
PM2.5; forecasting; Graph neural network; Spatial-temporal data; AIR; PREDICTION; POLLUTION; IMPACT; TIME;
D O I
10.1007/s00521-022-08036-0
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Accurate PM2.5 forecasting is significant for improving quality of life and human health. However, it is very challenging to capture the high spatiotemporal correlations and the complex diffusion processes of PM2.5. Most existing PM2.5 prediction methods only focus on spatiotemporal dependencies. In addition, the PM2.5 diffusion process with domain knowledge in deep learning is rarely considered. Therefore, how to simultaneously capture comprehensive spatiotemporal dependencies and model the complicated diffusion process of PM2.5 is still a challenge. To address this problem, we propose a dual-channel spatial-temporal difference graph neural network (DC-STDGN) to forecast future PM2.5 concentrations. DC-STDGN first constructs a dual-channel structure to obtain distance-based local neighboring information and the global hidden spatial correlation of the data. Then, a temporal convolution layer is designed to handle the long-term dependency. Finally, the spatial difference with domain knowledge is introduced to model the complex diffusion process and capture more comprehensive spatiotemporal correlations. The extensive experiments with three real-world datasets demonstrate the improved prediction performance of DC-STDGN over state-of-the-art baselines. DC-STDGN outperforms the second-best model by up to 16.9% improvement in mean absolute error, 8.9% improvement in root mean square error and 18.2% improvement in mean absolute scaled error.
引用
收藏
页码:7475 / 7494
页数:20
相关论文
共 40 条
[1]   Evaluation of the Community Multiscale Air Quality (CMAQ) model version 4.5: Sensitivities impacting model performance Part I - Ozone [J].
Appel, K. Wyat ;
Gilliland, Alice B. ;
Sarwar, Golam ;
Gilliam, Robert C. .
ATMOSPHERIC ENVIRONMENT, 2007, 41 (40) :9603-9615
[2]  
Chung J., 2014, P ADV NEUR INF PROC, P1
[3]   Deep Air Quality Forecasting Using Hybrid Deep Learning Framework [J].
Du, Shengdong ;
Li, Tianrui ;
Yang, Yan ;
Horng, Shi-Jinn .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2021, 33 (06) :2412-2424
[4]   Multivariate time series forecasting via attention-based encoder-decoder framework [J].
Du, Shengdong ;
Li, Tianrui ;
Yang, Yan ;
Horng, Shi-Jinn .
NEUROCOMPUTING, 2020, 388 :269-279
[5]   Multi-scale spatiotemporal graph convolution network for air quality prediction [J].
Ge, Liang ;
Wu, Kunyan ;
Zeng, Yi ;
Chang, Feng ;
Wang, Yaqian ;
Li, Siyu .
APPLIED INTELLIGENCE, 2021, 51 (06) :3491-3505
[6]   Recurrent Air Quality Predictor Based on Meteorology- and Pollution-Related Factors [J].
Gu, Ke ;
Qiao, Junfei ;
Lin, Weisi .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2018, 14 (09) :3946-3955
[7]  
Guan WJ, 2016, LANCET, V388, P1939, DOI 10.1016/S0140-6736(16)31597-5
[8]  
Han J, 2021, P 35 AAAI C ART INT
[9]   Identity Mappings in Deep Residual Networks [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
COMPUTER VISION - ECCV 2016, PT IV, 2016, 9908 :630-645
[10]  
Hochreiter S, 1997, NEURAL COMPUT, V9, P1735, DOI [10.1162/neco.1997.9.1.1, 10.1007/978-3-642-24797-2]