Enhanced Air Quality Prediction through Spatio-temporal Feature Sxtraction and Fusion: A Self-tuning Hybrid Approach with GCN and GRU

作者：Bao Liu, Zhi Qi, Lei Gao · 发表于：Water Air & Soil Pollution · 年份：2024 · DOI：10.1007/s11270-024-07346-4 · 被引用次数：33 · 研究领域：Air Quality Monitoring and Forecasting、Air Quality and Health Impacts、Urban Heat Island Mitigation

Abstract Accurate prediction of air quality change is essential for air pollution control and human daily mobility. Due to the strong spatial and temporal correlation of air quality changes, existing air quality prediction methods often face the problem of low prediction accuracy due to insufficient extraction of spatio-temporal features. In this paper, we proposed a self-tuning spatio-temporal neural network (ST2NN) to enhance air quality prediction. ST2NN model consisted of four modules. First, ST2NN model constructed a temporal feature extraction module and a spatial feature extraction module based on gated recurrent unit (GRU) and graph convolutional neural network (GCN), respectively, and the two feature extraction modules adopted a parallel structure, which could effectively extract the spatio-temporal features in data. Additionally, ST2NN model constructed a feature fusion module based on gating mechanism to delineate the contribution of spatio-temporal features to the predicted values. Further, ST2NN model constructed a Hyperband hyperparameter optimization module based on Hyperband optimization algorithm to automatically adjust the network hyperparameters. The structure of ST2NN model endowed it with excellent spatio-temporal feature extraction and parameter adaptability. ST2NN model was evaluated and compared with existing models, including convolutional long short-term memory neural network (ConvLSTM), GRU, combined convolutional neural network and long short-term ...