Generalizable neural-network parameterization of mesoscale eddies in idealized and global ocean models [Dataset]

This repository contains the following data:

• factor-*.zip Is the training data generated from the global ocean model CM2.6, where factor value stands for the coarsening ratio between high-resolution and coarse grids
• Data required to plot Figure 1 is in offline-skill-Figure-1.zip
• Online metrics required to reproduce Figures 2,3 and 4 are in online-simulation-metrics.zip
• NW2-*.zip is raw simulation data in the idealized configuration NeverWorld 2
• OM4-*.zip is raw simulation data in the global configuration OM4

Note: the additional OM4 simulation (OM4-ZB20.zip) appears only in a subsequent paper, "Hybrid AI Coupled Sea-Ice Ocean Climate Models", while OM4-ANN.zip and OM4-Control.zip are the same in both papers.

Additional data required to plot Figure S5:

• Offline skill of the subgrid model used in the Paper for online simulations is shown in offline-skill-of-ANN-used-in-online-simulations.zip
• Offline skill of a bigger ANN trained on all data and with dimensional scaling is provided offline-skill-ANN-32-32-neurons.zip

Split of the dataset into train/test/validate is performed in time. param.nc contains grid information, permanent_features.nc contains time-independent features that potentially can be used for training (Coriolis frequency, beta-effect). 

Files train/test/validate.nc contain input and output features to the neural network:

• u and v  [m/s] are horizontal velocities on Arakawa C grid 
• SGSx and SGSy [m/s^2] are zonal and meridional subfilter forcing such that:

          du/dt=SGSx 

          dv/dt=SGSy 

• Txx, Txy, Tyy [m^2/s^2]are horizontal subfilter momentum fluxes such that:

           du/dt = d Txx / dx + d Txy / dy

           dv/dt = d Txy / dx + d Tyy / dy

• rel_vort_h = dv/dx - du/dy [1/s] is relative vorticity
• sh_xx=du/dx - dv/dy [1/s] is horizontal tension
• sh_xy_h=du/dy + dv/dx [1/s] is shearing strain. Note: Index h stands for the horizontal interpolation to the center of the computational cell where thickness (h) resides.
• div=du/dx + dv/dy [1/s] is horizontal divergence
• deformation_radius [m] is the first baroclinic Rossby deformation radius. 
• deltaU [m/s] is the scale of vertical shear
• NH [m/s] is the vertical integral of the Buoyancy frequency

The deformation radius, vertical shear and Buoyancy frequency can be used in future studies to address generalization to different turbulence regimes.