Parallelized hybrid ensemble machine learning framework for scalable and accurate rainfall prediction

Read full text
Authors: S Ramakrishnan, Christopher Columbus Chinnappan
Cited by: 1
References: 13

Abstract

Accurate rainfall forecasting is essential for agriculture, hydrological planning, and disaster preparedness, while balancing prediction quality with computational constraints. This study establishes a stacking-based ensemble framework for binary rainfall prediction using the Australian RainTomorrow dataset (over 140,000 daily observations), showing learned meta-feature combination outperforms probability averaging for imbalanced weather prediction. This study compares voting architectures against two-layer stacking ensembles where an XGBoost meta-learner optimizes weights from leakage-safe out-of-fold probabilities. Two ensemble families are proposed: RXL (Random Forest + XGBoost + LightGBM) and TriBoost (CatBoost + XGBoost + HistGradientBoosting), evaluated as voting and stacking configurations with tuned single learners. The methodology ensures rigor through unified preprocessing, leakage control, and evaluation under three protocols: stratified holdout (Protocol A), 5-fold cross-validation (Protocol B), and temporal split (Protocol C). Class imbalance is addressed via class weighting, fold-internal SMOTE, and threshold tuning, while temporal features mitigate distribution drift. Results confirm advantages for stacking architectures. TriBoost Stack achieves state-of-the-art performance under Protocol A (Accuracy 0.864, Macro F1 0.776, PR AUC 0.751, ROC AUC 0.893) with strong minority-event detection (Recall Yes 0.565, Precision Yes 0.752), exceeding voting ensembles and benchmarks. RXL Stack follows (Accuracy 0.861, Macro F1 0.772), while voting variants underperform on minority-sensitive metrics. Cross-validation confirms stability (TriBoost Stack: Macro F1 0.772 ± 0.004), and temporal evaluation validates deployment readiness (Macro F1 0.754). Analysis shows practical viability: parallel CPU execution and GPU acceleration deliver training speedups without sacrificing quality. The framework provides a validated solution for operational weather forecasting systems, balancing predictive performance and computational constraints.

Fields of Study & Subjects

Hydrological Forecasting Using AI Precipitation Measurement and Analysis Hydrology and Drought Analysis

Publication details

Journal / source
Machine Learning with Applications
Publisher
Elsevier BV
Volume
24
Pages
100863–100863
Publication date
10 February 2026
Language
EN