Interpretable and explainable AI (XAI) model for spatial drought prediction

Accurate prediction of any type of natural hazard is a challenging task. Of all the various hazards, drought prediction is challenging as it lacks a universal definition and is getting adverse with climate change impacting drought events both spatially and temporally. The problem becomes more complex as drought occurrence is dependent on a multitude of factors ranging from hydro-meteorological to climatic variables. A paradigm shift happened in this field when it was found that the inclusion of climatic variables in the data-driven prediction model improves the accuracy. However, this understanding has been primarily using statistical metrics used to measure the model accuracy. The present work tries to explore this finding using an explainable artificial intelligence (XAI) model. The explainable deep learning model development and comparative analysis were performed using known understandings drawn from physical-based models. The work also tries to explore how the model achieves specific results at different spatio-temporal intervals, enabling us to understand the local interactions among the predictors for different drought conditions and drought periods. The drought index used in the study is Standard Precipitation Index (SPI) at 12 month scales applied for five different regions in New South Wales, Australia, with the explainable algorithm being SHapley Additive exPlanations (SHAP). The conclusions drawn from SHAP plots depict the importance of climatic variables at a monthly scale and varying ranges of annual scale. We observe that the results obtained from SHAP align with the physical model interpretations, thus suggesting the need to add climatic variables as predictors in the prediction model.