A Hybrid Framework Based on Lstm for Predicting Karst Spring Discharge Using Historical Data

Wenrui Zhang,Limin Duan,Tingxi Liu,Zheming Shi,Xiaoqing Shi,Yong Chang,Shen Qu,Guoqiang Wang
DOI: https://doi.org/10.1016/j.jhydrol.2024.130946
IF: 6.4
2024-01-01
Journal of Hydrology
Abstract:Accurately predicting karst spring discharge is crucial for effective regional water resource management. However, the uneven distribution (or absence) of meteorological stations within karst spring catchments and the temporal discontinuity of meteorological data make the prediction of spring discharge difficult. To overcome a model’s dependence on meteorological data, this paper proposes a novel hybrid framework that uses historical spring discharge as the input for the LSTM model. The framework consists of four components—hyperparameter optimization, variable screening, decomposition-combination, and error suppression—to improve prediction accuracy. Specifically, particle swarm optimization, genetic algorithms, and Bayesian optimization (BSO) were used to optimize the hyperparameters of LSTM. Partial correlation, grey correlation analysis, and partial mutual information (PMI) were adopted to analyze the correlations between historical and current discharge to determine the optimal input variables of the model. Seasonal and trend decomposition using loess, ensemble empirical mode decomposition, and variational mode decomposition (VMD) were used to extract multi-scale change features in the spring discharge sequence, which reduced the learning complexity of LSTM. Finally, the Autoregressive Moving Average (ARMA) model was used to correct the output error, ensuring prediction accuracy. In total, 57 models were established to evaluate the performance of the hybrid framework in three karst spring catchments with different hydrogeological conditions and strongly nonlinear discharge. The results show that the hybrid framework achieves excellent prediction performance for all three karst springs. In comparison to the five benchmark models that take meteorological data as the input, the hybrid framework outperforms the best benchmark models in two of the spring catchments. This study provides a high-quality alternative for predicting spring discharge in regions without meteorological data.
What problem does this paper attempt to address?