Physical Sciences and Mathematics Commons^™

Comparative Evaluation Of Machine Learning Models For Groundwater Quality Assessment, Shine Bedi, Ashok Samal, Chittaranjan Ray, Daniel D. Snow

School of Computing: Faculty Publications

Contamination from pesticides and nitrate in groundwater is a significant threat to water quality in general and agriculturally intensive regions in particular. Three widely used machine learning models, namely, artificial neural networks (ANN), support vector machines (SVM), and extreme gradient boosting (XGB), were evaluated for their efficacy in predicting contamination levels using sparse data with non-linear relationships. The predictive ability of the models was assessed using a dataset consisting of 303 wells across 12 Midwestern states in the USA. Multiple hydrogeologic, water quality, and land use features were chosen as the independent variables, and classes were based on measured concentration …