I4DVAR Tutorial

Incremental, Strong Constraint, 4D-Var (I4D-Var) Data Assimilation

Introduction

During this exercise, you will apply the primal form of incremental strong constraint 4D-Var (I4D-Var) to ROMS configured for the U.S. west coast and the California Current System (CCS). This configuration, referred to as WC13, has 30 km horizontal resolution, and 30 levels in the vertical. While 30 km resolution is inadequate for capturing much of the energetic mesoscale circulation associated with the CCS, WC13 captures the broad-scale features of the circulation quite well, and serves as a very useful and efficient illustrative example of I4D-Var.

Model Set-up

The WC13 model domain is shown in Fig. 1 and has open boundaries along the northern, western, and southern edges of the model domain.

Fig. 1: Model Bathymetry with 37°N Transect and Target Area

In the tutorial, you will perform a 4D-Var data assimilation cycle that spans the period 3-6 January, 2004. The 4D-Var control vector δz is comprised of increments to the initial conditions, δx(t₀), surface forcing, δf(t), and open boundary conditions, δb(t). The prior initial conditions, x_b(t₀), are taken from the sequence of 4D-Var experiments described by Moore et al. (2011b) in which data were assimilated every 7 days during the period July 2002- December 2004. The prior surface forcing, f_b(t), takes the form of surface wind stress, heat flux, and a freshwater flux computed using the ROMS bulk flux formulation, and using near surface air data from COAMPS (Doyle et al., 2009). Clamped open boundary conditions are imposed on (u,v) and tracers, and the prior boundary conditions, b_b(t), are taken from the global ECCO product (Wunsch and Heimbach, 2007). The free-surface height and vertically integrated velocity components are subject to the usual Chapman and Flather radiation conditions at the open boundaries. The prior surface forcing and open boundary conditions are provided daily and linearly interpolated in time. Similarly, the increments δf(t) and δb(t) are also computed daily and linearly interpolated in time.

The observations assimilated into the model are satellite SST, satellite SSH in the form of a gridded product from Aviso, and hydrographic observations of temperature and salinity collected from Argo floats and during the GLOBEC/LTOP and CalCOFI cruises off the coast of Oregon and southern California, respectively. The observation locations are illustrated in Fig. 2.

Figure 2: WC13 Observations

a) Aviso SSH	b) Blended SST
c) In Situ Temperature	d) In Situ Salinity

Running I4D-Var

To run this exercise, go first to the directory WC13/I4DVAR. Instructions for compiling and running the model are provided below or can be found in the Readme file. The recommended configuration for this exercise is one outer-loop and 25 inner-loops, and roms_wc13.in is configured for this default case. The number of inner-loops is controlled by the parameter Ninner in roms_wc13.in.

Important CPP Options

The following C-preprocessing options are activated in the build script:

IS4DVAR I4D-Var driver
ANA_SPONGE Analytical enhanced viscosity/diffusion sponge
WC13 Application CPP option

Input NetCDF Files

WC13 requires the following input NetCDF files:

Grid File: ../Data/wc13_grd.nc
Nonlinear Initial File: wc13_ini.nc
Forcing File 01: ../Data/coamps_wc13_lwrad_down.nc
Forcing File 02: ../Data/coamps_wc13_Pair.nc
Forcing File 03: ../Data/coamps_wc13_Qair.nc
Forcing File 04: ../Data/coamps_wc13_rain.nc
Forcing File 05: ../Data/coamps_wc13_swrad.nc
Forcing File 06: ../Data/coamps_wc13_Tair.nc
Forcing File 07: ../Data/coamps_wc13_wind.nc
Boundary File: ../Data/wc13_ecco_bry.nc

Initial Conditions STD File: ../Data/wc13_std_i.nc
Boundary Conditions STD File: ../Data/wc13_std_b.nc
Surface Forcing STD File: ../Data/wc13_std_f.nc
Initial Conditions Norm File: ../Data/wc13_nrm_i.nc
Boundary Conditions Norm File: ../Data/wc13_nrm_b.nc
Surface Forcing Norm File: ../Data/wc13_nrm_f.nc
Observations File: wc13_obs.nc

