Using long short-term memory networks for river flow prediction

Abstract

Deep learning has made significant advances in methodologies and practical applications in recent years. However, there is a lack of understanding on how the long short-term memory (LSTM) networks perform in river flow prediction. This paper assesses the performance of LSTM networks to understand the impact of network structures and parameters on river flow predictions. Two river basins with different characteristics, i.e., Hun river and Upper Yangtze river basins, are used as case studies for the 10-day average flow predictions and the daily flow predictions, respectively. The use of the fully connected layer with the activation function before the LSTM cell layer can substantially reduce learning efficiency. On the contrary, non-linear transformation following the LSTM cells is required to improve learning efficiency due to the different magnitudes of precipitation and flow. The batch size and the number of LSTM cells are sensitive parameters and should be carefully tuned to achieve a balance between learning efficiency and stability. Compared with several hydrological models, the LSTM network achieves good performance in terms of three evaluation criteria, i.e., coefficient of determination, Nash–Sutcliffe Efficiency and relative error, which demonstrates its powerful capacity in learning non-linear and complex processes in hydrological modelling. HIGHLIGHTS Long short-term memory (LSTM) networks are assessed for river flow prediction.; The impacts of network structures and parameters on learning efficiency are analysed.; The batch size and the number of LSTM cells are sensitive parameters for learning.; LSTM has good predictive accuracy compared to hydrological and data-driven models tested.;

Keywords

• hydrological modelling
• lstm
• machine learning
• river flow prediction