# Enhancing streamflow forecast and extracting insights using long-short term memory networks with data integration at continental scales

arXiv: https://arxiv.org/abs/1912.08949 · HTML: https://mlsanity.com/paper/1912.08949/

## Key facts

- Cited by 27 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 18 Dec 2019.

## Built on by (most cited first, 27 of 27)

- [Differentiable, learnable, regionalized process-based models with physical outputs can approach state-of-the-art hydrologic prediction accuracy](https://mlsanity.com/paper/2203.14827/) (2022)
- [Robust Inverse Framework using Knowledge-guided Self-Supervised Learning: An application to Hydrology](https://mlsanity.com/paper/2109.06429/) (2021)
- [Flood forecasting with machine learning models in an operational framework](https://mlsanity.com/paper/2111.02780/) (2021)
- [Physics-aware Machine Learning Revolutionizes Scientific Paradigm for Machine Learning and Process-based Hydrology](https://mlsanity.com/paper/2310.05227/) (2023)
- [The data synergy effects of time-series deep learning models in hydrology](https://mlsanity.com/paper/2101.01876/) (2021)
- [Probing the limit of hydrologic predictability with the Transformer network](https://mlsanity.com/paper/2306.12384/) (2023)
- [A Mass-Conserving-Perceptron for Machine Learning-Based Modeling of Geoscientific Systems](https://mlsanity.com/paper/2310.08644/) (2023)
- [Enhancing predictive skills in physically-consistent way: Physics Informed Machine Learning for Hydrological Processes](https://mlsanity.com/paper/2104.11009/) (2021)
- [Towards Interpretable Physical-Conceptual Catchment-Scale Hydrological Modeling using the Mass-Conserving-Perceptron](https://mlsanity.com/paper/2401.14521/) (2024)
- [Continental-scale streamflow modeling of basins with reservoirs: towards a coherent deep-learning-based strategy](https://mlsanity.com/paper/2101.04423/) (2021)
- [Differentiable modeling to unify machine learning and physical models and advance Geosciences](https://mlsanity.com/paper/2301.04027/) (2023)
- [Temporal Fusion Transformers for Streamflow Prediction: Value of Combining Attention with Recurrence](https://mlsanity.com/paper/2305.12335/) (2023)
- [Using Machine Learning to Discover Parsimonious and Physically-Interpretable Representations of Catchment-Scale Rainfall-Runoff Dynamics](https://mlsanity.com/paper/2412.04845/) (2024)
- [Earthquake Nowcasting with Deep Learning](https://mlsanity.com/paper/2201.01869/) (2021)
- [Utilizing Earth Foundation Models to Enhance the Simulation Performance of Hydrological Models with AlphaEarth Embeddings](https://mlsanity.com/paper/2601.01558/) (2026)
- [Towards Generalized Hydrological Forecasting using Transformer Models for 120-Hour Streamflow Prediction](https://mlsanity.com/paper/2406.07484/) (2024)
- [Joint distribution of upstream runoff governs downstream river-discharge prediction uncertainty in distributed ML models](https://mlsanity.com/paper/2607.03217/) (2026)
- [Machine Learning for Postprocessing Ensemble Streamflow Forecasts](https://mlsanity.com/paper/2106.09547/) (2021)
- [Toward Routing River Water in Land Surface Models with Recurrent Neural Networks](https://mlsanity.com/paper/2404.14212/) (2024)
- [Hierarchically Disentangled Recurrent Network for Factorizing System Dynamics of Multi-scale Systems: An application on Hydrological Systems](https://mlsanity.com/paper/2407.20152/) (2024)
- [Update hydrological states or meteorological forcings? Comparing data assimilation methods for differentiable hydrologic models](https://mlsanity.com/paper/2502.16444/) (2025)
- [Distinct hydrologic response patterns and trends worldwide revealed by physics-embedded learning](https://mlsanity.com/paper/2504.10707/) (2025)
- [HydroFusion-LMF: Semi-Supervised Multi-Network Fusion with Large-Model Adaptation for Long-Term Daily Runoff Forecasting](https://mlsanity.com/paper/2510.03744/) (2025)
- [Rethinking deep learning: linear regression remains a key benchmark in predicting terrestrial water storage](https://mlsanity.com/paper/2510.10799/) (2025)
- [Process-Aware AI for Rainfall-Runoff Modeling: A Mass-Conserving Neural Framework with Hydrological Process Constraints](https://mlsanity.com/paper/2603.25093/) (2026)
- [A Differentiable Framework for Global Circulation Model Precipitation Bias Correction](https://mlsanity.com/paper/2604.23045/) (2026)
- [Transformer-based Diffusion models for Hydrological Time Series Probabilistic Imputation and Forecasting](https://mlsanity.com/paper/2607.21200/) (2026)

## Abstract (arXiv)

Recent observations with varied schedules and types (moving average, snapshot, or regularly spaced) can help to improve streamflow forecasts, but it is challenging to integrate them effectively. Based on a long short-term memory (LSTM) streamflow model, we tested multiple versions of a flexible procedure we call data integration (DI) to leverage recent discharge measurements to improve forecasts. DI accepts lagged inputs either directly or through a convolutional neural network (CNN) unit. DI ubiquitously elevated streamflow forecast performance to unseen levels, reaching a record continental-scale median Nash-Sutcliffe Efficiency coefficient value of 0.86. Integrating moving-average discharge, discharge from the last few days, or even average discharge from the previous calendar month could all improve daily forecasts. Directly using lagged observations as inputs was comparable in performance to using the CNN unit. Importantly, we obtained valuable insights regarding hydrologic processes impacting LSTM and DI performance. Before applying DI, the base LSTM model worked well in mountainous or snow-dominated regions, but less well in regions with low discharge volumes (due to either low precipitation or high precipitation-energy synchronicity) and large inter-annual storage variability. DI was most beneficial in regions with high flow autocorrelation: it greatly reduced baseflow bias in groundwater-dominated western basins and also improved peak prediction for basins with dynamical surface water storage, such as the Prairie Potholes or Great Lakes regions. However, even DI cannot elevate high-aridity basins with one-day flash peaks. Despite this limitation, there is much promise for a deep-learning-based forecast paradigm due to its performance, automation, efficiency, and flexibility.
