A Comparison of Ensemble and Deep Learning Algorithms to Model Groundwater Levels in a Data-Scarce Aquifer of Southern Africa
Open Access
Abstract
Machine learning and deep learning have demonstrated usefulness in modelling various groundwater phenomena. However, these techniques require large amounts of data to develop reliable models. In the Southern African Development Community, groundwater datasets are generally poorly developed. Hence, the question arises as to whether machine learning can be a reliable tool to support groundwater management in the data-scarce environments of Southern Africa. This study tests two machine learning algorithms, a gradient-boosted decision tree (GBDT) and a long short-term memory neural network (LSTM-NN), to model groundwater level (GWL) changes in the Shire Valley Alluvial Aquifer. Using data from two boreholes, Ngabu (sample size = 96) and Nsanje (sample size = 45), we model two predictive scenarios: (I) predicting the change in the current month’s groundwater level, and (II) predicting the change in the following month’s groundwater level. For the Ngabu borehole, GBDT achieved R2 scores of 0.19 and 0.14, while LSTM achieved R2 scores of 0.30 and 0.30, in experiments I and II, respectively. For the Nsanje borehole, GBDT achieved R2 of −0.04 and −0.21, while LSTM achieved R2 scores of 0.03 and −0.15, in experiments I and II, respectively. The results illustrate that LSTM performs better than the GBDT model, especially regarding slightly greater time series and extreme GWL changes. However, closer inspection reveals that where datasets are relatively small (e.g., Nsanje), the GBDT model may be more efficient, considering the cost required to tune, train, and test the LSTM model. Assessing the full spectrum of results, we concluded that these small sample sizes might not be sufficient to develop generalised and reliable machine learning models.Keywords
Funding Information
- Big Data and Transboundary Water Collaboration
- IBM Research
This publication has 51 references indexed in Scilit:
- Astronomy in the Big Data EraData Science Journal, 2015
- Big data analytics in healthcare: promise and potentialHealth Information Science and Systems, 2014
- Transboundary aquifer mapping and management in Africa: a harmonised approachHydrogeology Journal, 2013
- What are decision trees?Nature Biotechnology, 2008
- A working guide to boosted regression treesJournal of Animal Ecology, 2008
- Estimating groundwater storage changes in the Mississippi River basin (USA) using GRACEHydrogeology Journal, 2006
- Groundwater Level Forecasting in a Shallow Aquifer Using Artificial Neural Network ApproachWater Resources Management, 2006
- The gravity recovery and climate experiment: Mission overview and early resultsGeophysical Research Letters, 2004
- Methods for inferring regional surface‐mass anomalies from Gravity Recovery and Climate Experiment (GRACE) measurements of time‐variable gravityJournal of Geophysical Research, 2002
- Time variability of the Earth's gravity field: Hydrological and oceanic effects and their possible detection using GRACEJournal of Geophysical Research, 1998