U-Cast: A Surprisingly Simple and Efficient Frontier Probabilistic AI Weather Forecaster
Salva Ruhling Cachay, Duncan Watson-Parris, Rose Yu
摘要
AI-based weather forecasting now rivals traditional physics-based ensembles, but state-of-theart (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 U-Cast, a probabilistic forecaster built on a standard U-Net backbone trained with a simple recipe: deterministic pretraining 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 ˝resolution while reducing training compute by over 10ˆcompared to leading CRPS-based models and inference latency by over 10ˆcompared to diffusion-based models. U-Cast trains in under 12 H200 GPUdays 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 domainspecific designs at a fraction of the cost, opening the training of frontier probabilistic weather models to the broader community. Our code is available at: https://github.com/Rose-STL-Lab/u-cast .
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper12
- Diffusion Models Beat GANs on Image SynthesisPrafulla Dhariwal, Alexander Quinn NicholNeurIPS 2021 · 被引用 13,211 次
- Elucidating the Design Space of Diffusion-Based Generative ModelsTero Karras, Miika Aittala, Timo Aila, Samuli LaineNeurIPS 2022 · 被引用 3,959 次
- Spherical Fourier Neural Operators: Learning Stable Dynamics on the SphereBoris Bonev, Thorsten Kurth, Christian Hundt, Jaideep Pathak 等ICML 2023 · 被引用 280 次
- Scaling transformer neural networks for skillful and reliable medium-range weather forecastingTung Nguyen, Rohan Shah, Hritik Bansal, Troy Arcomano 等NeurIPS 2024 · 被引用 165 次
- Uncertainty Quantification over Graph with Conformalized Graph Neural NetworksKexin Huang, Ying Jin, Emmanuel J. Candès, Jure LeskovecNeurIPS 2023 · 被引用 124 次
相关 Paper
- Continuous Ensemble Weather Forecasting with Diffusion modelsMartin Andrae, Tomas Landelius, Joel Oskarsson, Fredrik LindstenICLR 2025
- Estimating Epistemic and Aleatoric Uncertainty with a Single ModelMatthew Chan, Maria Molina, Chris MetzlerNeurIPS 2024 · 被引用 76 次
- CasCast: Skillful High-resolution Precipitation Nowcasting via Cascaded ModellingJunchao Gong, Lei Bai, Peng Ye, Wanghan Xu 等ICML 2024 · 被引用 53 次
- OmniCast: A Masked Latent Diffusion Model for Weather Forecasting Across Time ScalesTung Nguyen, Tuan Pham, Troy Arcomano, Rao Kotamarthi 等NeurIPS 2025 · 被引用 12 次
- Fixing the Double Penalty in Data-Driven Weather Forecasting Through a Modified Spherical Harmonic Loss FunctionChristopher Subich, Syed Zahid Husain, Leo Separovic, Jing YangICML 2025
