Real-time sensor-based explainable machine learning models for harmful algal bloom prediction and driver thresholds in the Lower Charles River

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.

Authors

Institutions

Publication Details

Journal
Desalination and Water Treatment
Published
2026-06-12
DOI
https://doi.org/10.1016/j.dwt.2026.101831
Primary Topic
Hydrological Forecasting Using AI
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Real-time sensor-based explainable machine learning models for harmful algal bloom prediction and driver thresholds in the Lower Charles River

Sabina Khadka, Shekhar Mahat, Sudip Mahat, Kyaron Kadel
Desalination and Water Treatment
Hydrological Forecasting Using AI
article

Real-time sensor-based explainable machine learning models for harmful algal bloom prediction and driver thresholds in the Lower Charles River

Sabina Khadka, Shekhar Mahat, Sudip Mahat, Kyaron Kadel
article en

Abstract

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.

Desalination and Water TreatmentVol. 327
Louisiana State University (US), Government of Nepal (NP), Kolej Poly-Tech MARA (MY), Manmohan Memorial Institute of Health Sciences (NP), Nepal Academy of Science and Technology (NP)
Climate action
Openalex Percentile: Top 10%
Hydrological Forecasting Using AI
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.