Meta-LSTM in hydrology: Advancing runoff predictions through model-agnostic meta-learning

Kaixuan Cai,Jinxin He,Qingliang Li,Wei Shangguan,Lu Li,Huiming Hu
DOI: https://doi.org/10.1016/j.jhydrol.2024.131521
IF: 6.4
2024-06-20
Journal of Hydrology
Abstract:In the field of hydrology, deep learning has become a prevalent tool for runoff simulation. However, the limitations stem from their primary focus on normality, which fails to accurately capture rare events. This study introduces a novel Long Short-Term Memory (LSTM) model, termed Meta-LSTM, which is based on the Model-Agnostic Meta-Learning (MAML) framework. The Meta-LSTM model is capable of dynamically fine-tuning its parameters to ensure effective adaptation to various runoff scenarios. From this Meta-LSTM model, the Meta-enhanced paradigm is subsequently proposed, ensuring effective adaptation to diverse models and datasets. We conducted experiments with various models on the CAMELS and CAMELS-AUS datasets. Compared to conventional deep learning models for runoff simulation, Meta-enhanced models exhibit substantial improvements. Specifically, the Meta-LSTM model reduces the PBIAS from −13.86 % to −5.18 %, increasing the Kling-Gupta Efficiency (KGE) by 7 % on the CAMELS dataset. For the Meta-GR4J-CNN model, the PBIAS decreases by 21 %, and the Nash-Sutcliffe Efficiency (NSE) increases by 2 % on the CAMELS-AUS dataset. This demonstrates the ability of the Meta-enhanced model to more accurately represent observed data without significant additional time costs. Our approach overcomes the problem of uniform training in simulations of complex scenarios and promises to significantly improve the accuracy of future hydrologic studies.
geosciences, multidisciplinary,water resources,engineering, civil
What problem does this paper attempt to address?