HESS Opinions: Never train a Long Short-Term Memory (LSTM) network on a single basin
Abstract. 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.
Authors
- Frederik Kratzert (ORCID: https://orcid.org/0000-0002-8897-7689)
- Daniel Klotz (ORCID: https://orcid.org/0000-0002-9843-6798)
- Martin Gauch (ORCID: https://orcid.org/0000-0002-4587-898X)
- Grey Nearing (ORCID: https://orcid.org/0000-0001-7031-6770)
Institutions
- Google (United States) (US)
- Helmholtz Centre for Environmental Research (DE)
- Google (Switzerland) (CH)
Publication Details
- Journal
- Hydrology and earth system sciences
- Published
- 2024-09-12
- DOI
- https://doi.org/10.5194/hess-28-4187-2024
- Citations
- 104
- Primary Topic
- Hydrological Forecasting Using AI
- Type
- article
- Field-Weighted Citation Impact
- 21.40