Various Scripts and Include Files

The following files will be found in WC13/I4DVAR directory after downloading from ROMS test cases SVN repository:

Readme instructions
build_roms.bash bash shell script to compile application
build_roms.sh csh Unix script to compile application
job_i4dvar.sh job configuration script
roms_wc13.in ROMS standard input script for WC13
s4dvar.in 4D-Var standard input script template
wc13.h WC13 header with CPP options

Instructions

To run this application you need to take the following steps:

1. We need to run the model application for a period that is long enough to compute meaningful circulation statistics, like mean and standard deviations for all prognostic state variables (zeta, u, v, T, and S). The standard deviations are written to NetCDF files and are read by the 4D-Var algorithm to convert modeled error correlations to error covariances. The error covariance matrix, D=diag(B_x, B_b, B_f, Q), is very large and not well known. B is modeled as the solution of a diffusion equation as in Weaver and Courtier (2001). Each covariance matrix is factorized as B = K Σ C Σ^T K^T, where C is a univariate correlation matrix, Σ is a diagonal matrix of error standard deviations, and K is a multivariate balance operator.
    
   In this application, we need standard deviations for initial conditions, surface forcing (ADJUST_WSTRESS and ADJUST_STFLUX), and open boundary conditions (ADJUST_BOUNDARY). If the balance operator is activated (BALANCE_OPERATOR and ZETA_ELLIPTIC), the standard deviations for the initial and boundary conditions error covariance are in terms of the unbalanced error covariance (K B_u K^T). The balance operator imposes a multivariate constraint on the error covariance such that the unobserved variable information is extracted from observed data by establishing balance relationships (i.e., T-S empirical formulas, hydrostatic balance, and geostrophic balance) with other state variables (Weaver et al., 2005). The balance operator is not used in the tutorial.
    
   The standard deviations for WC13 have already been created for you:
   ../Data/wc13_std_i.nc initial conditions
   ../Data/wc13_std_b.nc open boundary conditions
   ../Data/wc13_std_f.nc surface forcing (wind stress and net heat flux)
2. Since we are modeling the error covariance matrix, B, we need to compute the normalization coefficients to ensure that the diagonal elements of the associated correlation matrix C are equal to unity. There are two methods to compute normalization coefficients: exact and randomization (an approximation).
    
   The exact method is very expensive on large grids. The normalization coefficients are computed by perturbing each model grid cell with a delta function scaled by the area (2D state variables) or volume (3D state variables), and then by convolving with the squared-root adjoint and tangent linear diffusion operators.
    
   The approximated method is cheaper: the normalization coefficients are computed using the randomization approach of Fisher and Courtier (1995). The coefficients are initialized with random numbers having a uniform distribution (drawn from a normal distribution with zero mean and unit variance). Then, they are scaled by the inverse squared-root of the cell area (2D state variable) or volume (3D state variable) and convolved with the squared-root adjoint and tangent diffusion operators over a specified number of iterations, Nrandom.
    
   Check following parameters in the 4D-Var input script s4dvar.in (see input script for details):
   Nmethod == 0  ! normalization method: 0=Exact (expensive) or 1=Approximated (randomization)
   Nrandom == 5000  ! randomization iterations
   
   LdefNRM == T T T T  ! Create a new normalization files
   LwrtNRM == T T T T  ! Compute and write normalization
   
   CnormI(isFsur) = T  ! IC error covariance, 2D variable at RHO-points
   CnormI(isUbar) = T  ! IC error covariance, 2D variable at U-points
   CnormI(isVbar) = T  ! IC error covariance, 2D variable at V-points
   CnormI(isUvel) = T  ! IC error covariance, 3D variable at U-points
   CnormI(isVvel) = T  ! IC error covariance, 3D variable at V-points
   CnormI(isTvar) = T T  ! IC error covariance, NT tracers
   
   CnormB(isFsur) = T  ! BC error covariance, 2D variable at RHO-points
   CnormB(isUbar) = T  ! BC error covariance, 2D variable at U-points
   CnormB(isVbar) = T  ! BC error covariance, 2D variable at V-points
   CnormB(isUvel) = T  ! BC error covariance, 3D variable at U-points
   CnormB(isVvel) = T  ! BC error covariance, 3D variable at V-points
   CnormB(isTvar) = T T  ! BC error covariance, NT tracers
   
   CnormF(isUstr) = T  ! surface forcing error covariance, U-momentum stress
   CnormF(isVstr) = T  ! surface forcing error covariance, V-momentum stress
   CnormF(isTsur) = T T  ! surface forcing error covariance, NT tracers fluxes
   These normalization coefficients have already been computed for you (WC13/Normalization) using the exact method since this application has a small grid (54x53x30):
   ../Data/wc13_nrm_i.nc initial conditions
   ../Data/wc13_nrm_b.nc open boundary conditions
   ../Data/wc13_nrm_f.nc surface forcing (wind stress and net heat flux)
   Notice that the switches LdefNRM and LwrtNRM are all false (F) since we already computed these coefficients.
    
   The normalization coefficients need to be computed only once for a particular application provided that the grid, land/sea masking (if any), and decorrelation scales (HdecayI, VdecayI, HdecayB, VdecayV, and HdecayF) remain the same. Notice that large spatial changes in the normalization coefficient structure are observed near the open boundaries and land/sea masking regions.
