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
- Sabina Khadka (ORCID: https://orcid.org/0000-0003-4157-197X)
- Shekhar Mahat (ORCID: https://orcid.org/0000-0002-5743-1640)
- Sudip Mahat (ORCID: https://orcid.org/0009-0008-7177-1044)
- Kyaron Kadel (ORCID: https://orcid.org/0009-0000-1712-4657)
Institutions
- 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)
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