Enhancing Air Quality Prediction with an Adaptive PSO-Optimized CNN-Bi-LSTM Model

Abstract

Effective air quality prediction models are crucial for the timely prevention and control of air pollution. However, previous models often fail to fully consider air quality’s temporal and spatial distribution characteristics. In this study, Xi’an City is used as the study area. Data from 1 January 2019 to 31 October 2020 are used as the training set, while data from 1 November 2020 to 31 December 2020 are used as the test set. This paper proposes a multi-time and multi-site air quality prediction model for Xi’an, leveraging a deep learning network model based on APSO-CNN-Bi-LSTM. The CNN model extracts the spatial features of the input data, the Bi-LSTM model extracts the time series features, and the PSO algorithm with adaptive inertia weight (APSO) optimizes the model’s hyperparameters. The results show that the model achieves the best results in terms of MAE and RMSE. Compared to the PSO-SVR, BPTT, CNN-LSTM, and GA-ACO-BP models, the MAE improved by 9.375%, 6.667%, 2.276%, and 4.975%, while the RMSE improved by 8.371%, 8.217%, 6.327%, and 5.293%. These significant improvements highlight the model’s accuracy and its promising application prospects.