Articles | Volume 28, issue 4
https://doi.org/10.5194/hess-28-945-2024
https://doi.org/10.5194/hess-28-945-2024
Research article
 | 
27 Feb 2024
Research article |  | 27 Feb 2024

Toward interpretable LSTM-based modeling of hydrological systems

Luis Andres De la Fuente, Mohammad Reza Ehsani, Hoshin Vijai Gupta, and Laura Elizabeth Condon

Data sets

A large-sample watershed-scale hydrometeorological dataset for the contiguous USA Andrew Newman et al. https://doi.org/10.5065/D6MW2F4D

Model code and software

ldelafue/Hydro-LSTM: HydroLSTM (v1.0.0) Luis Andres De la Fuente and Andrew Bennett https://doi.org/10.5281/zenodo.10694927

Download
Short summary
Long short-term memory (LSTM) is a widely used machine-learning model in hydrology, but it is difficult to extract knowledge from it. We propose HydroLSTM, which represents processes like a hydrological reservoir. Models based on HydroLSTM perform similarly to LSTM while requiring fewer cell states. The learned parameters are informative about the dominant hydrology of a catchment. Our results show how parsimony and hydrological knowledge extraction can be achieved by using the new structure.