PM concentration prediction using a whale optimization algorithm based hybrid deep learning model in Beijing, China.

Journal: Environmental pollution (Barking, Essex : 1987)
PMID:

Abstract

PM is a significant global atmospheric pollutant impacting visibility, climate, and public health. Accurate prediction of PM concentrations is critical for assessing air pollution risks and providing early warnings for effective management. This study proposes a novel hybrid machine learning model that combines the whale optimization algorithm (WOA) with a convolutional neural network (CNN), long short-term memory (LSTM), and an attention mechanism (AM) to predict daily PM concentrations. Tested with meteorological and air pollution daily data from 2014 to 2018, the WOA-CNN-LSTM-AM model demonstrates substantial improvements. It achieves MAE, RMSE, MBE, and R values of 14.29, 21.96, -0.23, and 0.93, respectively, showing a reduction in prediction errors by 39% compared to CNN and 34% compared to LSTM models. In the medium-term forecast, the accuracy of the hybrid model is 30%-54% over WOA-CNN-LSTM and 26%-39% over CNN-LSTM-AM. The R value decreases by 2.5% from the 1-day to 5-day forecast, maintaining high accuracy. SHAP analysis reveals that NO and CO are the primary drivers for PM predictions. This study provides a reliable tool for short and medium-term PM prediction and air pollution control.

Authors

  • Qing Wei
    Department of Computer Science and Department of Biological Science, Purdue University, West Lafayette, IN 47907, USA.
  • Huijin Zhang
    College of Environmental Science and Engineering, Tongji University, Shanghai, 200092, China; Key Laboratory of Urban Water Supply, Water Saving and Water Environment Governance in the Yangtze River Delta of Ministry of Water Resources, State Key Laboratory of Pollution Control and Resource Reuse, Tongji University, Shanghai, 200092, China.
  • Ju Yang
    Guangdong Institute of Water Resources and Hydropower Research, Guangzhou, 510000, China.
  • Bin Niu
    PowerChina East China Survey, Design and Research Institute Co. Ltd., Hangzhou, 310000, China.
  • Zuxin Xu
    State Key Laboratory of Pollution Control and Resource Reuse, College of Environmental Science and Engineering, Tongji University, 1239 Siping Road, Yangpu District, Shanghai 200092, China; Ministry of Education Key Laboratory of Yangtze River Water Environment, Tongji University, Shanghai 200092, China; Shanghai Institute of Pollution Control and Ecological Security, Shanghai, 200092, China.