Neural Hierarchical Interpolation for Standardized Precipitation Index Forecasting

Abstract

In the context of climate change, studying changes in rainfall patterns is a crucial area of research, remarkably so in arid and semi-arid regions due to the susceptibility of human activities to extreme events such as droughts. Employing predictive models to calculate drought indices can be a useful method for the effective characterization of drought conditions. This study applies two type of machine learning methods—long short-term memory (LSTM) and Neural Hierarchical Interpolation for Time Series Forecasting (N-HiTS)—to develop and deploy artificial neural network models with the aim of predicting the regional standardized precipitation index (SPI) in four regions of Zacatecas, Mexico. The predictor variables were a set of climatological time series data spanning from 1964 to 2020. The results suggest that the N-HiTS model outperforms the LSTM model in the prediction and forecasting of SPI time series for all regions in terms of performance metrics: the Mean Squared Error, Mean Absolute Error, Coefficient of Determination and ξ correlation coefficient range from 0.0455 to 0.5472, from 0.1696 to 0.6661, from 0.9162 to 0.9684 and from 0.9222 to 0.9368, respectively, for the regions under study. Consequently, the outcomes revealed the successful performance of the N-HiTS models in accurately predicting the SPI across the four examined regions.