U-Cast: A Surprisingly Simple and Efficient Frontier Probabilistic AI Weather Forecaster
About
AI-based weather forecasting now rivals traditional physics-based ensembles, but state-of-the-art (SOTA) models rely on specialized architectures and massive computational budgets, creating a high barrier to entry. We demonstrate that such complexity is unnecessary for frontier performance. We introduce \ours, a probabilistic forecaster built on a standard U-Net backbone trained with a simple recipe: deterministic pre-training on Mean Absolute Error followed by short probabilistic fine-tuning on the Continuous Ranked Probability Score (CRPS) using Monte Carlo Dropout for stochasticity. As a result, our model matches or exceeds the probabilistic skill of GenCast and IFS ENS at $1.5^\circ$ resolution while reducing training compute by over $10\times$ compared to leading CRPS-based models and inference latency by over $10\times$ compared to diffusion-based models. U-Cast trains in under 12 H200 GPU-days and generates a 15-day ensemble forecast in 3 seconds. These results suggest that scalable, general-purpose architectures paired with efficient training curricula can match complex domain-specific designs at a fraction of the cost, opening the training of frontier probabilistic weather models to the broader community.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Geopotential at 500hPa forecasting | WeatherBench 2 1.5° resolution 2020 (test) | CRPS (1d)19.6 | 6 | |
| Probabilistic Weather Forecasting | Weather Forecasting Coarser Resolution ~1.5° | Training Cost (days)8.2 | 6 | |
| 10m u-component of wind forecasting | WeatherBench 1.5° resolution 2 2020 (test) | CRPS (1d)0.345 | 5 | |
| Probabilistic Weather Forecasting | Weather Forecasting High Resolution ~0.25° | -- | 5 | |
| Probabilistic Weather Forecasting | Weather Forecasting Mid-resolution ~1° | Training Cost (days)15 | 4 |