Frederik Kratzert, Daniel Klotz, Guy Shalev, Günter Klambauer, Sepp Hochreiter, and Grey Nearing

LSTM cells

The standard LSTM cell and the proposed Entity-Aware LSTM.

Regional rainfall-runoff modeling remains a largely unsolved problem in hydrological sciences. This is because traditional hydrological models degrade significantly in performance when calibrated for a set of multiple basins compared to a single basin. Here, we use Long Short-Term Memory networks (LSTMs) and demonstrate that this problem can be overcome with data-driven models.

We are able to significantly improve performance compared to several different hydrological benchmark models. We train a single LSTM model on 531 basins from the Catchment Attributes and Meteorology for Large-sample Studies (CAMELS) dataset using meteorological time series data and catchment features. Our model significantly outperforms the models calibrated regionally, and achieves better performance compared to models calibrated for each basin individually.

Furthermore, we propose an adaption to the standard LSTM architecture, which we call an Entity-Aware-LSTM (EA-LSTM). The EA-LSTM model allows evaluating the catchment characteristics and embedding it as a feature layer in a deep learning model. The EA-LSTM model is also able to learn complex interactions between the catchments attributes that allows grouping of different basins.

Hydrology and Earth System Sciences, 23, 12, 5089–5110, 2019-12-17.

View paper
IARAI Authors
Dr Sepp Hochreiter
Weather and Climate
Deep Learning, Hydrology, Long Short-Term Memory, Rainfall-Runoff


Imprint | Privacy Policy

Stay in the know with developments at IARAI

We can let you know if there’s any

updates from the Institute.
You can later also tailor your news feed to specific research areas or keywords (Privacy)

Log in with your credentials

Forgot your details?

Create Account