A new attention-based CNN_GRU model for spatial-temporal PM2.5 prediction.

ABSTRACT

Accurately predicting the spatial-temporal distribution of PM2.5 is challenging due to missing data and selecting an appropriate modeling method. Effective imputation of missing data must consider the relationships between variables while preserving their inherent variability and uncertainty. In this study, we employed machine learning techniques to impute missing data by analyzing the relationships between meteorological variables and other pollutants. Subsequently, we introduced an innovative spatiotemporal hybrid model, AC_GRU, which integrates a one-dimensional convolutional neural network (CNN), GRU, and an attention-based network to predict PM2.5 concentrations in urban areas. The AC_GRU model utilizes meteorological variables, PM2.5 concentrations from nearby air quality monitoring stations, and concentrations of other pollutants as inputs. This approach allows the model to learn spatiotemporal correlations within the time-series data, enhancing the accuracy of PM2.5 predictions. Additionally, the attention mechanism improves prediction accuracy by automatically weighting the past input variables based on their importance for future PM2.5 predictions. The experimental results demonstrate that our AC_GRU model outperforms state-of-the-art methods, making it a valuable tool for urban air quality management and public health protection.