The machine learning and geostatistical approach for assessment of arsenic contamination levels using physicochemical properties of water

• Published in: Water science and technology Vol. 88; no. 3; pp. 595 - 614
• Main Authors: Chattopadhyay, Arghya, Singh, Anand Prakash, Kumar, Siddharth, Pati, Jayadeep, Rakshit, Amitava
• Format: Journal Article
• Language: English
• Published: England IWA Publishing 01.08.2023
• Subjects: Accuracy; Algorithms; Analytical chemistry; Anthropogenic factors; Aquifers; Arsenic; Carcinogens; Contamination; Data collection; Decision trees; Geostatistics; Groundwater; Groundwater pollution; Learning algorithms; Machine learning; Mathematical models; multilayer perceptron; Multilayer perceptrons; naïve bayes; Physicochemical processes; Physicochemical properties; random forest algorithm; Regression analysis; Sensitivity; Spatial analysis; spatial variability; Specificity; Statistical analysis; Water analysis; Water pollution; Water sampling; Wind; Bangladesh; India
• ISSN: 0273-1223; 1996-9732
• DOI: 10.2166/wst.2023.231

Arsenic contamination in groundwater due to natural or anthropogenic sources is responsible for carcinogenic and non-carcinogenic risks to humans and the ecosystem. The physicochemical properties of groundwater in the study area were determined in the laboratory using the samples collected across the Varanasi region of Uttar Pradesh, India. This paper analyses the physicochemical properties of water using machine learning, descriptive statistics, geostatistical and spatial analysis. Pearson correlation was used for feature selection and highly correlated features were selected for model creation. Hydrochemical facies of the study area were analyzed and the hyperparameters of machine learning models, i.e., multilayer perceptron, random forest (RF), naïve Bayes, and decision tree were optimized before training and testing the groundwater samples as high (1) or low (0) arsenic contamination levels based on the WHO 10 μg/L guideline value. The overall performance of the models was compared based on accuracy, sensitivity, and specificity value. Among all models, the RF algorithm outclasses other classifiers, as it has a high accuracy of 92.30%, a sensitivity of 100%, and a specificity of 75%. The accuracy result was compared to prior research, and the machine learning model may be used to continually monitor the amount of arsenic pollution in groundwater.