Abstract:Deep learning models have a high application value in runoff forecasting, but their prediction mechanism is difficult to interpret and their computational cost is high when dealing with complex hydrological relationships, limiting their feasibility in hydrological process mechanism analysis. To address these concerns, the paper first introduces an attention mechanism (AM) for building a long short-term memory network (LSTM) model with AM in the hidden layer (AM-LSTM). The AM-LSTM model employs attention layers to improve information extraction from hidden layers, resulting in a more accurate representation of the relationships between runoff-related elements. Furthermore, in the hidden layers of the AM-LSTM model, interpretable spatiotemporal attention units are established, which not only improves the model's prediction accuracy but also provides interpretability to the forecasting process. Furthermore, parallelization techniques are used in the paper to address the issue of model runtime cost. Simultaneously, to address the accuracy degradation caused by parallelization, the paper employs wavelet denoising (WD) techniques and builds the WD-AM-LSTM model. This accomplishment enables the runoff forecasting model to predict runoff in real time and with high accuracy. Based on validation using ten-day runoff data from the Huanren Reservoir in the Hun River's middle reaches, the results show that, with two layers and an eight-batch size, the AM-LSTM model outperforms the LSTM model in capturing spatiotemporal runoff features. During the model testing phase, the AM-LSTM model improves the MAE, RMSE, and NSE performance metrics by 8.46%, 13%, and 3.82%, respectively. The WD-AM-LSTM model effectively mitigates the noise impact caused by data parallelization under the conditions of two layers and a batch size of 512, achieving the same level of prediction performance while reducing computational cost by 92.01%. By incorporating attention mechanisms and wavelet denoising techniques, this study obtains high-speed and accurate predictions with interpretable results. It expands the deep learning models' applicability in ten-day runoff forecasting work.

Runoff prediction based on a VMD-LSTM model considering the decomposition error

A Hybrid Forecasting Model to Simulate the Runoff of the Upper Heihe River

Monthly Runoff Forecasting Using Variational Mode Decomposition Coupled with Gray Wolf Optimizer-Based Long Short-term Memory Neural Networks

Monthly runoff prediction based on a coupled VMD-SSA-BiLSTM model

Improved monthly runoff time series prediction using the CABES-LSTM mixture model based on CEEMDAN-VMD decomposition

An Ensemble Hybrid Forecasting Model for Annual Runoff Based on Sample Entropy, Secondary Decomposition, and Long Short-Term Memory Neural Network

A Novel Runoff Prediction Model Based on Support Vector Machine and Gate Recurrent unit with Secondary Mode Decomposition

Comparative Study of Multi-Combination Models for Medium- and Long-Term Runoff Prediction in Weihe River

Daily Runoff Forecasting Using Ensemble Empirical Mode Decomposition and Long Short-Term Memory

An adaptive daily runoff forecast model using VMD-LSTM-PSO hybrid approach

A Novel Runoff Forecasting Model Based on the Decomposition-Integration-Prediction Framework

Enhanced LSTM model for daily runoff prediction in the upper huai river basin, china

Data Decomposition, Seasonal Adjustment Method and Machine Learning Combined for Runoff Prediction: A Case Study

Monthly runoff time series interval prediction based on WOA-VMD-LSTM using non-parametric kernel density estimation

Time Series of Landslide Displacement Prediction based on VMD-LSTM

Runoff prediction of lower Yellow River based on CEEMDAN-LSSVM-GM(1,1) model

A runoff prediction method based on hyperparameter optimisation of a kernel extreme learning machine with multi-step decomposition

Prediction of Landslide Displacement Based on the Combined VMD-Stacked LSTM-TAR Model

A compound approach for ten-day runoff prediction by coupling wavelet denoising, attention mechanism, and LSTM based on GPU parallel acceleration technology

Rainfall-Runoff Modelling Based on Long Short-Term Memory (LSTM)

A Hybrid Model of Ensemble Empirical Mode Decomposition and Sparrow Search Algorithm-Based Long Short-Term Memory Neural Networks for Monthly Runoff Forecasting