arXiv Machine Learning

Diffusion Precipitation Generation with Medium-range Weather Forecasting

The paper introduces a plug‑and‑play framework called Dynamics‑Precipitation Decoupling that employs a diffusion generator to convert non‑precipitation forecasts from a frozen weather model into observation‑aligned daily precipitation. The generator is conditioned on atmospheric states and satellite‑gauge precipitation estimates, using a multi‑scale recurrent encoder and cross‑attention to capture heterogeneous rainfall distributions and align daily accumulations with evolving atmospheric signals. Experiments on global and regional datasets, including a heavy‑rainfall case study in China, show that the method improves daily precipitation accuracy, event detection, and localized accumulation patterns compared to several baseline weather models and ensemble forecasts.

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
Sep 25

Improving global precipitation forecasts with an AI weather model trained on satellite observations

The paper presents Laxmi, a retrained version of the AIFS weather model that uses satellite-based precipitation observations instead of ERA5 reanalysis data. Laxmi achieves a 19% improvement in global probabilistic accuracy, reduces drizzle overprediction by 33%, and boosts the 95th percentile Brier skill score by 57%. In a case study of 10 Indian tropical storms, Laxmi accurately forecasted 150 mm event-total precipitation in 7 events, outperforming both the original AIFS and the leading physical model IFS.

By Julian F. Schmitt, Bertrand Delorme, Robert C. King, Yashica Patodia, Tapio Schneider, Aditi Sheshadri, Ravi Jain
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 AI
Sep 16

Composable multi-satellite precipitation estimation for evolving observing systems

The paper introduces PRISMA, a generative framework that separates precipitation prior training from sensor-specific constraints, allowing flexible composition of heterogeneous satellite observations without retraining the core model. By integrating FY‑4B/AGRI, GPM/GMI, F16‑F18 SSMIS, and GPM/DPR‑Ka data, PRISMA consistently improves precipitation‑estimation accuracy and outperforms IMERG Final in CRPS and RMSE while maintaining positive Brier skill across thresholds. The framework supports rapid, accurate ensemble precipitation estimates, enhancing satellite‑based monitoring for hydrometeorological hazards.

By Yunfan Yang, Haofei Sun, Xiuyu Sun, Wei Han, Xiaoze Xu, Xingtao Song, Jun Li, Zhiqiu Gao, Wei Huang
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 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 Machine Learning
Aug 19

MAGPIE-Net: Predicting short-duration heavy-rainfall events in station neighborhoods from multitemporal FY-4A AGRI observations

MAGPIE‑Net is a deep‑learning framework that directly maps multitemporal FY‑4A AGRI infrared and water‑vapor observations to short‑duration heavy‑rainfall warnings for irregular station neighborhoods. By embedding a geographically adaptive, differentiable grid‑to‑station mapping and training with station‑neighborhood event losses, the model outperforms traditional gridded‑precipitation baselines, achieving higher detection rates and longer lead times in 2023 warm‑season tests over China.

By Xiang Lin, Yunying Li, Chengzhi Ye, Zitong Chen, Jing Sun
arXiv Machine Learning
Oct 9

Just Weather Scoring: Efficient End-to-end Nowcasting with Distributional Diffusion

Just Weather Scoring (JWS) is a single‑stage, end‑to‑end diffusion model that forecasts precipitation directly in radar space, eliminating the need for separate compression or deterministic components. By combining Masked Asynchronous Diffusion with a scoring‑rule objective, JWS adapts diffusion training to high‑dimensional spatio‑temporal data and enables few‑step generation. On SEVIR and MeteoNet benchmarks, JWS achieves state‑of‑the‑art probabilistic forecasting while using fewer parameters and delivering over 17× faster inference than existing methods.

By Jannik Wiese, Johannes Schusterbauer, Tommaso Martorella, Bj\"orn Ommer
arXiv Machine Learning
Jun 26

CSU-PCAST: A Dual-Branch Transformer Framework for medium-range ensemble Precipitation Forecasting

arXiv:2510. 20769v2 Announce Type: replace-cross Abstract: Accurate medium-range precipitation forecasting is essential for hydrometeorological risk management but remains challenging for both numerical weather prediction (NWP) systems and data-driven models.

By Tianyi Xiong, Haonan Chen, Kelly Mahoney, Jingyin Tang, Tim Smith, Janice Bytheway