Improvement of AI Forecast of Gridded PM2.5 Forecast in China Through ConvLSTM and Attention

Liu Pei,Yao Erlin,Liu Tao,Kong Lei,Tang Xiao,Tan Guangming
DOI: https://doi.org/10.1007/s42514-021-00087-4
2022-01-01
Abstract:The prediction of $${\mathrm{PM}}_{2.5}$$ concentration has attracted considerable research efforts in recent years. However, due to the lack of open dataset, the data processed by existing intelligent methods are only values at single stations or mean value in a small region, while the data in real applications are all gridded values in large regions. This incompatibility in data format makes intelligent methods cannot be integrated into the practical process of $${\mathrm{PM}}_{2.5}$$ prediction. In this paper, first we build a large dataset with gridded data obtained from the numerical prediction field, then an intelligent prediction method with gridded data as the basic input and output format is proposed. To capture both the spatial and temporal characteristics in data, the ConvLSTM (convolutional long short term memory) model is applied, which can utilize the advantages of both the CNN (convolutional neural network) and LSTM models. However, ConvLSTM has defects in processing multi-feature data: the more features model uses, the worse the forecasting result will be. To improve the prediction accuracy of ConvLSTM further, the attention mechanism is applied, which can describe more accurately the importance of different features and different regions for the prediction accuracy. On the built large dataset of $${\mathrm{PM}}_{2.5}$$ gridded concentrations, when we predict the next hour’s value using the past 6 h, the RMSE (root mean square error) of the conventional MLR (multi-linear regression) and ConvLSTM are respectively 6.44 $$\mu \mathrm{g}/{\mathrm{m}}^{3}$$ and 6.24 $$\mu \mathrm{g}/{\mathrm{m}}^{3}$$ , when the attention mechanism is incorporated into ConvLSTM, the RMSE can be decreased to 4.79 $$\mu \mathrm{g}/{\mathrm{m}}^{3}$$ .
What problem does this paper attempt to address?