AI-Driven Predictive Modelling for Groundwater Salinization in Israel

TL;DR

This study develops machine learning models to predict groundwater salinity in Israel, identifying key environmental and human factors influencing salinization, and providing insights for targeted mitigation strategies.

Contribution

It integrates multiple datasets and advanced AI techniques to identify and analyze the drivers of groundwater salinization at a national scale, including explainability and causality analyses.

Findings

Key drivers include precipitation, temperature, and proximity to saline bodies.

Treated wastewater significantly influences salinity in vulnerable areas.

Machine learning models effectively predict salinity and reveal important causal factors.

Abstract

Increasing salinity and contamination of groundwater is a serious issue in many parts of the world, causing degradation of water resources. The aim of this work is to form a comprehensive understanding of groundwater salinization underlying causal factors and identify important meteorological, geological and anthropogenic drivers of salinity. We have integrated different datasets of potential covariates, to create a robust framework for machine learning based predictive models including Random Forest (RF), XGBoost, Neural network, Long Short-Term Memory (LSTM), convolution neural network (CNN) and linear regression (LR), of groundwater salinity. Additionally, Recursive Feature Elimination (RFE) followed by Global sensitivity analysis (GSA) and Explainable AI (XAI) based SHapley Additive exPlanations (SHAP) were used to estimate the importance scores and find insights into the drivers of salinization. We also did causality analysis via Double machine learning using various predictive models. From these analyses, key meteorological (Precipitation, Temperature), geological (Distance from river, Distance to saline body, TWI, Shoreline distance), and anthropogenic (Area of agriculture field, Treated Wastewater) covariates are identified to be influential drivers of groundwater salinity across Israel. XAI analysis also identified Treated Wastewater (TWW) as an essential anthropogenic driver of salinity, its significance being context-dependent but critical in vulnerable hydro-climatic environment. Our approach provides deeper insight into global salinization mechanisms at country scale, reducing AI model uncertainty and highlighting the need for tailored strategies to address salinity.