Ensemble learning for air quality index prediction: integrating gradient boosting, XGBoost, and stacking with SHAP-based interpretability

基于集成学习的空气质量指数预测:融合梯度提升、XGBoost和堆叠算法以及基于SHAP的可解释性

阅读:1

Abstract

The increasing challenge of air pollution in cities requires smart methods to make proper predictions and manage the problem. Although machine learning and deep learning models have contributed greatly to weather and pollution forecasting, the main issue is the real-time flexibility, and scalability in the varying atmospheric conditions. This paper introduces a weighted Voting ensemble model that combines Gradient Boosting (×4), CatBoost (×3), XGBoost (×2) and LightGBM (×1) to improve the accuracy of Air Quality Index (AQI) forecasting. The full preprocessing (complete-case deletion, which retains extremes) and optimization of hyperparameters (GridSearchCV/Optuna, 5-fold CV) were used to enhance the robustness and generalizability of the model. The Taiwan Air Quality Dataset (2016-2024, [Formula: see text] hourly records from 74 stations), 6 major pollutants (PM2.5, PM10, [Formula: see text], [Formula: see text], [Formula: see text], CO), meteorological parameters (wind speed/direction), and 8-h averages) is used to model the data (spatial/temporal IDs are excluded, to allow deployment to a single station). Experimental validation of 60/16/24 splits (random + temporal validation) shows that the ensemble has validation MSE 0.6553 ([Formula: see text] 0.9969), which beats 15 baselines including the deep learning (LSTM MSE 45.4), but has temporal robustness (Δ [Formula: see text]= - 0.0037). Moreover, SHAP is implemented to offer explainability, as it gives more insights into the contribution of features in predicting AQI. The results indicate the promise of interpretable ensemble learning systems to underpin sustainable urban living, reinforce community health programs, and allow interventions in managing air quality in time.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。