GNN-Surrogate: A Hierarchical and Adaptive Graph Neural Network for Parameter Space Exploration of Unstructured-Mesh Ocean Simulations
Authors:
Neng Shi,
Jiayi Xu,
Skylar W. Wurster,
Hanqi Guo,
Jonathan Woodring,
Luke P. Van Roekel,
Han-Wei Shen
Abstract:
We propose GNN-Surrogate, a graph neural network-based surrogate model to explore the parameter space of ocean climate simulations. Parameter space exploration is important for domain scientists to understand the influence of input parameters (e.g., wind stress) on the simulation output (e.g., temperature). The exploration requires scientists to exhaust the complicated parameter space by running a…
▽ More
We propose GNN-Surrogate, a graph neural network-based surrogate model to explore the parameter space of ocean climate simulations. Parameter space exploration is important for domain scientists to understand the influence of input parameters (e.g., wind stress) on the simulation output (e.g., temperature). The exploration requires scientists to exhaust the complicated parameter space by running a batch of computationally expensive simulations. Our approach improves the efficiency of parameter space exploration with a surrogate model that predicts the simulation outputs accurately and efficiently. Specifically, GNN-Surrogate predicts the output field with given simulation parameters so scientists can explore the simulation parameter space with visualizations from user-specified visual mappings. Moreover, our graph-based techniques are designed for unstructured meshes, making the exploration of simulation outputs on irregular grids efficient. For efficient training, we generate hierarchical graphs and use adaptive resolutions. We give quantitative and qualitative evaluations on the MPAS-Ocean simulation to demonstrate the effectiveness and efficiency of GNN-Surrogate. Source code is publicly available at https://github.com/trainsn/GNN-Surrogate.
△ Less
Submitted 21 February, 2022; v1 submitted 17 February, 2022;
originally announced February 2022.
The KPP boundary layer scheme: revisiting its formulation and benchmarking one-dimensional ocean simulations relative to LES
Authors:
Luke P Van Roekel,
Alistair J. Adcroft,
Gokhan Danabasoglu,
Stephen M. Griffies,
Brian Kauffman,
William Large,
Michael Levy,
Brandon Reichl,
Todd Ringler,
Martin Schmidt
Abstract:
We evaluate the Community ocean Vertical Mixing (CVMix) project version of the K-profile parameterization (KPP). For this purpose, one-dimensional KPP simulations are compared across a suite of oceanographically relevant regimes against large eddy simulations (LES). The LES is forced with horizontally uniform boundary fluxes and has horizontally uniform initial conditions, allowing its horizontal…
▽ More
We evaluate the Community ocean Vertical Mixing (CVMix) project version of the K-profile parameterization (KPP). For this purpose, one-dimensional KPP simulations are compared across a suite of oceanographically relevant regimes against large eddy simulations (LES). The LES is forced with horizontally uniform boundary fluxes and has horizontally uniform initial conditions, allowing its horizontal average to be compared to one-dimensional KPP tests. We find the standard configuration of KPP consistent with LES across many forcing regimes, supporting the physical basis of KPP. Our evaluation motivates recommendations for "best practices" for using KPP within ocean circulation models, and identifies areas where further research is warranted. Further, our test suite can be used as a baseline for evaluation of a broad suite of boundary layer models. The original treatment of KPP recommends the matching of interior diffusivities and their gradients to the KPP predicted values computed in the ocean surface boundary layer (OSBL). However, we find that difficulties in representing derivatives of rapidly changing diffusivities near the base of the OSBL can lead to loss of simulation fidelity. We propose two alternative approaches. We find the KPP entrainment buoyancy flux to be sensitive to vertical grid resolution and details of how to diagnose the KPP boundary layer depth. We modify the KPP turbulent shear velocity parameterization to reduce resolution dependence. Additionally, our results show that the KPP parameterized non-local tracer flux is incomplete due to the assumption that it solely redistributes the surface tracer flux. However, examination of the LES vertical turbulent scalar flux budgets show that non-local fluxes can exist in the absence of surface tracer fluxes. This result motivates further studies of the non-local flux parameterization.
△ Less
Submitted 8 March, 2018; v1 submitted 6 October, 2017;
originally announced October 2017.
