A New Rainfall-Runoff Model Using Improved LSTM with Attentive Long and Short Lag-Time
Xi Chen, Jiaxu Huang, Sheng Wang, Gongjian Zhou, Hongkai Gao, Min Liu, Ye Yuan, Laiwen Zheng, Qingli Li, Honggang Qi
Abstract
It is important to improve the forecasting performance of rainfall-runoff models due to the high complexity of basin response and frequent data limitations. Recently, many studies have been carried out based on deep learning and have achieved significant performance improvements. However, their intrinsic characteristics remain unclear and have not been explored. In this paper, we pioneered the exploitation of short lag-times in rainfall-runoff modeling and measured its influence on model performance. The proposed model, long short-term memory with attentive long and short lag-time (LSTM-ALSL), simultaneously and explicitly uses new data structures, i.e., long and short lag-times, to enhance rainfall-runoff forecasting accuracy by jointly extracting better features. In addition, self-attention is employed to model the temporal dependencies within long and short lag-times to further enhance the model performance. The results indicate that LSTM-ALSL yielded superior performance at four mesoscale stations (1846~9208 km2) with humid climates (aridity index 0.77~1.16) in the U.S.A., for both peak flow and base flow, with respect to state-of-the-art counterparts.