Explainable AI for Interpreting Spatiotemporal Groundwater Predictions
Stephanie Clark, Guobin Fu, Sreekanth Janardhanan
Abstract
Abstract As machine learning models become more widely relied on for groundwater predictions, the ability to interpret and explain these predictions is increasingly important. Explainable AI (XAI) tools are addressing this challenge by enhancing model transparency. Importantly, XAI also offers an early indication of its potential in broadening the role of machine learning in groundwater research — shifting it from a predictive tool to one that deepens understanding of system dynamics. This study explores the capacity of XAI to provide comprehensive insights into groundwater system behavior over large geographic scales. Spatiotemporal variations in groundwater levels and trends across Australia's Murray‐Darling Basin (MDB) are investigated. Predominant drivers of groundwater changes are identified, revealing differences across subregions and extended timeframes, including during periods of drought. Insights are revealed on a geographic scale that would be difficult to obtain using physics‐based or conceptual models, though the approach is equally applicable to surrogates and emulators of these models. This framework advances the interpretability of spatiotemporal environmental predictions through the incorporation of machine learning with explainability and visualisations—demonstrating the potential for machine learning to add value in hydrological research beyond the production of accurate predictions. Although the application of explainability in hydrological machine learning models is still relatively new, it is poised to become a standard component of future analyses. Through the considered adaptation of XAI methods to hydrological settings, researchers will enhance the acceptance and applicability of machine learning models for sustainable water resource management.