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.

An efficient water quality index forecasting and categorization using optimized Deep Capsule Crystal Edge Graph neural network

Deep Learning-Based Water Quality Index Classification Using Stacked Ensemble Variational Mode Decomposition

Coastal groundwater quality prediction using objective-weighted WQI and machine learning approach

An advanced deep learning model for predicting water quality index

Water Quality Prediction using Neural Networks

Hybrid WT-CNN-GRU-based model for the estimation of reservoir water quality variables considering spatio-temporal features

Classification and Prediction of Developed Water Quality Indexes Using Soft Computing Tools

Optimization of water quality index models using machine learning approaches

Water Quality Prediction Based on Hybrid Deep Learning Algorithm

A novel hybrid framework based on the ANFIS, discrete wavelet transform, and optimization algorithm for the estimation of water quality parameters

Water quality assessment of deep learning-improved comprehensive pollution index: a case study of Dagu River, Jiaozhou Bay, China

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

Performance Analysis of the Water Quality Index Model for Predicting Water State Using Machine Learning Techniques

Water quality prediction using machine learning models based on grid search method

Monitoring the Industrial waste polluted stream - Integrated analytics and machine learning for water quality index assessment

WaterQualityNeT: Prediction of Seasonal Water Quality of Nepal Using Hybrid Deep Learning Models

Optimisation and interpretation of machine and deep learning models for improved water quality management in Lake Loktak

W-WaveNet: A multi-site water quality prediction model incorporating adaptive graph convolution and CNN-LSTM

Water quality prediction and classification based on principal component regression and gradient boosting classifier approach

Development of AI-based hybrid soft computing models for prediction of critical river water quality indicators

HDTO-DeepAR: A novel hybrid approach to forecast surface water quality indicators