A hierarchical ensemble approach for multi-country PM10 forecasting using LightGBM and residual neural network

Accurate day-ahead forecasting of particulate matter (PM10) concentrations is critical for public health interventions, regulatory compliance, and urban air quality management. However, existing approaches suffer from temporal leakage, single-city limitations, inadequate hierarchical modeling of geographic dependencies, and reliance on single-model architectures that fail to capture complex nonlinear pollution dynamics. This study presents a novel three-stage leakage-free stacked ensemble framework for city-level PM10 prediction across 25 countries and 380 cities using the World Air Quality Index (WAQI) dataset comprising 1,798,600 records. The framework integrates Light Gradient Boosting Machine (LightGBM) as the base learner, a residual neural network to model higher-order nonlinearities, and a Ridge regression meta-learner trained exclusively on out-of-fold predictions. Rigorous leakage prevention is achieved through chronological per-city train-test splits (70:30), leakage-safe target encoding of hierarchical city and country features fitted only on training data, and expanding-window rolling-origin cross-validation. The proposed model achieves exceptional performance on the held-out test set with R2 = 0.9983, RMSE = 1.01 µg/m3, MAE = 1.01 µg/m3, and NSE = 0.9983, representing 96.7% RMSE improvement over the persistence baseline and 76% improvement over XGBoost. Cross-validation results demonstrate consistent R2 values exceeding 0.99 across large sample countries and above 0.98 for smaller datasets. SHAP-based feature importance analysis confirms that lag-1 PM10, rolling means, and hierarchically encoded geographic features are the primary predictors. Temporal shuffle validation confirms the absence of information leakage. The framework significantly outperforms ARIMA, SARIMA, Random Forest, MLP, and traditional baselines while maintaining computational efficiency. This research establishes a reproducible, operationally viable, and theoretically rigorous benchmark for multi-country air quality forecasting with direct applications in public health policy and environmental monitoring systems.

Authors

Institutions

Publication Details

Journal
Discover Atmosphere
Published
2026-01-05
DOI
https://doi.org/10.1007/s44292-025-00072-4
Citations
3
Primary Topic
Air Quality Monitoring and Forecasting
Type
article
Field-Weighted Citation Impact
65.49
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

A hierarchical ensemble approach for multi-country PM10 forecasting using LightGBM and residual neural network

Haider Rajput, Saddam Umer, Syed Azeem Inam
3 citations
Discover Atmosphere
Air Quality Monitoring and Forecasting
65.49
article

A hierarchical ensemble approach for multi-country PM10 forecasting using LightGBM and residual neural network

Haider Rajput, Saddam Umer, Syed Azeem Inam
article en
3 citations

Abstract

Accurate day-ahead forecasting of particulate matter (PM10) concentrations is critical for public health interventions, regulatory compliance, and urban air quality management. However, existing approaches suffer from temporal leakage, single-city limitations, inadequate hierarchical modeling of geographic dependencies, and reliance on single-model architectures that fail to capture complex nonlinear pollution dynamics. This study presents a novel three-stage leakage-free stacked ensemble framework for city-level PM10 prediction across 25 countries and 380 cities using the World Air Quality Index (WAQI) dataset comprising 1,798,600 records. The framework integrates Light Gradient Boosting Machine (LightGBM) as the base learner, a residual neural network to model higher-order nonlinearities, and a Ridge regression meta-learner trained exclusively on out-of-fold predictions. Rigorous leakage prevention is achieved through chronological per-city train-test splits (70:30), leakage-safe target encoding of hierarchical city and country features fitted only on training data, and expanding-window rolling-origin cross-validation. The proposed model achieves exceptional performance on the held-out test set with R2 = 0.9983, RMSE = 1.01 µg/m3, MAE = 1.01 µg/m3, and NSE = 0.9983, representing 96.7% RMSE improvement over the persistence baseline and 76% improvement over XGBoost. Cross-validation results demonstrate consistent R2 values exceeding 0.99 across large sample countries and above 0.98 for smaller datasets. SHAP-based feature importance analysis confirms that lag-1 PM10, rolling means, and hierarchically encoded geographic features are the primary predictors. Temporal shuffle validation confirms the absence of information leakage. The framework significantly outperforms ARIMA, SARIMA, Random Forest, MLP, and traditional baselines while maintaining computational efficiency. This research establishes a reproducible, operationally viable, and theoretically rigorous benchmark for multi-country air quality forecasting with direct applications in public health policy and environmental monitoring systems.

Discover AtmosphereVol. 4(1)
Sindh Madressatul Islam University (PK)
Sustainable cities and communities
Openalex Percentile: Top 0%
Air Quality Monitoring and Forecasting
65.49
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.