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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gupta, Aayush, Subramaniam, Akshay, Pritchard, Michael S., Kashinath, Karthik, Frolov, Sergey, Lieberman, Kelsey, Miller, Christopher, Silverman, Nicholas, Brenowitz, Noah D.
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915990309175296
author Gupta, Aayush
Subramaniam, Akshay
Pritchard, Michael S.
Kashinath, Karthik
Frolov, Sergey
Lieberman, Kelsey
Miller, Christopher
Silverman, Nicholas
Brenowitz, Noah D.
author_facet Gupta, Aayush
Subramaniam, Akshay
Pritchard, Michael S.
Kashinath, Karthik
Frolov, Sergey
Lieberman, Kelsey
Miller, Christopher
Silverman, Nicholas
Brenowitz, Noah D.
contents AI 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 HEALPix grid, using a smaller sensor suite than operational NWP. 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 IFS ENS system by < 24 h. We find that forecast error growth in these models is unchanged from HealDA initialization, and the skill gap primarily arises from the larger initial 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--24h, 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, direct observation-to-state 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.
format Preprint
id arxiv_https___arxiv_org_abs_2601_17636
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle HealDA: Highlighting the importance of initial errors in end-to-end AI weather forecasts
Gupta, Aayush
Subramaniam, Akshay
Pritchard, Michael S.
Kashinath, Karthik
Frolov, Sergey
Lieberman, Kelsey
Miller, Christopher
Silverman, Nicholas
Brenowitz, Noah D.
Atmospheric and Oceanic Physics
AI 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 HEALPix grid, using a smaller sensor suite than operational NWP. 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 IFS ENS system by < 24 h. We find that forecast error growth in these models is unchanged from HealDA initialization, and the skill gap primarily arises from the larger initial 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--24h, 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, direct observation-to-state 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.
title HealDA: Highlighting the importance of initial errors in end-to-end AI weather forecasts
topic Atmospheric and Oceanic Physics
url https://arxiv.org/abs/2601.17636