arXiv Machine Learning

Learning global river system hydrodynamics with a topology-informed foundation model

The paper introduces GraphRiverCast (GRC), a topology-informed foundation model that predicts daily multivariate hydrodynamics across the global river network using sparse gauge data. By leveraging river connectivity and energetics, GRC outperforms both physics-based and traditional machine-learning models, maintaining skill over long lead times and generalizing to ungauged reaches and finer-resolution networks. The approach demonstrates how incorporating system structure can improve predictions in observation-limited Earth systems.

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
Oct 3

Physics-Refined Spatiotemporal Forecasting on Open-Boundary Hydrologic Graphs

The paper introduces a physics‑refined framework for spatiotemporal forecasting on open‑boundary hydrologic graphs, addressing instability caused by missing external boundary forcing. It learns ghost node proxies to approximate unobserved inputs and applies two physics refiners: one enforcing local consistency with two‑hop neighbors, and another using a physics‑guided graph neural operator to reduce long‑horizon drift. Experiments on two real‑world hydrologic graphs show improved prediction accuracy and stability compared to existing learning‑based and physics‑informed models.

By Haoyang Jiang, Zhengui Wang, Shenghan Gao, Y. Joseph Zhang, Xingquan Zhu, Yi He
Hugging Face Trending Papers
Sep 10

A Dataset and Model for Imputing Water Surface Elevation on a Large and Extremely Sparse Spatiotemporal Graph

The paper introduces AmazonSWE, a dataset covering over 19,000 river sections in the Amazon basin for 10 years, combining satellite altimetry and in‑situ gauge data to enable large‑scale spatiotemporal graph imputation. The authors highlight the extreme sparsity of observations—less than 1% of sections per day—and the directed acyclic topology of river networks, which challenge existing imputation methods. They propose a bidirectional selective state‑space model that samples connected subgraphs and uses topology‑aware positional encodings, achieving 18–39% lower RMSE than the current state‑of‑the‑art SWOT‑based approach while providing predictions for all river sections.

arXiv Machine Learning
Sep 11

A Dataset and Model for Imputing Water Surface Elevation on a Large and Extremely Sparse Spatiotemporal Graph

The paper introduces AmazonSWE, a dataset covering over 19,000 river sections in the Amazon basin for 10 years (2016‑2026) that integrates satellite altimetry, including SWOT, to enable large‑scale spatiotemporal graph imputation. The dataset is extremely sparse—fewer than 1% of sections are observed daily—and features a directed acyclic river topology that is larger and structurally distinct from existing benchmarks. The authors demonstrate that conventional imputation methods struggle with this topology, scale, and sparsity, and propose a bidirectional selective state‑space model that outperforms prior approaches, reducing RMSE against in‑situ gauges by 18‑39% and providing predictions for every river section. whyItMatters":"AmazonSWE offers a novel, real‑world use case that could improve flood forecasting and water resource management by enabling more accurate and comprehensive water surface elevation estimates across a vast, sparsely monitored river network."

By Ruben Cartuyvels, Karim Douch, Gabriele Bertoli, Mounia El Baz, Artemis Vrettou, S\'ebastien Lef\`evre, Diego Fernandez Prieto
arXiv Machine Learning
Jun 11

LakeFM: Toward a Foundation Model for Aquatic Ecosystems Using Irregular Multivariate Multi-depth Time Series Data

arXiv:2606. 11268v1 Announce Type: new Abstract: Understanding and forecasting lake dynamics is critical for monitoring water quality and ecosystem health across lakes and reservoirs.

By Abhilash Neog, Sepideh Fatemi, Medha Sawhney, Kazi Sajeed Mehrab, Aanish Pradhan, Bennett J. McAfee, Emma Marchisin, Arka Daw, Robert Ladwig, Cayelan C. Carey, Paul Hanson, Anuj Karpatne
arXiv AI
Oct 27

EvoBench: An Evolving, Diagnostic Benchmark Framework for Environmental Modeling

EvoBench is a diagnostic benchmark framework designed to reveal where and when machine‑learning models for environmental variables fail, rather than relying on a single average accuracy score. It tracks model performance over time and across spatial extrapolation, highlighting issues such as non‑stationarity, poor performance at unmonitored sites, and under‑prediction of rare events like floods. In a case study on daily streamflow prediction across 9,000 U.S. basins, EvoBench shows that no model dominates all scenarios and that targeted fixes can mitigate identified weaknesses.

By Yiming Sun, Yuan Zhang, Qi Cheng, Yingda Fan, Runlong Yu, Yiqun Xie, Xiaowei Jia