Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Swift: An Autoregressive Consistency Model for Efficient Weather Forecasting

About

Diffusion models offer a physically grounded framework for probabilistic weather forecasting, but their typical reliance on slow, iterative solvers during inference makes them impractical for subseasonal-to-seasonal (S2S) applications where long lead-times and domain-driven calibration are essential. To address this, we introduce Swift, a single-step consistency model that, for the first time, enables autoregressive finetuning of a probability flow model with a continuous ranked probability score (CRPS) objective. This eliminates the need for multi-model ensembling or parameter perturbations. Results show that Swift produces skillful 6-hourly forecasts that remain stable for up to 75 days, running $39\times$ faster than state-of-the-art diffusion baselines while achieving forecast skill competitive with the numerical-based, operational IFS ENS. This marks a step toward efficient and reliable ensemble forecasting from medium-range to seasonal-scales.

Jason Stock, Troy Arcomano, Rao Kotamarthi• 2025

Related benchmarks

TaskDatasetResultRank
Probabilistic Weather ForecastingWeather Forecasting Coarser Resolution ~1.5°
Training Cost (days)360
6
Weather forecastingERA5 6-hour lead time
U10 Error (6h Lead)0.85
6
Weather forecastingERA5 7-day lead time
U-Wind (10m) Error5.14
6
Weather forecastingERA5 10-day lead time
U10 Error (10m U-wind)5.24
6
Weather forecastingERA5 1-day lead time
U101.46
6
Weather forecastingERA5 3-Day lead time
RMSE (u10)2.81
6
Showing 6 of 6 rows

Other info

Follow for update