3. Customize your preferred build script and provide the appropriate values for:
   • Root directory, MY_ROOT_DIR
   • ROMS source code, MY_ROMS_SRC
   • Fortran compiler, FORT
   • MPI flags, USE_MPI and USE_MPIF90
   • Path of MPI, NetCDF, and ARPACK libraries according to the compiler are set in my_build_paths.sh. Notice that you need to provide the correct places of these libraries for your computer. If you want to ignore this section, set USE_MY_LIBS value to no.
4. Notice that the most important CPP option for this application is specified in the build script instead of wc13.h:
   setenv MY_CPP_FLAGS "${MY_CPP_FLAGS} -DIS4DVAR"
   setenv MY_CPP_FLAGS "${MY_CPP_FLAGS} -DANA_SPONGE"
   This is to allow flexibility with different CPP options.
    
   For this to work, however, any #undef directives MUST be avoided in the header file wc13.h since it has precedence during C-preprocessing.
5. You MUST use the build script to compile.
6. Customize the ROMS input script roms_wc13.in and specify the appropriate values for the distributed-memory partition. It is set by default to:
   NtileI == 2  ! I-direction partition
   NtileJ == 4  ! J-direction partition
   Notice that the adjoint-based algorithms can only be run in parallel using MPI. This is because of the way that the adjoint model is constructed.
7. Customize the configuration script job_i4dvar.sh and provide the appropriate place for the substitute Perl script:
   set SUBSTITUTE=${ROMS_ROOT}/ROMS/Bin/substitute
   This script is distributed with ROMS and it is found in the ROMS/Bin sub-directory. Alternatively, you can define ROMS_ROOT environmental variable in your .cshrc login script. For example, I have:
   setenv ROMS_ROOT /home/arango/ocean/toms/repository/trunk
8. Execute the configuration job_i4dvar.sh before running the model. It copies the required files and creates i4dvar.in input script from template s4dvar.in. This has to be done every time that you run this application. We need a clean and fresh copy of the initial conditions and observation files since they are modified by ROMS during execution.
9. Run ROMS with data assimilation:
   mpirun -np 8 romsM roms_wc13.in > & log &
