HealDA: Highlighting the importance of initial errors in end-to-end AI weather forecasts

Machine-learning (ML) weather models now rival leading numerical weather prediction (NWP) systems in medium-range skill. However, almost all still rely on NWP data assimilation (DA) to provide initial conditions, tying them to expensive infrastructure and limiting the practical speed and accuracy gains of ML. More recently, ML-based DA systems have been proposed, which are often trained and evaluated end-to-end with a forecast model, making it difficult to assess the quality of their analysis fields. We introduce HealDA, a global ML-based DA system that maps a short window of satellite and conventional observations directly to a 1° atmospheric state on the Hierarchical Equal Area isoLatitude Pixelation (HEALPix) grid, using a smaller sensor suite than operational NWP and no background forecast at runtime. We treat HealDA strictly as a DA module: its analyses are used to initialize off-the-shelf ML forecast models without any fine-tuning of either. For a variety of off-the-shelf ML forecast models, including FourCastNet3 (FCN3), Aurora, and FengWu, HealDA-initialized forecasts lose less than one day of effective lead time when scored against ERA5. HealDA-initialized FCN3 ensembles similarly trail those of the ECMWF Integrated Forecasting System Ensemble (IFS ENS) system by less than 24~h. We find that forecast error growth in these models is largely unchanged from HealDA initialization, and the skill gap primarily arises from the larger initial-condition error of the HealDA analysis. Spectral analysis reveals that this stems from overfitting to the large scales and upper-tropospheric fields. We also demonstrate that small changes in the verification setup can shift apparent skill by 12--24~h, underscoring the need for consistent scoring. Taken together, these results clarify the current performance of ML-based DA systems and show that a relatively simple, background-free network can already provide initial conditions that are usable by state-of-the-art ML forecast models with only modest loss in medium-range skill.

Authors

Aayush Gupta (NVIDIA)
Akshay Subramaniam (NVIDIA)
Michael S. Pritchard (NVIDIA)
Karthik Kashinath (NVIDIA)
Sergey Frolov (NOAA)
Kelsey Lieberman (MITRE Corporation)
Christopher Miller (MITRE Corporation)
Nicholas Silverman (MITRE Corporation)
Noah D. Brenowitz (NVIDIA)

Publication Date

Uploaded Files