arXiv Machine Learning

Learning Intrinsic Water-Quality Dynamics with Rainfall for Data-Driven Forecasting

The paper introduces RaiNet, a data‑driven model that jointly learns multiscale water‑quality dynamics and station‑specific rainfall effects. It uses LocTrend to capture irregular water‑quality patterns, constructs station‑oriented rainfall events from gridded precipitation, and applies XGateFusion for lag‑aware fusion across scales. Experiments on three new multimodal datasets show RaiNet surpasses existing time‑series, water‑quality, diffusion‑based, and spatiotemporal models by over 20%, with each module contributing uniquely to performance.

arXiv AI
Sep 10

PCSDiff: Diffusion-Based Bias Correction and Super Resolution Toward Practical Operational Medium-Term Precipitation Forecast

PCSDiff is a diffusion-based framework designed to correct systematic biases and enhance spatial resolution in medium-term (10‑day) precipitation forecasts. It uses a Precipitation Intensity‑aware Multi‑branch Decoder to mitigate dynamic multi‑day errors and a two‑phase conditional diffusion super‑resolution module to restore fine‑scale rainfall patterns. Evaluated over China, PCSDiff reduces RMSE by 16.1% and increases ACC by 13.9% compared to raw ECMWF forecasts, outperforming mainstream deep‑learning baselines and enabling low‑latency rolling forecasts for operational use.

By Yuze Sun, Shiyi Wang, Jiancheng Pan, Die Wang, Andreas F. Prein, Wentao Luo, Linhan Jiang, Jie Wu, Quan Zhang, Xiaomeng Huang
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 Machine Learning
Jun 4

Uncovering Insights of Compound Flooding with Data-Driven AI

arXiv:2506. 04281v2 Announce Type: replace Abstract: Compound flooding, driven by nonlinear interactions between multiple hydrometeorological factors, poses a significant challenge to hazard prevention.

By Xu Zheng, Chaohao Lin, Sipeng Chen, Zhuomin Chen, Jimeng Shi, Jayantha Obeysekera, Jingchao Ni, Wei Cheng, Jason Liu, Dongsheng Luo
arXiv Machine Learning
Sep 22

Gaussian Process Decorrelation for Spatiotemporal Deep Learning-Based Snow Water Equivalent Prediction

The paper proposes a method for predicting future snow water equivalent (SWE) across the Western United States by first removing spatial correlations using a Gaussian Process-based linear transformation, then training a long short-term memory (LSTM) neural network on the decorrelated data. This separation of spatial and temporal components improves predictive accuracy compared to baseline models. Additionally, the authors incorporate conformal prediction to provide distribution‑free uncertainty estimates for SWE forecasts.

By Colin Fenster, Adrienne Marshall, Soutir Bandyopadhyay, Daniel McKenzie
arXiv AI
Sep 7

MZ-Rain: Moisture-Budget-Guided Zero-Inflated Model for Station-Level Precipitation Nowcasting

MZ‑Rain is a moisture‑budget‑guided zero‑inflated sLSTM framework designed for station‑level precipitation nowcasting. It decomposes precipitation formation into moisture storage, transport, surface evaporation, and persistence pathways, each modeled by dedicated sLSTM branches, and employs an adaptive Tweedie strategy to handle the dataset’s severe zero inflation. Experiments across varied climates show MZ‑Rain outperforms strong baselines on metrics such as CSI, FAR, MSE, and MAE, especially for heavy precipitation events.

By Yifang Zhang, Shengwu Xiong, Henan Wang, Wenjie Yin, Yuqiang Zhang, Chen Zhou, Hua Chen, Qile Zhao, Pengfei Duan
arXiv Statistics ML
Sep 11

Stress-Testing Dynamical and Generative Downscaling Using Subseasonal Extreme Precipitation Forecasts

The study compares the Weather Research and Forecasting (WRF) dynamical model with an unpaired diffusion-based generative model for downscaling extreme precipitation events up to three weeks ahead. Both models outperform raw European Centre for Medium-Range Weather Forecasts forecasts when evaluated against Swiss rain gauge-radar observations, but their strengths differ by atmospheric regime: WRF excels in a multicell, non‑stationary event, while the diffusion model performs more consistently and better in a stationary supercell event.

By Mauricio Lima, Marika Koukoula, Romain Pilon, Monika Feldmann, Erwan Koch, Daniela I. V. Domeisen, Tom Beucler