Long-term drought prediction using deep neural networks based on geospatial weather data

Alexander Marusov,Vsevolod Grabar,Yury Maximov,Nazar Sotiriadi,Alexander Bulkin,Alexey Zaytsev
2024-07-01
Abstract:The problem of high-quality drought forecasting up to a year in advance is critical for agriculture planning and insurance. Yet, it is still unsolved with reasonable accuracy due to data complexity and aridity stochasticity. We tackle drought data by introducing an end-to-end approach that adopts a spatio-temporal neural network model with accessible open monthly climate data as the input. Our systematic research employs diverse proposed models and five distinct environmental regions as a testbed to evaluate the efficacy of the Palmer Drought Severity Index (PDSI) prediction. Key aggregated findings are the exceptional performance of a Transformer model, EarthFormer, in making accurate short-term (up to six months) forecasts. At the same time, the Convolutional LSTM excels in longer-term forecasting. Both models achieved high ROC AUC scores: 0.948 for one month ahead and 0.617 for twelve months ahead forecasts, becoming closer to perfect ROC-AUC by $54\%$ and $16\%$, respectively, c.t. classic approaches.
Machine Learning
What problem does this paper attempt to address?
### Problems the Paper Aims to Solve This paper aims to address the problem of year-long drought prediction. Specifically, the authors seek to improve the quality of drought predictions by introducing an end-to-end approach to handle drought data, employing spatiotemporal neural network models, and utilizing publicly available monthly climate data as input. ### Main Contributions 1. **Comprehensive Study of Diverse Spatiotemporal Models**: The authors employed advanced deep learning methods from different fields to benchmark the most prominent models, including the Transformer-based models EarthFormer and FourCastNet. Additionally, they constructed classical logistic regression and gradient boosting methods to account for spatial and temporal dependencies. 2. **Development of a Testing Platform Covering Five Different Global Regions**: This allows the research to objectively evaluate the models and provides systematic answers on how to accurately predict droughts within a time range of 1 to 12 months, and whether deep learning is necessary for accurate predictions. 3. **Identification that EarthFormer-based Neural Networks are Suitable for Short to Medium-term Predictions (up to 6 months), while ConvLSTM is Suitable for Long-term Predictions**: This indicates that deep learning methods perform excellently in prediction time frames of four months and beyond, which is crucial for decision-making. 4. **Continuous Improvement of Models Across Different Time Ranges from 1 to 12 Months**: This contrasts with previous studies that typically focused only on very short-term (up to 1 month) or long-term predictions spanning decades. Through these contributions, the paper demonstrates the effectiveness and superiority of advanced deep learning methods in drought prediction.