Understanding Drought through Spatial-Temporal Learning

Xuwei Tan, Qian Zhao, Yanlan Liu, Xueru Zhang

Keywords: Climate Science, Natural Science, AI4Science

Abstract: Drought is one of the most destructive and expensive natural disasters, severely impacting natural resources and risks by depleting water resources and diminishing agricultural yields. Under climate change, accurately predicting drought is critical for mitigating drought-induced risks. However, the intricate interplay among the physical and biological drivers that regulate droughts limits the predictability and understanding of drought, particularly at a subseasonal to seasonal (S2S) time scale. While deep learning has demonstrated potential in addressing complex climate forecasting challenges, its application to drought prediction has received relatively less attention. Therefore, in this work, we integrate predictive features and three drought indices from multiple remote sensing and reanalysis datasets across the contiguous United States and propose a dataset specifically for predicting the spatiotemporal variation of drought: DroughtSet. DroughtSet also provides the machine learning community with a new real-world dataset to benchmark time-series forecasting methods. Furthermore, We propose a spatial-temporal model to predict and interpret S2S droughts in the contiguous U.S. Our model learns from the spatial and temporal information of physical and biological features to predict three types of droughts simultaneously. Multiple strategies are employed to quantify the importance of physical and biological features for drought prediction. These results also give insight for researchers to better understand the predictability and sensitivity of drought to biological and physical conditions. We aim to contribute to the climate field by proposing a new tool to predict and understand the occurrence of droughts and provide AI community with a new benchmark to study deep learning in the climate science field.

Submission Number: 30