Towards a Self-contained Data-driven Global Weather Forecasting Framework
Yi Xiao, Lei Bai, Wei Xue, Hao Chen, Kun Chen, Kang Chen, Tao Han, Wanli Ouyang
Abstract
Data-driven weather forecasting models are advancing rapidly, yet they rely on initial states (i.e., analysis states) typically produced by traditional data assimilation algorithms. Four-dimensional variational assimilation (4DVar) is one of the most widely adopted data assimilation algorithms in numerical weather prediction centers; it is accurate but computationally expensive. In this paper, we aim to couple the AI forecasting model, FengWu, with 4DVar to build a self-contained data-driven global weather forecasting framework, FengWu-4DVar. To achieve this, we propose an AI-embedded 4DVar algorithm that includes three components: (1) a 4DVar objective function embedded with the FengWu forecasting model and its error representation to enhance efficiency and accuracy; (2) a spherical-harmonictransform-based (SHT-based) approximation strategy for capturing the horizontal correlation of background error; and (3) an auto-differentiation (AD) scheme for determining the optimal analysis fields. Experimental results show that under the ERA5 simulated observational data with varying proportions and noise levels, FengWu-4DVar can generate accurate analysis fields; remarkably, it has achieved stable self-contained global weather forecasts for an entire year for the first time, demonstrating its potential for realworld applications. Additionally, our framework is approximately 100 times faster than the traditional 4DVar algorithm under similar experimental conditions, highlighting its significant computational efficiency.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers8
- Generalizing Weather Forecast to Fine-grained Temporal Scales via Physics-AI Hybrid ModelingWanghan Xu, Fenghua Ling, Wenlong Zhang, Tao Han et al.NeurIPS 2024 · 31 citations
- MeteorPred: A Meteorological Multimodal Large Model and Dataset for Severe Weather Event PredictionShuo Tang, Jian Xu, Jiadong Zhang, Yi Chen et al.CVPR 2026 · 3 citations
- DAWP: A framework for global observation forecasting via Data Assimilation and Weather Prediction in satellite observation spaceJunchao Gong, Jingyi Xu, Ben Fei, Fenghua Ling et al.NeurIPS 2025 · 2 citations
- Task-Adaptive Parameter-Efficient Fine-Tuning for Weather Foundation ModelsShilei Cao, Hehai Lin, Jiashun Cheng, Yang Liu et al.ICLR 2026 · 2 citations
- Synthetic Series-Symbol Data Generation for Time Series Foundation ModelsWenxuan Wang, Kai Wu, Yujian Betterest Li, Dan Wang et al.NeurIPS 2025 · 1 citation
Builds on3
- Swin Transformer: Hierarchical Vision Transformer using Shifted WindowsZe Liu, Yutong Lin, Yue Cao, Han Hu et al.ICCV 2021 · 31,683 citations
- Spherical Fourier Neural Operators: Learning Stable Dynamics on the SphereBoris Bonev, Thorsten Kurth, Christian Hundt, Jaideep Pathak et al.ICML 2023 · 280 citations
- Variational Data Assimilation with a Learned Inverse Observation OperatorThomas Frerix, Dmitrii Kochkov, Jamie A. Smith, Daniel Cremers et al.ICML 2021 · 42 citations
Related papers
- VAE-Var: Variational Autoencoder-Enhanced Variational Methods for Data Assimilation in MeteorologyYi Xiao, Qilong Jia, Kun Chen, Lei Bai et al.ICLR 2025
- EWMoE: An Effective Model for Global Weather Forecasting with Mixture-of-ExpertsLihao Gan, Xin Man, Chenghong Zhang, Jie ShaoAAAI 2025 · 10 citations
- Rapid simulations of atmospheric data assimilation of hourly-scale phenomena with modern neural networksYiyuan Li, Xiting Ju, Yi Xiao, Qilong Jia et al.SC 2023 · 4 citations
- FNP: Fourier Neural Processes for Arbitrary-Resolution Data AssimilationKun Chen, Peng Ye, Hao Chen, Kang Chen et al.NeurIPS 2024 · 15 citations
- Tensor-Var: Efficient Four-Dimensional Variational Data AssimilationYiming Yang, Xiaoyuan Cheng, Daniel Giles, Sibo Cheng et al.ICML 2025
