Hydrology and Earth System Sciences (Sep 2024)

HESS Opinions: Never train a Long Short-Term Memory (LSTM) network on a single basin

  • F. Kratzert,
  • M. Gauch,
  • D. Klotz,
  • G. Nearing

DOI
https://doi.org/10.5194/hess-28-4187-2024
Journal volume & issue
Vol. 28
pp. 4187 – 4201

Abstract

Read online

Machine learning (ML) has played an increasing role in the hydrological sciences. In particular, Long Short-Term Memory (LSTM) networks are popular for rainfall–runoff modeling. A large majority of studies that use this type of model do not follow best practices, and there is one mistake in particular that is common: training deep learning models on small, homogeneous data sets, typically data from only a single hydrological basin. In this position paper, we show that LSTM rainfall–runoff models are best when trained with data from a large number of basins.