Advancing water quality assessment and prediction using machine learning models, coupled with explainable artificial intelligence (XAI) techniques like shapley additive explanations (SHAP) for interpreting the black-box nature
International Water Management Institute · The University of Tokyo · +7 more institutions
Abstract
Water quality assessment and prediction play crucial roles in ensuring the sustainability and safety of freshwater resources. This study aims to enhance water quality assessment and prediction by integrating advanced machine learning models with XAI techniques. Traditional methods, such as the water quality index, often require extensive data collection and laboratory analysis, making them resource-intensive. The weighted arithmetic water quality index is employed alongside machine learning models, specifically RF, LightGBM, and XGBoost, to predict water quality. The models' performance was evaluated using metrics such as MAE, RMSE, R2, and R. The results demonstrated high predictive accuracy, with XGBoost…
Citation impact
- FWCI
- 29.54
- Percentile
- 100%
- References
- 58
Authors
8Topics & keywords
- Water quality
- Index (typography)
- Quality (philosophy)
- Predictive modelling
- Mean squared error
- Machine learning
- Computer science
- Artificial intelligence