A multi-scale evolutionary deep learning model based on CEEMDAN, improved whale optimization algorithm, regularized extreme learning machine and LSTM for AQI prediction

Abstract

With the rapid development of economy, air pollution occurs frequently, which has a huge negative impact on human health and urban ecosystem. Air quality index (AQI) can directly reflect the degree of air pollution. Accurate AQI trend prediction can provide reliable information for the prevention and control of air pollution, but traditional forecasting methods have limited performance. To this end, a dual-scale ensemble learning framework is proposed for the complex AQI time series prediction. First, complete ensemble empirical mode decomposition adaptive noise (CEEMDAN) and sample entropy (SE) are used to decompose and reconstruct AQI series to reduce the difficulty of direct modeling. Then, according to the characteristics of high and low frequencies, the high-frequency components are predicted by the long short-term memory neural network (LSTM), and the low-frequency items are predicted by the regularized extreme learning machine (RELM). At the same time, the improved whale optimization algorithm (WOA) is used to optimize the hyper-parameters of RELM and LSTM models. Finally, the hybrid prediction model proposed in this paper predicts the AQI of four cities in China. This work effectively improves the prediction accuracy of AQI, which is of great significance to the sustainable development of the cities.

Keywords: CEEMDAN; Long short-term memory; Regularized extreme learning machine; Whale optimization algorithm.