Desalination and Water Treatment · Published 2026-06-12 · DOI 10.1016/j.dwt.2026.101831
Shekhar Mahat, Kyaron Kadel, Sabina Khadka, Sudip Mahat
Cyanobacterial harmful algal blooms (CyanoHABs) pose significant risks to freshwater ecosystems, necessitating accurate and interpretable forecasting approaches. This study develops seven machine learning (ML) models, namely Extreme Gradient Boosting (XGB), Gradient Boosting Machine (GBM), Artificial Neural Network (ANN), Random Forest (RF), Ridge Regression (Ridge reg.), and Support Vector Regression with radial bias function and linear kernels (SVR-rbf, SVR-linear), to predict phycocyanin concentration in the Lower Charles River using high-frequency monitoring data. Predictors were selected using the Boruta algorithm, and datasets were chronologically split into training (80%), validation (10%), and testing (10%) sets. XGB achieved the highest predictive accuracy (average r = 0.965, RMSE = 0.197, MAE = 0.147, and MAPE = 9.373%), while SVR-linear showed the lowest. Boruta identified water temperature, specific conductivity, and chlorophyll-a as key predictors. SHAP based explainable artificial intelligence (XAI) further quantified feature contributions, identifying specific conductivity, chlorophyll-a, and turbidity as dominant drivers governing bloom dynamics and revealing nonlinear threshold behavior. The proposed framework integrates feature selection with explainable AI, demonstrating the superiority of boosting-based model while providing both high predictive accuracy and mechanistic interpretability. This approach offers advanced tools for real-time HAB prediction and enables data-driven early bloom warning system for water quality management.
Abstract from DOAJ. Public domain (CC0 1.0).
Read the article at the publisher →
Mahat, S., Kadel, K., Khadka, S., et al. (2026). Real-time sensor-based explainable machine learning models for harmful algal bloom prediction and driver thresholds in the Lower Charles River. Desalination and Water Treatment. https://doi.org/10.1016/j.dwt.2026.101831