-
MITgcm-AD v2: Open source tangent linear and adjoint modeling framework for the oceans and atmosphere enabled by the Automatic Differentiation tool Tapenade
Authors:
Shreyas Sunil Gaikwad,
Sri Hari Krishna Narayanan,
Laurent Hascoet,
Jean-Michel Campin,
Helen Pillar,
An Nguyen,
Jan Hückelheim,
Paul Hovland,
Patrick Heimbach
Abstract:
The Massachusetts Institute of Technology General Circulation Model (MITgcm) is widely used by the climate science community to simulate planetary atmosphere and ocean circulations. A defining feature of the MITgcm is that it has been developed to be compatible with an algorithmic differentiation (AD) tool, TAF, enabling the generation of tangent-linear and adjoint models. These provide gradient i…
▽ More
The Massachusetts Institute of Technology General Circulation Model (MITgcm) is widely used by the climate science community to simulate planetary atmosphere and ocean circulations. A defining feature of the MITgcm is that it has been developed to be compatible with an algorithmic differentiation (AD) tool, TAF, enabling the generation of tangent-linear and adjoint models. These provide gradient information which enables dynamics-based sensitivity and attribution studies, state and parameter estimation, and rigorous uncertainty quantification. Importantly, gradient information is essential for computing comprehensive sensitivities and performing efficient large-scale data assimilation, ensuring that observations collected from satellites and in-situ measuring instruments can be effectively used to optimize a large uncertain control space. As a result, the MITgcm forms the dynamical core of a key data assimilation product employed by the physical oceanography research community: Estimating the Circulation and Climate of the Ocean (ECCO) state estimate. Although MITgcm and ECCO are used extensively within the research community, the AD tool TAF is proprietary and hence inaccessible to a large proportion of these users. The new version 2 (MITgcm-AD v2) framework introduced here is based on the source-to-source AD tool Tapenade, which has recently been open-sourced. Another feature of Tapenade is that it stores required variables by default (instead of recomputing them) which simplifies the implementation of efficient, AD-compatible code. The framework has been integrated with the MITgcm model main branch and is now freely available.
△ Less
Submitted 22 January, 2024;
originally announced January 2024.
-
Oceananigans.jl: A model that achieves breakthrough resolution, memory and energy efficiency in global ocean simulations
Authors:
Simone Silvestri,
Gregory Wagner,
Christopher Hill,
Matin Raayai Ardakani,
Johannes Blaschke,
Jean-Michel Campin,
Valentin Churavy,
Navid Constantinou,
Alan Edelman,
John Marshall,
Ali Ramadhan,
Andre Souza,
Raffaele Ferrari
Abstract:
Climate models must simulate hundreds of future scenarios for hundreds of years at coarse resolutions, and a handful of high-resolution decadal simulations to resolve localized extreme events. Using Oceananigans.jl, written from scratch in Julia, we report several achievements: First, a global ocean simulation with breakthrough horizontal resolution -- 488m -- reaching 15 simulated days per day (0…
▽ More
Climate models must simulate hundreds of future scenarios for hundreds of years at coarse resolutions, and a handful of high-resolution decadal simulations to resolve localized extreme events. Using Oceananigans.jl, written from scratch in Julia, we report several achievements: First, a global ocean simulation with breakthrough horizontal resolution -- 488m -- reaching 15 simulated days per day (0.04 simulated years per day; SYPD). Second, Oceananigans simulates the global ocean at 488m with breakthrough memory efficiency on just 768 Nvidia A100 GPUs, a fraction of the resources available on current and upcoming exascale supercomputers. Third, and arguably most significant for climate modeling, Oceananigans achieves breakthrough energy efficiency reaching 0.95 SYPD at 1.7 km on 576 A100s and 9.9 SYPD at 10 km on 68 A100s -- the latter representing the highest horizontal resolutions employed by current IPCC-class ocean models. Routine climate simulations with 10 km ocean components are within reach.
△ Less
Submitted 12 September, 2023;
originally announced September 2023.
-
Formulation and calibration of CATKE, a one-equation parameterization for microscale ocean mixing
Authors:
Gregory LeClaire Wagner,
Adeline Hillier,
Navid C. Constantinou,
Simone Silvestri,
Andre Souza,
Keaton Burns,
Chris Hill,
Jean-Michel Campin,
John Marshall,
Raffaele Ferrari
Abstract:
We describe CATKE, a parameterization for fluxes associated with small-scale or "microscale" ocean turbulent mixing on scales between 1 and 100 meters. CATKE uses a downgradient formulation that depends on a prognostic turbulent kinetic energy (TKE) variable and a diagnostic mixing length scale that includes a dynamic convective adjustment (CA) component. With its dynamic convective mixing length,…
▽ More
We describe CATKE, a parameterization for fluxes associated with small-scale or "microscale" ocean turbulent mixing on scales between 1 and 100 meters. CATKE uses a downgradient formulation that depends on a prognostic turbulent kinetic energy (TKE) variable and a diagnostic mixing length scale that includes a dynamic convective adjustment (CA) component. With its dynamic convective mixing length, CATKE predicts not just the depth spanned by convective plumes but also the characteristic convective mixing timescale, an important aspect of turbulent convection not captured by simpler static convective adjustment schemes. As a result, CATKE can describe the competition between convection and other processes such as shear-driven mixing and baroclinic restratification. To calibrate CATKE, we use Ensemble Kalman Inversion to minimize the error between 21 large eddy simulations (LES) and predictions of the LES data by CATKE-parameterized single column simulations at three different vertical resolutions. We find that CATKE makes accurate predictions of both idealized and realistic LES compared to microscale turbulence parameterizations commonly used in climate models.
△ Less
Submitted 22 June, 2024; v1 submitted 22 June, 2023;
originally announced June 2023.
-
How does salinity shape ocean circulation and ice geometry on Enceladus and other icy satellites?
Authors:
Wanying Kang,
Tushar Mittal,
Suyash Bire,
Jean-Michel Campin,
John Marshall
Abstract:
Of profound astrobiological interest, Enceladus appears to have a global subsurface ocean that is salty, indicating water-rock reaction at present or in the past, important for its habitability. Here, we investigate how salinity and the partition of heat production between the silicate core and the ice shell affect ocean dynamics and the associated heat transport -- a key factor that determines th…
▽ More
Of profound astrobiological interest, Enceladus appears to have a global subsurface ocean that is salty, indicating water-rock reaction at present or in the past, important for its habitability. Here, we investigate how salinity and the partition of heat production between the silicate core and the ice shell affect ocean dynamics and the associated heat transport -- a key factor that determines the equilibrium ice shell geometry. Assuming steady state conditions, we show that the meridional overturning circulation of the ocean, driven by heat and salt exchange with the ice, has opposing signs at very low and very high salinities. Regardless of these differing circulations, heat and freshwater converge towards the equator, where the ice is thick, acting to homogenize thickness variations. In order to maintain the observed ice thickness variation, the polar-amplified ice dissipation needs to be strong enough and ocean heat convergence cannot overwhelm well-constrained heat loss rates through the thick equatorial ice sheet. This requirement is found violated if the main heat source is in the core rather than the ice shell, or if the ocean is very fresh or very salty. Instead, with a salinity of intermediate range, the temperature- and salinity-induced density gradient largely cancel one another, leading to much reduced overturning and equatorial heat convergence rates and consistent budgets in appearance of a significant ice dissipation.
△ Less
Submitted 12 May, 2022; v1 submitted 14 April, 2021;
originally announced April 2021.
-
Capturing missing physics in climate model parameterizations using neural differential equations
Authors:
Ali Ramadhan,
John Marshall,
Andre Souza,
Xin Kai Lee,
Ulyana Piterbarg,
Adeline Hillier,
Gregory LeClaire Wagner,
Christopher Rackauckas,
Chris Hill,
Jean-Michel Campin,
Raffaele Ferrari
Abstract:
We explore how neural differential equations (NDEs) may be trained on highly resolved fluid-dynamical models of unresolved scales providing an ideal framework for data-driven parameterizations in climate models. NDEs overcome some of the limitations of traditional neural networks (NNs) in fluid dynamical applications in that they can readily incorporate conservation laws and boundary conditions an…
▽ More
We explore how neural differential equations (NDEs) may be trained on highly resolved fluid-dynamical models of unresolved scales providing an ideal framework for data-driven parameterizations in climate models. NDEs overcome some of the limitations of traditional neural networks (NNs) in fluid dynamical applications in that they can readily incorporate conservation laws and boundary conditions and are stable when integrated over time. We advocate a method that employs a 'residual' approach, in which the NN is used to improve upon an existing parameterization through the representation of residual fluxes which are not captured by the base parameterization. This reduces the amount of training required and providing a method for capturing up-gradient and nonlocal fluxes. As an illustrative example, we consider the parameterization of free convection of the oceanic boundary layer triggered by buoyancy loss at the surface. We demonstrate that a simple parameterization of the process - convective adjustment - can be improved upon by training a NDE against highly resolved explicit models, to capture entrainment fluxes at the base of the well-mixed layer, fluxes that convective adjustment itself cannot represent. The augmented parameterization outperforms existing commonly used parameterizations such as the K-Profile Parameterization (KPP). We showcase that the NDE performs well independent of the time-stepper and that an online training approach using differentiable simulation via the Julia scientific machine learning software stack improves accuracy by an order-of-magnitude. We conclude that NDEs provide an exciting route forward to the development of representations of sub-grid-scale processes for climate science, opening up myriad new opportunities.
△ Less
Submitted 6 March, 2023; v1 submitted 23 October, 2020;
originally announced October 2020.
-
Differing Enceladean ocean circulation and ice shell geometries driven by tidal heating in the ice versus the core
Authors:
Wanying Kang,
Suyash Bire,
Jean-Michel Campin,
Christophe Sotin,
Christopher German,
Andreas Thurnherr,
John Marshall
Abstract:
Beneath the icy shell encasing Enceladus, a small icy moon of Saturn, a global ocean of liquid water ejects geyser-like plumes into space through fissures in the ice, making it an attractive place to investigate habitability and to search for extraterrestrial life. The existence of an ocean on Enceladus has been attributed to the heat generated in dissipative processes associated with deformation…
▽ More
Beneath the icy shell encasing Enceladus, a small icy moon of Saturn, a global ocean of liquid water ejects geyser-like plumes into space through fissures in the ice, making it an attractive place to investigate habitability and to search for extraterrestrial life. The existence of an ocean on Enceladus has been attributed to the heat generated in dissipative processes associated with deformation by tidal forcing. However, it remains unclear whether that heat is mostly generated in its ice shell or silicate core. Answering this question is crucial if we are to unravel patterns of ocean circulation and tracer transport that will impact both the habitability of Enceladus and our ability to interpret putative evidence of any habitability and/or life. Using a nonhydrostatic ocean circulation model, we describe and contrast the differing circulation patterns and implied ice shell geometries to be expected as a result of heating in the ice shell above and heating in the core below Enceladus' ocean layer. If heat is generated primarily in the silicate core we would predict enhanced melting rates at the equator. In contrast, if heat is primarily generated in the ice shell we would infer a poleward-thinning ice geometry consistent with Cassini Mission observations.
△ Less
Submitted 9 August, 2020;
originally announced August 2020.