CRPS-LAM: Regional ensemble weather forecasting from matching marginals
Erik Larsson, Joel Oskarsson, Tomas Landelius, Fredrik Lindsten
TL;DR
This work addresses probabilistic regional weather forecasting for Limited-Area Modeling (LAM) where diffusion-based sampling is computationally expensive. It introduces CRPS-LAM, a CRPS-based probabilistic LAM that samples an ensemble in a single forward pass by injecting a latent vector $z \,\sim\, \mathcal{N}(0,I)$ through conditional normalization, and uses autoregressive rollout to extend horizons. The method trains with the unbiased fair CRPS estimator to align both marginal and joint distributions, achieving fast sampling while preserving fine-scale detail. On the MEPS-lam dataset, CRPS-LAM delivers competitive RMSE and CRPS with ensemble calibration comparable to state-of-the-art methods, while being up to ~39x faster than diffusion-based approaches, supporting parallel ensemble sampling in operational settings. This work offers a practical, scalable path for probabilistic regional forecasts and points to future directions in latent-variable integration and training strategies.
Abstract
Machine learning for weather prediction increasingly relies on ensemble methods to provide probabilistic forecasts. Diffusion-based models have shown strong performance in Limited-Area Modeling (LAM) but remain computationally expensive at sampling time. Building on the success of global weather forecasting models trained based on Continuous Ranked Probability Score (CRPS), we introduce CRPS-LAM, a probabilistic LAM forecasting model trained with a CRPS-based objective. By sampling and injecting a single latent noise vector into the model, CRPS-LAM generates ensemble members in a single forward pass, achieving sampling speeds up to 39 times faster than a diffusion-based model. We evaluate the model on the MEPS regional dataset, where CRPS-LAM matches the low errors of diffusion models. By retaining also fine-scale forecast details, the method stands out as an effective approach for probabilistic regional weather forecasting
