Explainable deep learning hybrid modeling framework for total suspended particles concentrations prediction

  • Sujan Ghimire
  • , Ravinesh C. Deo*
  • , Ningbo Jiang
  • , A. A.Masrur Ahmed
  • , Salvin S. Prasad
  • , David Casillas-Pérez
  • , Sancho Salcedo-Sanz
  • , Zaher Mundher Yaseen
  • *Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

2 Scopus citations

Abstract

Total Suspended Particles (TSP) is an important indicator of air quality, yet traditional prediction models lack comprehensive consideration of spatio-temporal interactions of different meteorological and air pollution phenomena. To address these limitations, this study introduces an explainable (X) deep hybrid (H) network, integrating Convolutional Neural Networks (CNN) and Bidirectional Gated Recurrent Units (BGRU), for hourly TSP concentration prediction. The model was trained and evaluated using meteorological and air quality data from Canon Hill, Australia. By combining CNN's spatial feature extraction capabilities with BGRU's temporal dependencies, the model effectively captures complex spatial–temporal patterns in the data. The X-H-CBGRU model outperforms fifteen competing benchmark models such as deep neural network, extreme learning machine, multilayer perceptron, support vector regression, random forest regression, light gradient boosting, gradient boosting regression, long short-term memory network, as well as their hybrid CNN counterparts in terms of the accuracy evidenced by a lower Root Mean Square Error (RMSE ≈6.302μg/m3) and higher Correlation Coefficient (r ≈0.91) compared to other models. Moreover, the model demonstrates strong probabilistic performance with a high Prediction Interval Coverage Probability (PICP ≈0.98) and low Prediction Interval Normalized Average Width (PINAW ≈0.18), indicating its reliable prediction intervals. To enhance model interpretability, Shapley Additive Explanations (SHAP) and Local Interpretable Model-Agnostic Explanations (LIME) methods were employed, revealing PM10 concentration, relative humidity, air temperature, and wind speed as key predictors of TSP concentrations. The Diebold–Mariano statistical test further confirmed the model's superior performance. This study contributes towards advancing TSP prediction by providing a robust, accurate, and interpretable model which has particular importance in locations such as mining regions. The X-H-CBGRU model holds potential for improving public health protection and informing air pollution mitigation strategies.

Original languageEnglish
Article number121079
JournalAtmospheric Environment
Volume347
DOIs
StatePublished - 15 Apr 2025

Bibliographical note

Publisher Copyright:
© 2025

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 3 - Good Health and Well-being
    SDG 3 Good Health and Well-being

Keywords

  • Air pollutant prediction
  • Bidirectional GRU
  • Deep learning
  • Explainable artificial intelligence (xAI)
  • Optuna optimization
  • “Black-box” model

ASJC Scopus subject areas

  • General Environmental Science
  • Atmospheric Science

Fingerprint

Dive into the research topics of 'Explainable deep learning hybrid modeling framework for total suspended particles concentrations prediction'. Together they form a unique fingerprint.

Cite this