SREL Reprint #3708

Optimized machine learning model for predicting groundwater contamination

Hirak Mazumdar1, Michael P. Murphy1, Shilpa Bhatkande1, Hilary P. Emerson2, Daniel I. Kaplan3,
and Hardik A. Gohel1

1University Of Houston-Victoria, Department of Computer Science, Victoria, Texas, USA
2Pacific Northwest National Laboratory, Richland, WA, USA
3University of Georgia, Savannah River Ecology Lab, Aiken, SC, USA

Abstract: The use of physical models to predict groundwater contaminant movement remains technically challenging due to the complexity of the phenomena, the heterogeneity of key parameters in nature, and the presence of poorly defined interactive and feedback processes. New approaches to address these challenges are needed. In this study, we evaluate various Artificial Intelligence (AI)-based approaches to understand hexavalent chromium (Cr(VI)) plumes located on the U.S. Department of Energy's (DOE) Hanford Site in Richland, WA. The groundwater monitoring dataset used in this study included data from the 100 Area along the Columbia River and included data collected between 2010 to 2019. This study investigates the most prominent contaminant, Cr(VI), with the Extreme Gradient Boosting (XGBoost) machine learning model. The XGBoost model was compared with an optimized version using an Empirical Bayes Search Cross-Validation technique for better prediction. The optimized XGBoost model yielded an R2 value of 0.99 on the training set and 0.85 on the testing set, whereas XGBoost without optimization yielded a value of 0.83 on the training set and 0.73 on the testing set. This paper provides an overview of a computational method for groundwater contamination modeling that shows promise for improving current remediation efforts.

Keywords: Groundwater, Contamination, Prediction, Machine Learning, Optimization

SREL Reprint #3708

Mazumdar, H., M. P. Murphy, S. Bhatkande, H. P. Emerson, D. I. Kaplan, and H. A. Gohel. 2022. Optimized machine learning model for predicting groundwater contamination. 2022 IEEE MetroCon.

This information was provided by the University of Georgia's Savannah River Ecology Laboratory (srel.uga.edu).