Abstract:In the past few decades, global industrial development and population growth have led to a scarcity of water resources, making sustainable management of groundwater a global challenge. The Water Quality Index (WQI) serves as a comprehensive method for assessing water quality and can provide valuable recommendations at the water quality level, optimizing policies for groundwater management. However, the subjectivity and uncertainty of the traditional WQI have negative impacts on evaluation outcomes, particularly in determining indicator weights and selecting aggregation functions. The proposed water quality index for groundwater based on the random forest (RFWQI) model in this study addresses these issues. It selects water quality indicators based on the actual pollution situation in the study area, employs an advanced random forest model to rank water quality indicators, determines indicator weights using the rank centroid method, scores the indicators using a sub-index function designed for groundwater development, and compares the results of two commonly used aggregation functions to identify the optimal one. Based on the aggregated scores, the water quality at 137 monitoring sites is classified into five levels: “Excellent”, “Good”, “Medium”, “Poor”, or “Unacceptable”. Among the 11 water quality indicators (sodium, sulfate, chloride, bicarbonate, total dissolved solids, fluoride, boron, nitrate, pH, CODMn, and hardness), chloride was given the highest weight (0.236), followed by total dissolved solids (0.156), and sodium was given the lowest weight (0.008). The random forest model exhibits a good prediction capability before hyperparameter tuning (86% accuracy, RMSE of 0.378), and after grid search and five-fold cross-validation, the optimal hyperparameter combination is determined, further improving the performance of the random forest model (94% accuracy, F1-Score of 0.967, AUC of 0.91, RMSE of 0.232). For the newly developed groundwater sub-index function, interpolation is used to score each indicator, and after comparing two aggregation functions, the NSF aggregation function is selected as the most suitable for groundwater assessment. Overall, most of the groundwater in the study area was of poor quality (52.5% of low quality) and not suitable for drinking.

Predictive Modeling of Groundwater Nitrate Pollution and Evaluating Its Main Impact Factors Using Random Forest.

Mapping specific groundwater nitrate concentrations from spatial data using machine learning: A case study of chongqing, China

Assessing impacts of land use/land cover patterns to shallow groundwater nitrate pollution in an agricultural-dominant area in northwest China using random forest

Identification of Critical Effect Factors for Prediction of Spatial and Intra-Annual Variability of Shallow Groundwater Nitrate in Agricultural Areas

Risk assessment of nitrate groundwater contamination using GIS-based machine learning methods: A case study in the northern Anhui plain, China

Predictive Modeling of Nitrogen and Phosphorus Concentrations in Rivers Using a Machine Learning Framework: A Case Study in an Urban-Rural Transitional Area in Wenzhou China.

Assessment of aquifer specific vulnerability to total nitrate contamination using ensemble learning and geochemical evidence

Key factors affecting groundwater nitrate levels in the Yinchuan Region, Northwest China: Research using the eXtreme Gradient Boosting (XGBoost) model with the SHapley Additive exPlanations (SHAP) method

Application of machine learning and deep neural networks for spatial prediction of groundwater nitrate concentration to improve land use management practices

Modelling the Vegetation Response to Climate Changes in the Yarlung Zangbo River Basin Using Random Forest

Spatial-temporal variation analysis and prediction of water quality in the Yellow River basin, China

Predicting monthly high-resolution PM2.5 concentrations with random forest model in the North China Plain

Nitrogen and Phosphorus Retention Risk Assessment in a Drinking Water Source Area under Anthropogenic Activities

Enhanced groundwater vulnerability assessment to nitrate contamination in Chongqing, Southwest China: Integrating novel explainable machine learning algorithms with DRASTIC-LU

Assessing the land use type and environment factors affecting groundwater nitrogen in an arid oasis in northwestern China

Health risk assessment of groundwater nitrogen pollution in Yinchuan plain

Critical role of climate factors for groundwater potential mapping in arid regions: Insights from random forest, XGBoost, and LightGBM algorithms

Electrical conductivity and dissolved oxygen as predictors of nitrate concentrations in shallow groundwater in Erhai Lake region

Predictive Modeling of Nitrogen Transport and Transformation in Unsaturated Zone

Multi-machine learning methods to predict spatial variation characteristics of total nitrogen at watershed scale: Evidences from the largest watershed (Yangtze River Watershed), Asian

Groundwater Quality Assessment Based on the Random Forest Water Quality Index—Taking Karamay City As an Example