arXiv Machine Learning

Flow Matching for Convective-Scale Precipitation Downscaling

arXiv:2606. 00281v1 Announce Type: cross Abstract: Generative machine learning is an increasingly important complement to dynamical downscaling for producing high-resolution precipitation projections, with diffusion models currently the leading approach.

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
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 30

High-Resolution Climate Projections Using Diffusion-Based Downscaling of a Lightweight Climate Emulator

arXiv:2602. 13416v2 Announce Type: replace Abstract: The proliferation of data-driven models in weather and climate sciences has marked a significant paradigm shift, with advanced models demonstrating exceptional skill in medium-range forecasting.

By Haiwen Guan, Dibyajyoti Chakraborty, Moein Darman, Troy Arcomano, Ashesh Chattopadhyay, Romit Maulik
arXiv Machine Learning
Jun 9

SwAIther-Precip: Lead-Time-Aware Bias Correction Enables Kilometer-Scale Downscaling of Global AI Precipitation Forecasts over Switzerland

arXiv:2605. 16163v2 Announce Type: replace-cross Abstract: Skillful medium-range precipitation forecasting at kilometer scale remains challenging over complex terrain because precipitation arises from multiscale nonlinear processes that global models cannot explicitly resolve at affordable cost.

By Dan Assouline, Erwan Koch, Federico Amato, Filippo Quarenghi, Daniele Nerini, Thibaut Loiseau, Kyle van de Langemheen, Tom Beucler
arXiv Machine Learning
Sep 25

Evaluating Cross-region Generalization for Wavelet-Diffusion Precipitation Downscaling

The study evaluates how well wavelet-diffusion models (WDM) for precipitation downscaling generalize across different U.S. regions and precipitation regimes. A WDM trained only on Oklahoma data performs competitively in other regions, while a model trained on all six regions achieves the best overall image-domain and detection performance, though improvements vary with precipitation intensity. The analysis shows a strong correlation between spatial autocorrelation (Moran's I) and detection success, indicating that model performance is linked to the spatial organization of precipitation fields.

By Weikang Qian, Yixin Wen, Chugang Yi, Zhi Li, Lingcheng Li, Haizhao Yang
arXiv Machine Learning
Aug 27

Precipitation Downscaling Using Foundation Model-Conditioned Diffusion

The study evaluates three conditioning strategies for a denoising diffusion probabilistic model to downscale daily precipitation for the Colorado River Basin. Channel concatenation of upsampled coarse predictors yields the lowest point‑wise CRPS and MSE but tends to over‑smooth high‑intensity events. Cross‑attention conditioning—both with a learned encoder and with the frozen encoder of the pretrained Prithvi‑WxC weather foundation model—provides better distributional realism, improved spectral fidelity, and stronger performance on extreme events, especially when data are limited.

By Victor Nascimento Ribeiro, Jorge Guevara, Jorge Sebastian Moraga, Chris Lucas, Natalie Lord, Andrew Taylor, Edward Lockhart, Will Trojak, Johannes Schmude, Anne Jones
arXiv Machine Learning
Sep 4

Improving precipitation forecasts in an AI weather model using observational data

The paper presents a graph-transformer AI weather model that is fine‑tuned with high‑resolution IMERG precipitation observations, moving beyond the traditional reliance on the ERA5 reanalysis dataset. This approach yields up to a 19% improvement in medium‑range continuous ranked probability scores and a 57% better Brier skill score for extreme rainfall compared to leading operational models, while also excelling in tropical storm and drizzle prediction. The study demonstrates that directly incorporating observation‑based precipitation data into AI training can markedly enhance forecast accuracy, though physics‑based models still outperform for the heaviest events.

By Julian F. Schmitt, Bertrand Delorme, Robert C. King, Yashica Patodia, Tapio Schneider, Aditi Sheshadri, Ravi Jain
arXiv Machine Learning
Sep 11

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.

By Ziqi Wang, Hailiang Zhao, Cheng Bao, Daojiang Hu, Wenzhuo Qian, Shuiguang Deng