Abstract:In Saudi Arabia, water pollution and drinking water scarcity pose a major challenge and jeopardise the achievement of sustainable development goals. The urgent need for rapid and accurate monitoring and assessment of water quality requires sophisticated, data-driven solutions for better decision-making in water management. This study aims to develop optimised data-driven models for comprehensive water quality assessment to enable informed decisions that are critical for sustainable water resources management. We used an entropy-weighted arithmetic technique to calculate the Water Quality Index (WQI), which integrates the World Health Organization (WHO) standards for various water quality parameters. Our methodology incorporated advanced machine learning (ML) models, including decision trees, random forests (RF) and correlation analyses to select features essential for identifying critical water quality parameters. We developed and optimised data-driven models such as gradient boosting machines (GBM), deep neural networks (DNN) and RF within the H2O API framework to ensure efficient data processing and handling. Interpretation of these models was achieved through a three-pronged explainable artificial intelligence (XAI) approach: model diagnosis with residual analysis, model parts with permutation-based feature importance and model profiling with partial dependence plots (PDP), accumulated local effects (ALE) plots and individual conditional expectation (ICE) plots. The quantitative results revealed insightful findings: fluoride and residual chlorine had the highest and lowest entropy weights, respectively, indicating their differential effects on water quality. Over 35% of the water samples were categorised as 'unsuitable' for consumption, highlighting the urgency of taking action to improve water quality. Amongst the optimised models, the Random Forest (model 79) and the Deep Neural Network (model 81) proved to be the most effective and showed robust predictive abilities with R 2 values of 0.96 and 0.97 respectively for testing dataset. Model profiling as XAI highlighted the significant influence of key parameters such as nitrate, total hardness and pH on WQI predictions. These findings enable targeted water quality improvement measures that are in line with sustainable water management goals. Therefore, our study demonstrates the potential of advanced, data-driven methods to revolutionise water quality assessment in Saudi Arabia. By providing a more nuanced understanding of water quality dynamics and enabling effective decision-making, these models contribute significantly to the sustainable management of valuable water resources.

Explainable artificial intelligence for the interpretation of ensemble learning performance in algal bloom estimation

Interpretation of ensemble learning to predict water quality using explainable artificial intelligence

Comparative Analysis of Machine Learning Methods for Prediction of Chlorophyll-a in a River with Different Hydrology Characteristics: A Case Study in Fuchun River, China

Spatial-temporal characteristics and the importance of environmental factors in relation to algal blooms in coastal seas

A Soft Sensor for Simulating Algal Cell Density Based on Dynamic Response to Environmental Changes in a Eutrophic Shallow Lake

Identification of driving factors for chlorophyll‐a in multi‐stable shallow lakes of China employing machine learning methods

Probabilistic Prediction of Algal Blooms from Basic Water Quality Parameters by Bayesian Scale-Mixture of Skew-Normal Model

Early warning of algal blooms based on the optimization support vector machine regression in a typical tributary bay of the Three Gorges Reservoir, China

Predicting and analyzing the algal population dynamics of a grass-type lake with explainable machine learning

Algal community structure prediction by machine learning

Predicting the Short-Time-scale Variability of Chlorophyll a in the Elbe River Using a Lagrangian-based Multi-Criterion Analog Model

Algal bloom forecasting with time-frequency analysis: A hybrid deep learning approach.

Response Surfaces Method and Artificial Intelligence Approaches for Modeling the Effects of Environmental Factors on Chlorophyll a in Isochrysis galbana

Prediction and sensitivity analysis of chlorophyll a based on a support vector machine regression algorithm

Interpreting optimised data-driven solution with explainable artificial intelligence (XAI) for water quality assessment for better decision-making in pollution management

A novel hybrid model based on two-stage data processing and machine learning for forecasting chlorophyll-a concentration in reservoirs

Employing hybrid deep learning for near-real-time forecasts of sensor-based algal parameters in a Microcystis bloom-dominated lake

Recent advances in algal bloom detection and prediction technology using machine learning

An ensemble machine learning model for water quality estimation in coastal area based on remote sensing imagery

Assessing the algal population dynamics using multiple machine learning approaches: Application to Macao reservoirs

A Novel Single-Parameter Approach for Forecasting Algal Blooms.