Conditional Flow Matching for Probabilistic Downscaling of Maximum 3-day Snowfall in Alaska

arXiv:2604.25172v1 Announce Type: cross Abstract: Precipitation in complex terrain is governed by orographic processes operating at scales of a few kilometers, yet climate models typically run at resolutions of 50--100km where this topographic detail is absent. Dynamical downscaling with high-resolution regional models such as WRF can resolve these processes, but the computational cost -- months of wall-clock time per scenario -- precludes the large ensembles needed for uncertainty quantification. We present WxFlow, a conditional generative model based on flow matching that learns to map coarse-resolution climate model output and high-resolution topography to calibrated probabilistic ensembles of fine-scale precipitation fields. Applied to 4km WRF simulations of maximum 3-day snowfall over southeast Alaska, WxFlow achieves 87.8% improvement in spectral fidelity and dramatically lower Continuous Ranked Probability Scores relative to conventional lapse-rate-corrected bicubic downscaling, while generating 50-member ensembles in seconds on a laptop. Ensemble spread is spatially coherent and governed by topography, reflecting physically plausible uncertainty structure. All code is available at https://github.com/glide-ism/wrf-flow.

Related

• HardFlow: Hard-Constrained Sampling for Flow-Matching Models via Trajectory Optimization
• climt-paraformer: Stable Emulation of Convective Parameterization using a Temporal Memory-aware Transformer
• FlowRefiner: Flow Matching-Based Iterative Refinement for 3D Turbulent Flow Simulation
• Improvements to the post-processing of weather forecasts using machine learning and feature selection

Source: arXiv cs.LG | 2026-04-29