Prediction of groundwater quality using efficient machine learning technique

• Published in: Chemosphere (Oxford) Vol. 276; p. 130265
• Main Authors: Singha, Sudhakar, Pasupuleti, Srinivas, Singha, Soumya S., Singh, Rambabu, Kumar, Suresh
• Format: Journal Article
• Language: English
• Published: Elsevier Ltd 01.08.2021
• Subjects: Entropy weight-based groundwater quality index; Machine learning algorithms; Prediction models; Variable importance; Machine learning algorithms; Entropy weight-based groundwater quality index; Prediction models; Variable importance
• ISSN: 0045-6535; 1879-1298
• DOI: 10.1016/j.chemosphere.2021.130265

To ensure safe drinking water sources in the future, it is imperative to understand the quality and pollution level of existing groundwater. The prediction of water quality with high accuracy is the key to control water pollution and the improvement of water management. In this study, a deep learning (DL) based model is proposed for predicting groundwater quality and compared with three other machine learning (ML) models, namely, random forest (RF), eXtreme gradient boosting (XGBoost), and artificial neural network (ANN). A total of 226 groundwater samples are collected from an agriculturally intensive area Arang of Raipur district, Chhattisgarh, India, and various physicochemical parameters are measured to compute entropy weight-based groundwater quality index (EWQI). Prediction performances of models are determined by introducing five error metrics. Results showed that DL model is the best prediction model with the highest accuracy in terms of R2, i.e., R2 = 0996 against the RF (R2 = 0.886), XGBoost (R2 = 0.0.927), and ANN (R2 = 0.917). The uncertainty of the DL model output is cross-verified by running the proposed algorithm with newly randomized dataset for ten times, where minor deviations in the mean value of performance metrics are observed. Moreover, input variable importance computed by prediction models highlights that DL model is the most realistic and accurate approach in the prediction of groundwater quality. •Groundwater quality is assessed using EWQI method.•Machine learning (ML) algorithms are used for predicting groundwater quality.•Prediction performance of RF, XGBoost, ANN and DL models are compared.•DL based quality prediction model performs much better than other ML models.