arXiv Machine Learning

Joint distribution of upstream runoff governs downstream river-discharge prediction uncertainty in distributed ML models

arXiv:2607. 03217v1 Announce Type: new Abstract: Uncertainty quantification of hydrological predictions is necessary to inform operational decisions.

arXiv Machine Learning
Jun 16

Towards CONUS-Wide ML-Augmented Conceptually-Interpretable Modeling of Catchment-Scale Precipitation-Storage-Runoff Dynamics

arXiv:2510. 02605v2 Announce Type: replace Abstract: While many modern studies are dedicated to ML-based large-sample hydrologic modeling, these efforts have not necessarily translated into predictive improvements that are grounded in enhanced physical-conceptual understanding.

By Yuan-Heng Wang, Yang Yang, Fabio Ciulla, Hoshin V. Gupta, Charuleka Varadharajan
arXiv AI
Sep 10

Resolving sources of uncertainty in AI weather forecasting

The paper introduces Pangu‑Bayes, a probabilistic forecasting hierarchy that separates atmospheric‑state uncertainty from learned‑model uncertainty as distinct stochastic variables, allowing cross‑flow perturbations of the evolving state with Bayesian parameter samples. In tests on 90 held‑out 2023 tropical cyclones, Pangu‑Bayes reduces track, pressure, and wind errors by 54.2%, 17.2%, and 24.9% respectively, and improves rapid‑intensification detection. The study finds that atmospheric‑state variability more consistently improves track prediction, while learned‑model variability more often enhances intensity prediction, demonstrating how model‑defined uncertainty resolution can be linked to target‑dependent value and dynamical interpretation.

By Wenbo Hu, Xinlei Xiong, Shuxun Zhou, Kaifeng Bi, Lingxi Xie, Jun Zhu, Richang Hong, Qi Tian
arXiv Machine Learning
Sep 2

Process-Aware AI for Rainfall-Runoff Modeling: A Mass-Conserving Neural Framework with Hydrological Process Constraints

The paper introduces the Mass‑Conserving Perceptron (MCP), a physics‑aware AI framework that enforces conservation laws while learning hydrological process relationships from data. By progressively adding physically meaningful components—such as bounded soil storage, state‑dependent conductivity, variable porosity, infiltration capacity, surface ponding, vertical drainage, and nonlinear water‑table dynamics—to a single MCP storage unit, the authors demonstrate that predictive skill for daily streamflow improves across 15 U.S. catchments. The study finds that the impact of each process representation varies with hydroclimate, with vertical drainage boosting performance in arid and snow‑dominated basins but hindering it in rainfall‑dominated ones, while surface ponding has minimal effect; the best MCP configurations rival LSTM benchmarks while retaining explicit physical interpretability.

By Mohammad A. Farmani, Hoshin V. Gupta, Ali Behrangi, Muhammad Jawad, Sadaf Moghisi, Guo-Yue Niu
arXiv Machine Learning
Aug 26

Topology enables learning-based hydrodynamic prediction of the global river system

arXiv:2602.22293v2 Announce Type: replace Abstract: Accurate river prediction is essential for water, food and energy security, yet remains challenging across entire river networks. Machine learning...

By Hancheng Ren, Gang Zhao, Shuo Wang, Louise Slater, Dai Yamazaki, Shu Liu, Jingfang Fan, Xueying Li, Shibo Cui, Ziming Yu, Shengyu Kang, Depeng Zuo, Dingzhi Peng, Zongxue Xu, Bo Pang
arXiv Machine Learning
Aug 28

SimCast-S2S: An Efficient Generative Model for Subseasonal Precipitation Forecasting via Transfer Learning from Climate Simulations

SimCast‑S2S is a generative latent‑diffusion model designed for probabilistic subseasonal‑to‑seasonal precipitation forecasting. It tackles three key challenges: it uses a diffusion pipeline to capture uncertainty, operates in a compact latent space to enable efficient large‑ensemble generation, and leverages transfer learning with low‑rank adaptation to train on limited reanalysis data after pretraining on climate simulations. The model outperforms deep‑learning baselines and competes with, or surpasses, operational systems such as the ECMWF‑S2S baseline without requiring extensive post‑processing.

By Hiep V. Dang, Antonios Mamalakis
arXiv Machine Learning
Sep 4

SimCast-S2S: A Computationally Efficient Diffusion Model for Subseasonal Precipitation Forecasting

SimCast‑S2S is a generative latent‑diffusion framework designed for probabilistic subseasonal‑to‑seasonal precipitation forecasting. It tackles three key challenges: it uses a diffusion‑based generative pipeline for uncertainty quantification, operates in a compact latent space learned by VAEs for efficient large‑ensemble generation, and employs transfer learning with LoRA to overcome limited training data. On reanalysis data, it outperforms deep‑learning baselines and competes with or surpasses state‑of‑the‑art operational systems such as ECMWF‑S2S.

By Hiep V. Dang, Antonios Mamalakis