Preprints
https://doi.org/10.5194/egusphere-2026-2950
https://doi.org/10.5194/egusphere-2026-2950
16 Jun 2026
 | 16 Jun 2026
Status: this preprint is open for discussion and under review for Hydrology and Earth System Sciences (HESS).

Technical note: Regional fine-tuning of LSTMs for improved streamflow predictions in ungauged catchments

Ashkan Shokri, James C. Bennett, and David E. Robertson

Abstract. Predicting streamflow in ungauged basins (PUB) remains a central challenge in hydrology. Long short-term memory (LSTM) networks trained on large samples of catchments ("global" LSTMs) have emerged as a state-of-the-art approach for PUB, outperforming conceptual rainfall–runoff models with traditional regionalisation approaches. However, global LSTMs are spatially agnostic, relying solely on static catchment attributes to differentiate regional hydrological behaviour. This study introduces Regionalised Fine-Tuning (ReFT), a strategy that adapts a pretrained global LSTM to the region surrounding each ungauged target catchment by fine-tuning on a spatially weighted set of donor catchments using an inverse-distance weighting scheme. ReFT is evaluated on 218 catchments from the CAMELS-AUS dataset under a spatial out-of-sample cross-validation framework, comparing two fine-tuning configurations: updating all model parameters versus updating only the prediction head while keeping the recurrent backbone frozen. ReFT improves Nash–Sutcliffe Efficiency relative to the base global LSTM in more than 66 % of catchments, with the largest gains occurring for catchments of moderate baseline performance. The ReFT framework combines the broad process generalisation of large-sample deep learning with the local specificity of regional adaptation, providing an efficient route to improved streamflow predictions in data-sparse regions.

Publisher's note: Copernicus Publications remains neutral with regard to jurisdictional claims made in the text, published maps, institutional affiliations, or any other geographical representation in this paper. While Copernicus Publications makes every effort to include appropriate place names, the final responsibility lies with the authors. Views expressed in the text are those of the authors and do not necessarily reflect the views of the publisher.
Share
Ashkan Shokri, James C. Bennett, and David E. Robertson

Status: open (until 28 Jul 2026)

Comment types: AC – author | RC – referee | CC – community | EC – editor | CEC – chief editor | : Report abuse
Ashkan Shokri, James C. Bennett, and David E. Robertson
Ashkan Shokri, James C. Bennett, and David E. Robertson
Metrics will be available soon.
Latest update: 16 Jun 2026
Download
Short summary
Predicting streamflow in ungauged river catchments, where no flow measurements exist, is a long-standing challenge in hydrology. We developed a method that takes an artificial-intelligence model trained on many gauged catchments across Australia and finetuned it to each ungauged catchment using data from nearby gauged ones, giving closer catchments more weight. Tested on 218 catchments, it improved streamflow predictions, offering an efficient way to predicting flow in poorly gauged regions.
Share