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

• Published in: Environmental research 2022-12, Vol.215, p.114228, Article 114228
• Main Authors: Ji, Chunlei, Zhang, Chu, Hua, Lei, Ma, Huixin, Nazir, Muhammad Shahzad, Peng, Tian
• Format: Article
• Language: English
• Subjects: CEEMDAN; Long short-term memory; Regularized extreme learning machine; Whale optimization algorithm
• ISSN: 0013-9351; 1096-0953
• DOI: 10.1016/j.envres.2022.114228

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. •A dual-scale ensemble framework is proposed for air quality index prediction.•CEEMDAN technology is adopted to obtain multiple intrinsic mode functions.•An improved WOA algorithm based on dimension learning-based hunting is proposed.•Long short-term memory network is used to predict high-frequency components.•Regularized extremum learning machine is used to predict low-frequency components.