Please login first
Comparative Analysis of Machine Learning and Deep Learning Models for Groundwater Potability Classification
* 1, 2 , 3, 4 , 5
1  Fundamental and Applied Sciences Department, Universiti Teknologi PETRONAS 32610 Seri Iskandar, Perak Darul Ridzuan, Malaysia
2  Department of Statistics, Kano University of Science and Technology, Wudil 713281, Nigeria
3  Department of Civil and Environmental Engineering, Universiti Teknologi PETRONAS, Seri Iskandar 32610, Perak, Malaysia
4  Department of Civil Engineering, University of Engineering & Technology Peshawar 25000, Pakistan
5  Department of Computer and Information Sciences, Universiti Teknologi PETRONAS, Seri Iskandar 32610, Perak, Malaysia
Academic Editor: Simeone Chianese

Abstract:

Ensuring access to safe drinking water is a critical concern, particularly in regions with limited resources. This study evaluates groundwater potability using a range of machine learning models, including logistic regression, K-Nearest Neighbors (KNN), Support Vector Classifier (SVC), and Random Forest, as well as deep learning models such as Artificial Neural Networks (ANN), Convolutional Neural Networks (CNN), Feedforward Neural Networks (FNN), and Long Short-Term Memory (LSTM). We collected thirty groundwater samples from residential and industrial locations in Jaen, Kano State, Nigeria, focusing on nine crucial physicochemical parameters: electric conductivity, pH, total dissolved solids, calcium, magnesium, chloride, zinc, manganese, and copper. Machine learning models, such as logistic regression and random forest, achieved accuracy scores of 0.833. They were closely followed by deep learning models, such as ANN with an accuracy score of 0.833, and LSTM, which scored 0.666. KNN and SVC provided moderately accurate predictions, scoring 0.667, while CNN and FNN achieved lower scores of 0.333 and 0.5, respectively. This study represents a significant step toward ensuring safe drinking water for communities and preserving the sustainability of natural resources.

Keywords: groundwater; artificial intelligence; machine learning; deep learning; classification, logistic regres-sion, random forest, artificial neural network, convolutional neural network

 
 
Top