Huge Ensembles Part II: Properties of a Huge Ensemble of Hindcasts Generated with Spherical Fourier Neural Operators

πŸ“… 2024-08-02
πŸ›οΈ arXiv.org
πŸ“ˆ Citations: 1
✨ Influential: 0
πŸ“„ PDF
πŸ€– AI Summary
This work addresses three key challenges in extreme weather forecasting: insufficient sampling of tail distributions, biased uncertainty quantification, and inadequate representation of internal variability. To this end, we propose HENSβ€”a hyper-scale ensemble forecasting framework comprising 7,424 members. Methodologically, HENS is the first to integrate spherical Fourier neural operators (SFNO), bred-vector initial-condition perturbations, and multi-checkpoint model-parameter perturbations on a ten-million-node spherical grid, enabling a high-fidelity, low-overhead parallel ensemble generation system. Compared with operational systems, HENS achieves comparable physical consistency while reducing computational cost by several orders of magnitude. It significantly improves tail-sampling accuracy for 4Οƒ extreme events, enhances single-member skill and trajectory coverage, and reduces ensemble outlier rates. Moreover, HENS provides a more comprehensive characterization of internal variability, concurrently improving both grid-scale forecast coverage and reliability.

Technology Category

Application Category

πŸ“ Abstract
In Part I, we created an ensemble based on Spherical Fourier Neural Operators. As initial condition perturbations, we used bred vectors, and as model perturbations, we used multiple checkpoints trained independently from scratch. Based on diagnostics that assess the ensemble's physical fidelity, our ensemble has comparable performance to operational weather forecasting systems. However, it requires several orders of magnitude fewer computational resources. Here in Part II, we generate a huge ensemble (HENS), with 7,424 members initialized each day of summer 2023. We enumerate the technical requirements for running huge ensembles at this scale. HENS precisely samples the tails of the forecast distribution and presents a detailed sampling of internal variability. For extreme climate statistics, HENS samples events 4$sigma$ away from the ensemble mean. At each grid cell, HENS improves the skill of the most accurate ensemble member and enhances coverage of possible future trajectories. As a weather forecasting model, HENS issues extreme weather forecasts with better uncertainty quantification. It also reduces the probability of outlier events, in which the verification value lies outside the ensemble forecast distribution.
Problem

Research questions and friction points this paper is trying to address.

Evaluates huge ensemble forecasting performance
Samples extreme climate statistics accurately
Improves weather forecast uncertainty quantification
Innovation

Methods, ideas, or system contributions that make the work stand out.

Spherical Fourier Neural Operators
Bred vectors initialization
Large-scale ensemble diagnostics
πŸ”Ž Similar Papers
No similar papers found.
Ankur Mahesh
Ankur Mahesh
University of California, Berkeley, and Lawrence Berkeley National Lab
Climate DynamicsMachine LearningAtmospheric Rivers
W
William Collins
Earth and Environmental Sciences Area, Lawrence Berkeley National Laboratory (LBNL), Berkeley, California, USA; Department of Earth and Planetary Science, University of California, Berkeley, USA
B
B. Bonev
NVIDIA Corporation, Santa Clara, California, USA
N
Noah D. Brenowitz
NVIDIA Corporation, Santa Clara, California, USA
Y
Y. Cohen
NVIDIA Corporation, Santa Clara, California, USA
Peter Harrington
Peter Harrington
NVIDIA
Deep learningartificial intelligence
K
K. Kashinath
NVIDIA Corporation, Santa Clara, California, USA
T
Thorsten Kurth
NVIDIA Corporation, Santa Clara, California, USA
J
Joshua North
Earth and Environmental Sciences Area, Lawrence Berkeley National Laboratory (LBNL), Berkeley, California, USA
T
Travis OBrien
Department of Earth and Atmospheric Sciences, Indiana University, Bloomington, Indiana, USA; Earth and Environmental Sciences Area, Lawrence Berkeley National Laboratory (LBNL), Berkeley, California, USA
Michael S. Pritchard
Michael S. Pritchard
Director of Climate Simulation Research, NVIDIA Research
Climate sciencecloud physicscloud superparameterizationmachine learninghigh performance computing
D
David Pruitt
NVIDIA Corporation, Santa Clara, California, USA
M
Mark Risser
Earth and Environmental Sciences Area, Lawrence Berkeley National Laboratory (LBNL), Berkeley, California, USA
Shashank Subramanian
Shashank Subramanian
Lawrence Berkeley National Laboratory
scientific machine learninglarge-scale optimizationinverse problemshigh-performance computing
J
Jared D. Willard
National Energy Research Scientific Computing Center (NERSC), LBNL, Berkeley, California, USA