Understanding the evolutionary processes and causes of groundwater drought using an interpretable machine learning model
Zhiyuan Gan, Xianjun Xie, Chunli Su, Weili Ge, Hongjie Pan, Liangping Yang
Abstract
Drought is a widespread natural disaster, and rapid assessment of groundwater drought has become a challenge due to the lack of direct spatiotemporal observation of groundwater. We employed machine learning models and the Shapley Additive Explanation (SHAP), a game theory-based interpretability method, to understand and predict the evolution of groundwater drought by evaluating eight models with SHAP analysis in the West Liao River Plain (WLRP), with a semi-arid climate. The research showed: (1) The XGBoost model, optimized by the Sparrow Search Algorithm (SSA), achieved the highest performance (AUC: 0.922, F1-score: 0.84). (2) SHAP analysis revealed that the Standardized Precipitation Evapotranspiration Index (SPEI) at 12- and 24-month scales (SPEI12 and SPEI24) were key predictors, with long-term meteorological drought causing delayed groundwater drought, exacerbated by over-extraction and urbanization. The local SHAP values confirmed the robust importance of long-term meteorological drought and precipitation, and identified the interaction between precipitation and meteorological factors responsible for groundwater drought. (3) Future projections under the SSP5-8.5 climate scenario indicated a significant increase in drought-affected areas, with earlier onset, broader extent, and greater severity. This work provides a machine learning framework for evaluating groundwater drought characteristics driven by multiple factors.