Prediction of air particulate matter in Beijing, China, based on the improved particle swarm optimization algorithm and long short-term memory neural network

Abstract

Intelligent algorithms using deep learning can help learn feature data with nonlinearity and uncertainty, such as time-series particle concentration data. This paper proposes an improved particle swarm optimization (IPSO) algorithm using nonlinear decreasing weights to optimize the hyperparameters, such as the number of hidden layer neurons, learning rate, and maximum number of iterations of the long short-term memory (LSTM) neural network, to predict the time series for air particulate concentration and capture its data dependence. The IPSO algorithm uses nonlinear decreasing weights to make the inertia weights nonlinearly decreasing during the iteration process to improve the convergence speed and capability of finding the global optimization of the PSO. This study addresses the limitations of the traditional method and exhibits accurate predictions. The results of the improved algorithm reveal that the root means square, mean absolute percentage error, and mean absolute error of the IPSO-LSTM model predicted changes in six particle concentrations, which decreased by 1.59% to 5.35%, 0.25% to 3.82%, 7.82% to 13.65%, 0.7% to 3.62%, 0.01% to 3.55%, and 1.06% to 17.21%, respectively, compared with the LSTM and PSO-LSTM models. The IPSO-LSTM prediction model has higher accuracy than the other models, and its accurate prediction model is suitable for regional air quality management and effective control of the adverse effects of air pollution.