10. We recommend creating a new subdirectory EX1, and saving the solution in it for analysis and plotting to avoid overwriting solutions when playing with different parameters. For example
    mkdir EX1
    mv Build_roms psas.in *.nc log EX1
    cp -p romsM roms_wc13.in EX1
    where log is the ROMS standard output specified in the previous step.

For Exercise 2 you will only be experimenting the Ninner and Nouter parameters so you do not need to recompile. However, you will need to edit your roms_wc13.in and execute the job_i4dvar.sh script before rerunning the model.

Plotting your Results

Several Matlab and plotting package scripts are provided in the directory WC13/plotting to plot some of the I4D-Var output:

plot_i4dvar_cost.m plots cost function values
plot_i4dvar_increments.m plots initial conditions surface increments
ccnt_i4dvar_increments.in plots initial conditions increments at z=100m
csec_i4dvar_increments.in plots initial conditions increments section at 37°N

Recall that I4D-Var minimizes the cost function given by:

$$J={\frac {1}{2}}\,\delta {\bf {z}}^{T}\,{\bf {D}}^{-1}\delta {\bf {z}}\,+\,\,{\frac {1}{2}}\,{\left({\bf {G}}\,\delta {\bf {z}}-{\bf {d}}\right)}^{T}\,{\bf {R}}^{-1}\,\left({\bf {G}}\,\delta {\bf {z}}-{\bf {d}}\right)=J_{b}+J_{o}$$

Plot first the I4D-Var cost function $J$ and its components $J_{b}$, $J_{o}$ and the theoretical minimum value $J_{min}=N_{obs}/2$ using the Matlab script plot_i4dvar_cost.m.

Next, plot the surface initial conditions increments $\delta {\bf {x}}(t_{0})$ and the surface forcing increments $\delta {\bf {f}}(t)$ at initial time using Matlab script plot_i4dvar_increments.m or ROMS plotting package script ccnt_i4dvar_increments.in for horizontal plots at 100 m or csec_i4dvar_increments.in for cross-sections along 37°N.

Results

The I4D-Var cost function value for each inner loop iteration is shown below:

I4D-Var Cost Function 25 Iterations

I4D-Var Cost Function 50 Iterations

I4D-Var Cost Function 2 outer and 25 inner loops

We have also included a plot with 50 inner loops so that you can see that the solution has more or less converged by 25 iterations. The total cost function J (black curve), observation cost function J_o (blue curve), and background cost function J_b (red curve) are plotted on a log₁₀ scale. The value of the nonlinear cost function J_NL at the end of the inner loops is also shown (red X). The horizontal, dashed line shows the theoretical J_min.

File:i4dvar cost 25iter daily 2019.png I4D-Var Cost Function 25 Iterations 2 hours instead of daily

File:i4dvar cost 50iter daily 2019.png I4D-Var Cost Function 50 Iterations 2 hours instead of daily

File:i4dvar cost 2inner 25outer daily 2019.png I4D-Var Cost Function 2 outer and 25 inner loops 2 hours instead of daily

The above plots. The total cost function J (black curve), observation cost function J_o (blue curve), and background cost function J_b (red curve) are plotted on a log₁₀ scale. The value of the nonlinear cost function J_NL at the end of the inner loops is also shown (red X). The horizontal, dashed line shows the theoretical J_min.

---

The I4D-Var initial conditions increments for free-surface (m), surface wind stress components (Pa), and surface net heat flux (W/m²) are shown below:

a) Free-surface

b) τx

c) τy

d) Net Heat Flux

The I4D-Var initial conditions increments at the surface for temperature (°C), salinity, and momentum components (m/s) are shown below:

a) Temperature

b) Salinity

c) U-Momentum

d) V-Momentum

The I4D-Var initial conditions increments at 100m for temperature (°C), salinity, and momentum components (m/s) are shown below:

a) Temperature

b) Salinity

c) U-Momentum

d) V-Momentum

A cross-section along 37°N for the I4D-Var initial conditions increments is shown below.

a) Temperature

b) Salinity

c) U-Momentum

d) V-Momentum