Dept. of Geosciences Colloquium: Combining physics and machine learning in the turbulenceconvection parameterization of the CliMA climate model
Dr. Yair Cohen, Nvidia Research (formerly Caltech)
Abstract:
The representation of turbulence and convection at the subgrid scale of climate models by various parameterization schemes is one of the largest sources of model uncertainty in long term climate predictions. The crux of these parameterization schemes is the formulation of closures for processes that cannot be observed or simulated in high resolution models. In this talk I will present a hybrid approach that combines a physics based parameterization scheme with machine learning (ML) closures. Such a hybrid approach cannot be trained using backpropagation which requires partial derivatives of the physics+ML setup in parameter space. Instead we use a gradient free (ensemble) method to train from high resolution LES data. The hybrid approach ensures conservation properties, allows for straightforward interpretability of its data driven components and reduces the degrees of freedom to allow us to train from spares data.
The physical model is based on the extended Eddy Diffusivity/Mass Flux (EDMF) scheme, derived by a systematic coarse-graining of the equations of motion and includes subgrid scale memory and prognostic equations for first and second moments. The closures for this scheme combine physical arguments with nondimensional functional forms that can be learned from data using various ML architectures. Our parameterization, in a single column of a climate models, reproduces well the corresponding LES of the full spectrum of boundary layer and turbulent motions: from polar boundary layers (vertical scale of 300m), through the stratocumulus-topped boundary layer (vertical scale of 1km), shallow convection (vertical scale of 3km), and deep convection (vertical scale of 15km). Furthermore, using ML closures we show that a hybrid model is able to extrapolate by training on current climate simulations to a validation set from 4K climate warming scenario.
Event Organizer: Dr. Roy Barkan