Assessment of Algorithm Performance on Predicting Total Dissolved Solids Using Artificial Neural Network and Multiple Linear Regression for the Groundwater Data

Abstract

Estimating groundwater quality parameters through conventional methods is time-consuming through laboratory measurements for megacities. There is a need to develop models that can help decision-makers make policies for sustainable groundwater reserves. The current study compared the efficiency of multivariate linear regressions (MLR) and artificial neural network (ANN) models in the prediction of groundwater parameters for total dissolved solids (TDS) for three sub-divisions in Lahore, Pakistan. The data for this study were collected every quarter of a year for six years. ANN was applied to investigate the feasibility of feedforward, backpropagation neural networks with three training functions T-BR (Bayesian regularization backpropagation), T-LM (Levenberg–Marquardt backpropagation), and T-SCG (scaled conjugate backpropagation). Two activation functions were used to analyze the performance of algorithmic training functions, i.e., Logsig and Tanh. Input parameters of pH, electrical conductivity (EC), calcium (Ca2+), magnesium (Mg2+), chloride (Cl−), and sulfate (SO42−) was used to predict TDS as an output parameter. The computed values of TDS by ANN and MLR were in close agreement with their respective measured values. Comparative analysis of ANN and MLR showed that TDS root means square error (RMSE) for city sub-division and Pearson’s coefficient of correlation (r) for ANN and MLR were 2.9% and 0.981 and 4.5% and 0.978, respectively. Similarly, for the Farrukhabad sub-division, RMSE and r for ANN were 4.9% and 0.952, while RMSE and r for MLR were 5.5% and 0.941, respectively. For the Shahadra sub-division, RMSE was 10.8%, r was 0.869 for ANN, RMSE was 11.3%, and r was 0.860 for MLR. The results exhibited that the ANN model showed less error in results than MLR. Therefore, ANN can be employed successfully as a groundwater quality prediction tool for TDS assessment.

Novel Ocean Wave Height and Energy Spectrum Forecasting Approaches: An Application of Semi-Analytical and Machine Learning Models

“…The results of Newton's method will be extremely close to the minimum error rate. This algorithm is highly effective and stable [69,70]. Levenberg-Marquardt is designed to minimize sum-of-square error functions.…”

Providing predictive models for quality parameters of groundwater resources in arid areas of central Iran: A case study of kashan plain

“…regarding this issue, various studies have been done to provide predictive models of groundwater quality parameters [ 3 , [10] , [11] , [12] , [13] , [14] , [15] , [16] , [17] ]. Application of MLR models to predict the fluoride content of groundwater in Maharashtra, India showed good performance of the regression analysis.…”

Intelligent soft computational models integrated for the prediction of potentially toxic elements and groundwater quality indicators: a case study

“…Despite using sufficient input variables, Egbueri and Agbasi ( 2022a ) predicted two water quality indices but did not predict PTEs in water. Similarly, Farooq et al ( 2022 ) employed MLR and ANN for predicting water quality parameters; however, PTEs were not predicted. In the study region, Egbueri ( 2021 ) predicted PTEs in water using only ANN and also used a few input variables for the prediction.…”