Improving groundwater quality predictions in semi-arid regions using ensemble learning models

Abstract

Groundwater resources constitute one of the primary sources of freshwater in semi-arid and arid climates. Monitoring the groundwater quality is an essential component of environmental management. In this study, a comprehensive comparison was conducted to analyze the performance of nine ensembles and regular machine learning (ML) methods in predicting two water quality parameters including total dissolved solids (TDS) and pH, in an area with semi-arid climate conditions. The study area under consideration is an aquifer located in the Sirjan plain, Kerman, Iran. The developed models include standard multilayer perceptron neural network (MLPNN), classification and regression trees (CART), Chi-square automatic interaction detection (CHAID), and their ensemble versions in bagging (BG) and boosting (BT) ensemble structures. The analysis revealed that standard MLs yield comparable results in predicting TDS. The MLPNN, exhibiting a standard root mean square error (SRMSE) of 0.085, demonstrated superior accuracy in predicting TDS when contrasted with CART and CHAID models. Predicting pH poses a greater challenge for the models. Ensemble techniques significantly enhanced the accuracy of regular models. On average, the bagging and boosting techniques resulted in a 22.68% improvement in the accuracy of regular models, which represents a statistically significant enhancement. The boosting method, with an average SRMSE of 0.0602, is more accurate than bagging. Based on the results, the CHAID-BT with SRMSE of 0.0790 and CHAID-BG with SRMSE of 0.0330 are ranked the most accurate models for predicting TDS and pH, respectively. The performance of ensemble techniques in predicting TDS is more remarkable. In practical implementation, ensemble techniques can be considered an alternative method with high accuracy for sustainable water resources management in semi-arid regions, helping to address water shortages, climate change, water pollution, etc.