Probabilistic Emulation of a Global Climate Model with Spherical DYffusion

• DOI: 10.48550/arxiv.2406.14798
• Publication Date: 2024-01-01
• Authors: Cachay, Salva Rüh...; Henn, Brian; Watt-Meyer, Oliver; Bretherton, Chris...; Yu, Rose
• Keywords: FOS: Computer and information sciences; Computer Science - Machine Learning; Physics - Atmospheric and Oceanic Physics; Artificial Intelligence (cs.AI); Computer Science - Artificial Intelligence; Statistics - Machine Learning; Atmospheric and Oceanic Physics (physics.ao-ph); FOS: Physical sciences; Machine Learning (stat.ML); Machine Learning (cs.LG)

NeurIPS 2024; Code is available at https://github.com/Rose-STL-Lab/spherical-dyffusion<br/>Data-driven deep learning models are transforming global weather forecasting. It is an open question if this success can extend to climate modeling, where the complexity of the data and long inference rollouts pose significant challenges. Here, we present the first conditional generative model that produces accurate and physically consistent global climate ensemble simulations by emulating a coarse version of the United States' primary operational global forecast model, FV3GFS. Our model integrates the dynamics-informed diffusion framework (DYffusion) with the Spherical Fourier Neural Operator (SFNO) architecture, enabling stable 100-year simulations at 6-hourly timesteps while maintaining low computational overhead compared to single-step deterministic baselines. The model achieves near gold-standard performance for climate model emulation, outperforming existing approaches and demonstrating promising ensemble skill. This work represents a significant advance towards efficient, data-driven climate simulations that can enhance our understanding of the climate system and inform adaptation strategies.<br/>