Uncertainty-Aware End-to-End AI Weather Forecasting: Disentangling Observation and Model Contributions
Not on the map. Why?
Abstract (arXiv)
End-to-end weather forecasting systems produce skillful global gridded and station forecasts directly from raw Earth observations, replacing the numerical weather prediction pipeline, including data assimilation, at a fraction of its cost. These systems are deterministic and issue no uncertainty. Here we render the Aardvark Weather model probabilistic by attaching one stochastic mechanism to each component: learned, input-dependent noise at the observation encoder, capturing aleatoric uncertainty inherited from the observing system, and Monte Carlo dropout in the processor, capturing epistemic uncertainty in the learned dynamics. The resulting nested ensemble attributes forecast spread to the two sources through a law-of-total-variance decomposition, cross-checked by withholding observation streams. Probabilistic finetuning significantly improves the mean forecast, by 4.2% on average across variables and lead times. The ensemble is calibrated against ERA5 through the medium range (spread-skill ratio 0.98), keeps station RMSE within 2.4% of the deterministic model while beating it in CRPS at every lead time, and trails the operational ECMWF ensemble. The encoder branch behaves as observation-driven uncertainty. Component-attributed uncertainty makes end-to-end forecasts more transparent, a step toward observation-driven digital twins of the atmosphere.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 18 of its 62 references that are in the map. First posted on arXiv 31 Aug 2026.
Builds on
18 of 62 references.
References · 18
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,352
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
- 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 913
- 2020A Review of Uncertainty Quantification in Deep Learning: Techniques, Applications and Challengescited by 819
- 2022FourCastNet: A Global Data-driven High-resolution Weather Model using Adaptive Fourier Neural Operatorscited by 776
- 2023WeatherBench 2: A benchmark for the next generation of data-driven global weather modelscited by 236
- 2018Understanding Measures of Uncertainty for Adversarial Example Detectioncited by 177
- 2022A Deeper Look into Aleatoric and Epistemic Uncertainty Disentanglementcited by 87
6 more
- 2024GraphDOP: Towards skilful data-driven medium-range weather forecasts learnt and initialised directly from observationscited by 27
- 2024Uncertainty quantification for data-driven weather modelscited by 26
- 2024Explaining Predictive Uncertainty by Exposing Second-Order Effectscited by 10
- 2025Benchmarking Uncertainty and its Disentanglement in multi-label Chest X-Ray Classificationcited by 4
- 2025On the Predictive Skill of Artificial Intelligence-based Weather Models for Extreme Events using Uncertainty Quantificationcited by 1
- 2026Beyond Boundary Noise: Aggregated Aleatoric Uncertainty Fails to Capture Presence Ambiguity in 3D Lung Nodule Segmentationcited by 1
Built on by
Not cited yet.