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.

Assessment of sodium adsorption ratio (SAR) in groundwater: Integrating experimental data with cutting-edge swarm intelligence approaches

Groundwater Quality: The Application of Artificial Intelligence

Study of water resources parameters using artificial intelligence techniques and learning algorithms: a survey

Application of novel hybrid artificial intelligence algorithms to groundwater simulation

Groundwater quality parameters prediction based on data-driven models

Artificial intelligence for surface water quality monitoring and assessment: a systematic literature analysis

Groundwater quality modeling using a novel hybrid data-intelligence model based on gray wolf optimization algorithm and multi-layer perceptron artificial neural network: a case study in Asadabad Plain, Hamedan, Iran

Enhancing the accuracy of metaheuristic neural networks in predicting underground water levels using meteorological data and remote sensing: A case study of Ardabil Plain, Iran

Estimation of irrigation water quality index in a semi-arid environment using data-driven approach

Application of Water Quality Indices, Machine Learning Approaches, and GIS to Identify Groundwater Quality for Irrigation Purposes: A Case Study of Sahara Aquifer, Doucen Plain, Algeria

Spatiotemporal assessment of groundwater quality and quantity using geostatistical and ensemble artificial intelligence tools

Simple Predictive Tool Estimates Sodium Adsorption Ratio for Evaluation of Potential Infiltration Problems using Reclaimed Wastewater

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

Mapping of groundwater salinization and modelling using meta-heuristic algorithms for the coastal aquifer of eastern Saudi Arabia

Modeling and prediction of water quality parameters using a hybrid particle swarm optimization–neural fuzzy approach

A hybrid artificial neural network: An optimization-based framework for smart groundwater governance

Groundwater Quality Analysis and Drinkability Prediction using Artificial Intelligence

Modelling qualitative and quantitative parameters of groundwater using a new wavelet conjunction heuristic method: wavelet extreme learning machine versus wavelet neural networks

Artificial Intelligence for Water Consumption Assessment: State of the Art Review

Prediction of irrigation water quality indices based on machine learning and regression models

Drinking Water Resources Suitability Assessment Based on Pollution Index of Groundwater Using Improved Explainable Artificial Intelligence