English

DAWP: A framework for global observation forecasting via Data Assimilation and Weather Prediction in satellite observation space

Machine Learning 2025-10-21 v1 Artificial Intelligence Atmospheric and Oceanic Physics

Abstract

Weather prediction is a critical task for human society, where impressive progress has been made by training artificial intelligence weather prediction (AIWP) methods with reanalysis data. However, reliance on reanalysis data limits the AIWPs with shortcomings, including data assimilation biases and temporal discrepancies. To liberate AIWPs from the reanalysis data, observation forecasting emerges as a transformative paradigm for weather prediction. One of the key challenges in observation forecasting is learning spatiotemporal dynamics across disparate measurement systems with irregular high-resolution observation data, which constrains the design and prediction of AIWPs. To this end, we propose our DAWP as an innovative framework to enable AIWPs to operate in a complete observation space by initialization with an artificial intelligence data assimilation (AIDA) module. Specifically, our AIDA module applies a mask multi-modality autoencoder(MMAE)for assimilating irregular satellite observation tokens encoded by mask ViT-VAEs. For AIWP, we introduce a spatiotemporal decoupling transformer with cross-regional boundary conditioning (CBC), learning the dynamics in observation space, to enable sub-image-based global observation forecasting. Comprehensive experiments demonstrate that AIDA initialization significantly improves the roll out and efficiency of AIWP. Additionally, we show that DAWP holds promising potential to be applied in global precipitation forecasting.

Keywords

Cite

@article{arxiv.2510.15978,
  title  = {DAWP: A framework for global observation forecasting via Data Assimilation and Weather Prediction in satellite observation space},
  author = {Junchao Gong and Jingyi Xu and Ben Fei and Fenghua Ling and Wenlong Zhang and Kun Chen and Wanghan Xu and Weidong Yang and Xiaokang Yang and Lei Bai},
  journal= {arXiv preprint arXiv:2510.15978},
  year   = {2025}
}