English

Swift: An Autoregressive Consistency Model for Efficient Weather Forecasting

Machine Learning 2025-10-01 v1

Abstract

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×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.

Keywords

Cite

@article{arxiv.2509.25631,
  title  = {Swift: An Autoregressive Consistency Model for Efficient Weather Forecasting},
  author = {Jason Stock and Troy Arcomano and Rao Kotamarthi},
  journal= {arXiv preprint arXiv:2509.25631},
  year   = {2025}
}

Comments

17 pages and 15 figures

R2 v1 2026-07-01T06:06:31.893Z