RBL4D-Var Tutorial

Restricted B-preconditioned Lanczos 4D-Var (RBL4D-Var)

Introduction

Notice: This algorithm began based on the Physical-space Statistical Analysis System (PSAS) algorithm but has evolved into a Restricted B-preconditioned Lanczos 4D-Var (RBL4D-Var). Some plots on this page still have the PSAS title but remain correct. The algorithm and ROMS CPP Options relating to this data assimilation system were renamed in SVN revision 1022 (May 13, 2020) and are explained in Trac ticket #854.

In this tutorial you will apply the strong/weak constraint, dual form of 4-Dimensional Variational (4D-Var) data assimilation based on the Restricted B-preconditioned Lanczos 4D-Var (RBLanczos) algorithm (RBL4D-Var) to ROMS configured for the U.S. west coast and the California Current System (WC13). In RBL4D-Var the search for the best ocean circulation estimate proceeds in the space spanned only by the observations, as opposed to the full space spanned by the model (i.e. the primal form, I4D-Var). Formally, the primal and dual formulations yield identical estimates of the ocean circulation so one might wonder if there is any advantage of one form over the other? The practical advantages and disadvantages to both approaches are discussed in Moore et al. (2011b, c).

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 RBL4D-Var

To run this tutorial, go first to the directory WC13/RBL4DVAR. 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 26 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:

RBL4DVAR 4D-PSAS driver (observation space)
ANA_SPONGE Analytical enhanced viscosity/diffusion sponge
BGQC Backgound quality control of observations
MINRES Minimal Residual Method for 4D-Var minimization
RPCG Restricted B-preconditioned Lanczos minimization
POSTERIOR_EOFS Estimate posterior analysis error
POSTERIOR_ERROR_I Estimate initial posterior analysis error
TIME_CONV Weak-constraint 4D-Var time convolution
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
Model STD File: ../Data/wc13_std_m.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
Model Norm File: ../Data/wc13_nrm_m.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/RBL4DVAR directory after downloading from ROMS test cases SVN repository:

Exercise_3.pdf Exercise 3 instructions
Exercise_4.pdf Exercise 4 instructions
Readme instructions
build_roms.csh csh Unix script to compile application
build_roms.sh bash shell script to compile application
job_rbl4dvar.csh job configuration script
roms_wc13_2hours.in ROMS standard input script for WC13 2 hour averages
roms_wc13_daily.in ROMS standard input script for WC13 daily averages
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_m.nc model error (if weak constraint)
   ../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, D, 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 approximate 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
   
   CnormM(isFsur) = T  ! model error covariance, 2D variable at RHO-points
   CnormM(isUbar) = T  ! model error covariance, 2D variable at U-points
   CnormM(isVbar) = T  ! model error covariance, 2D variable at V-points
   CnormM(isUvel) = T  ! model error covariance, 3D variable at U-points
   CnormM(isVvel) = T  ! model error covariance, 3D variable at V-points
   CnormM(isTvar) = T T  ! model error covariance, NT tracers
   
   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 (../Normalization) using the exact method since this application has a small grid (54x53x30):
   ../Data/wc13_nrm_i.nc initial conditions
   ../Data/wc13_nrm_m.nc model error (if weak constraint)
   ../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.csh. 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 options for this application are specified in the build script instead of wc13.h:
   setenv MY_CPP_FLAGS "${MY_CPP_FLAGS} -DRBL4DVAR"
   setenv MY_CPP_FLAGS "${MY_CPP_FLAGS} -DANA_SPONGE"
   setenv MY_CPP_FLAGS "${MY_CPP_FLAGS} -DBGQC"
   setenv MY_CPP_FLAGS "${MY_CPP_FLAGS} -DMINRES"
   setenv MY_CPP_FLAGS "${MY_CPP_FLAGS} -DRPCG"
   setenv MY_CPP_FLAGS "${MY_CPP_FLAGS} -DPOSTERIOR_EOFS"
   setenv MY_CPP_FLAGS "${MY_CPP_FLAGS} -DPOSTERIOR_ERROR_I"
   setenv MY_CPP_FLAGS "${MY_CPP_FLAGS} -DTIME_CONV"
   
   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_rbl4dvar.csh 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_rbl4dvar.csh before running the model. It copies the required files and creates rbl4dvar.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 for each Case in this exercise (EX3_CONGRAD, EX3_MINRES, EX3_RPCG, and EX4_WEAK), and saving the solution in it for analysis and plotting to avoid overwriting solutions when playing with different parameters. For example
    mkdir EX3_CONGRAD
    mv Build_roms rbl4dvar.in *.nc log EX3_CONGRAD
    cp -p romsM roms_wc13.in EX3_CONGRAD
    where log is the ROMS standard output specified in the previous step.

Plotting your Results

Several Matlab scripts are provided in the directory WC13/plotting which will allow you to plot some of the RBL4D-Var output.

Recall that RBL4D-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 RBL4D-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_rbl4dvar_cost_compare.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_rbl4dvar_increments.m or ROMS plotting package script ccnt_rbl4dvar_increments.in for horizontal plots at 100 m or csec_rbl4dvar_increments.in for cross-sections along 37°N.

Results

The RBL4D-Var cost function value for each inner loop iteration is shown below for various minimization algorithms:

a) RBL4D-Var Cost Function prior saved daily

b) RBL4D-Var Cost Function, Background Quality Control prior saved daily

c) RBL4D-Var Cost Function, Weak Constraint prior saved daily

The convergence of various minimization algorithms (RBLanczos default, MINRES, and RPCG) is compared with the primal formulation I4DVAR.

a) RBL4D-Var Cost Function prior saved every 2 hours

b) RBL4D-Var Cost Function, Background Quality Control prior saved every 2 hours

c) RBL4D-Var Cost Function, Weak Constraint prior saved every 2 hours

In the above plots, the prior nonlinear trajectory is saved every 2 hours instead of daily (NHIS = NTLM = NSFF = NOBC = 4). It shows the convergence of various minimization algorithms (RBLanczos default, MINRES, and RPCG) is compared with the primal formulation I4DVAR.

---

The RBL4D-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 RBL4D-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 RBL4D-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 RBL4D-Var initial conditions increments is shown below.

a) Temperature

b) Salinity

c) U-Momentum

d) V-Momentum