Deep Transfer Learning Based on LSTM Model for Reservoir Flood Forecasting

Abstract

In recent years, deep learning has been widely used as an efficient prediction algorithm. However, this algorithm has strict requirements on the size of training samples. If there are not enough samples to train the network, it is difficult to achieve the desired effect. In view of the lack of training samples, this article proposes a deep learning prediction model integrating migration learning and applies it to flood forecasting. The model uses random forest algorithm to extract the flood characteristics, and then uses the transfer learning strategy to fine-tune the parameters of the model based on the model trained with similar reservoir data; and is used for the target reservoir flood prediction. Based on the calculation results, an autoregressive algorithm is used to intelligently correct the error of the prediction results. A series of experimental results show that our proposed method is significantly superior to other classical methods in prediction accuracy.