Ensemble Kalman filter

The ensemble Kalman filter (EnKF) is a recursive filter suitable for problems with a large number of variables, such as discretizations of partial differential equations in geophysical models. The EnKF originated as a version of the Kalman filter for large problems (essentially, the covariance matrix is replaced by the sample covariance), and it is now an important data assimilation component of ensemble forecasting. EnKF is related to the particle filter (in this context, a particle is the same thing as ensemble member) but the EnKF makes the assumption that all probability distributions involved are Gaussian; when it is applicable, it is much more efficient than the particle filter. This article briefly describes the derivation and practical implementation of the basic version of EnKF, and reviews several extensions.

Introduction

The Ensemble Kalman Filter (EnKF) is a Monte-Carlo implementation of the Bayesian update problem: Given a probability density function (pdf) of the state of the modeled system (the "prior", called often the forecast in geosciences) and the data likelihood, the Bayes theorem is used to obtain the pdf after the data likelihood has been taken into account (the "posterior", often called the analysis). This is called a Bayesian update. The Bayesian update is combined with advancing the model in time, incorporating new data from time to time. The original Kalman Filter R. E. Kalman, "A new approach to linear filtering and prediction problems", Transactions of the ASME -- Journal of Basic Engineering, Series D, 82 (1960), pp. 35--45.

] assumes that all pdfs are Gaussian (the Gaussian assumption) and provides algebraic formulas for the change of the mean and the covariance matrix by the Bayesian update, as well as a formula for advancing the covariance matrix in time provided the system is linear. However, maintaining the covariance matrix is not feasible computationally for high-dimensional systems. For this reason, EnKFs were developed. G. Evensen, "Sequential data assimilation with nonlinear quasi-geostrophic model using Monte Carlo methods to forecast error statistics", Journal of Geophysical Research, 99 (C5) (1994), pp. 143--162.

] P. Houtekamer and H. L. Mitchell, "Data assimilation using an ensemble Kalman filter technique", Monthly Weather Review, 126 (1998), pp. 796--811.

] EnKFs represent the distribution of the system state using a collection of state vectors, called an ensemble, and replace the covariance matrix by the sample covariance computed from the ensemble. The ensemble is operated with as if it were a random sample, but the ensemble members are really not independent - the EnKF ties them together. One advantage of EnKFs is that advancing the pdf in time is achieved by simply advancing each member of the ensemble. For a survey of EnKF and related data assimilation techniques, see. G. Evensen, "Data assimilation : The ensemble Kalman filter, Springer, Berlin, 2007.

] A version of this article is also available as the technical report. J. Mandel, "A brief tutorial on the ensemble Kalman filter", CCM Report 242, University of Colorado at Denver and Health Sciences Center, February 2007. [http://www.math.cudenver.edu/ccm/reports/rep242.pdf report]

]

A derivation of the EnKF

The Kalman Filter

Let us review first the Kalman filter. Let $mathbf\{x\}$ denote the $n$-dimensional state vector of a model, and assume that it has Gaussian probability distribution with mean $mathbf\{mu\}$ and covariance $Q$, i.e., its pdf is

:$p(mathbf\{x\})proptoexpleft(\; -frac\{1\}\{2\}(mathbf\{x\}-mathbf\{mu\; \})^\{mathrm\{TQ^\{-1\}(mathbf\{x\}-mathbf\{mu\})\; ight)\; .$

Here and below, $propto$ means proportional; a pdf is always scaled so that its integral over the whole space is one. This $p(mathbf\{x\})$, called the "prior", was evolved in time by running the model and now is to be updated to account for new data. It is natural to assume that the error distribution of the data is known; data have to come with an error estimate, otherwise they are meaningless. Here, the data $mathbf\{d\}$ is assumed to have Gaussian pdf with covariance $R$ and mean $Hmathbf\{x\}$, where $H$ is the so-called the observation matrix. The covariance matrix $R$ describes the estimate of the error of the data; if the random errors in the entries of the data vector $mathbf\{d\}$ are independent, $R$ is diagonal and its diagonal entries are the squares of the standard deviation (“error size”) of the error of the corresponding entries of the data vector $mathbf\{d\}$. The value $Hmathbf\{x\}$ is what the value of the data would be for the state $mathbf\{x\}$ in the absence of data errors. Then the probability density $p(mathbf\{d\}|mathbf\{x\})$ of the data $mathbf\{d\}$ conditional of the system state $mathbf\{x\}$, called the data likelihood, is

:$pleft(\; mathbf\{d\}|mathbf\{x\}\; ight)\; proptoexpleft(\; -frac\{1\}\{2\}(mathbf\{d\}-Hmathbf\{x\})^\{mathrm\{TR^\{-1\}(mathbf\{d\}-Hmathbf\{x\})\; ight)\; .$

The pdf of the state and the data likelihood are combined to give the new probability density of the system state $mathbf\{x\}$ conditional on the value of the data $mathbf\{d\}$ (the "posterior") by the Bayes theorem,

:$pleft(\; mathbf\{x\}|mathbf\{d\}\; ight)\; propto\; pleft(\; mathbf\{d\}|mathbf\{x\}\; ight)\; p(mathbf\{x\}).$

The data $mathbf\{d\}$ is fixed once it is received, so denote the posterior state by $mathbf\{hat\{x$ instead of $mathbf\{x\}|mathbf\{d\}$ and the posterior pdf by $pleft(\; mathbf\{hat\{x\; ight)$. It can be shown by algebraic manipulations B. D. O. Anderson and J. B. Moore, "Optimal filtering", Prentice-Hall, Englewood Cliffs, N.J., 1979.

] that the posterior pdf is also Gaussian,

:$pleft(\; mathbf\{hat\{x\; ight)\; proptoexpleft(\; -frac\{1\}\{2\}(mathbf\{hat\{x-mathbf\{hat\{mu)^\{mathrm\{TP^\{-1\}(mathbf\{hat\{x-mathbf\{hat\{mu)\; ight)\; ,$

with the posterior mean $mathbf\{hat\{mu$ and covariance $hat\{Q\}$ given by the Kalman update formulas

:$mathbf\{hat\{mu=mathbf\{mu\}+Kleft(\; mathbf\{d\}-Hmathbf\{mu\}\; ight)\; ,quadhat\{Q\}=left(\; I-KH\; ight)\; Q,$

where

:$K=QH^\{mathrm\{Tleft(\; HQH^\{mathrm\{T+R\; ight)\; ^\{-1\}$

is the so-called Kalman gain matrix.

The Ensemble Kalman Filter

The EnKF is a Monte Carlo approximation of the Kalman filter, which avoids evolving the covariance matrix of the pdf of the state vector $mathbf\{x\}$. Instead, the pdf is represented by an ensemble

:$X=left\; [\; mathbf\{x\}\_\{1\},ldots,mathbf\{x\}\_\{N\}\; ight]\; =left\; [\; mathbf\{x\}\_\{i\}\; ight]\; .$

$X$ is an $n\; imes\; N$ matrix whose columns are the ensemble members, and it is called the "prior ensemble". Ideally, ensemble members would form a sample from the prior distribution. However, the ensemble members are not in general independent except in the initial ensemble, since every EnKF step ties them together. They are deemed to be approximately independent, and all calculations proceed as if they actually were independent.

Replicate the data $mathbf\{d\}$ into an $m\; imes\; N$ matrix

:$D=left\; [\; mathbf\{d\}\_\{1\},ldots,mathbf\{d\}\_\{N\}\; ight]\; =left\; [\; mathbf\{d\}\_\{i\}\; ight]\; ,\; quad\; mathbf\{d\}\_\{i\}=mathbf\{d\}+epsilon\_\{i\},\; quad\; epsilon\_\{i\}\; =N(0,R,)$

so that each column $mathbf\{d\}\_\{i\}$ consists of the data vector $mathbf\{d\}$ plus a random vector from the $n$-dimensional normal distribution $N(0,R)$. If, in addition, the columns of $X$ are a sample from the prior probability distribution, then the columns of

:$hat\{X\}=X+K(D-HX)$

form a sample from the posterior probability distribution. The EnKF is now obtained C. J. Johns and J. Mandel, "A two-stage ensemble Kalman filter for smooth data assimilation". Environmental and Ecological Statistics, in print. Special issue, Conference on New Developments of Statistical Analysis in Wildlife, Fisheries, and Ecological Research, Oct 13-16, 2004, Columbia, MI. CCM Report 221, University of Colorado at Denver and Health Sciences Center, 2005. [http://www.math.cudenver.edu/ccm/reports/rep221.pdf report]

] simply by replacing the state covariance $Q$ in Kalman gain matrix $K=QH^\{mathrm\{Tleft(\; HQH^\{mathrm\{T+R\; ight)\; ^\{-1\}$ by the sample covariance $C$ computed from the ensemble members (called the "ensemble covariance").

Implementation

Basic formulation

Here we follow. G. Burgers, P. J. van Leeuwen, and G. Evensen, "Analysis scheme in the ensemble Kalman filter", Monthly Weather Review, 126 (1998), pp. 1719--1724.

] G. Evensen, "The ensemble Kalman filter: Theoretical formulation and practical implementation", Ocean Dynamics, 53 (2003), pp. 343--367.

] Suppose the ensemble matrix $X$ and the data matrix $D$ are as above. The ensemble mean and the covariance are

:$Eleft(\; X\; ight)\; =frac\{1\}\{N\}sum\_\{k=1\}^\{N\}mathbf\{x\}\_\{k\},quad\; C=frac\{AA^\{T\{N-1\},$

where

:$A=X-Eleft(\; X\; ight)\; =X-frac\{1\}\{N\}left(\; Xmathbf\{e\}\_\{N\; imes1\}\; ight)\; mathbf\{e\}\_\{1\; imes\; N\},$

and $mathbf\{e\}$ denotes the matrix of all ones of the indicated size.

The posterior ensemble $X^\{p\}$ is then given by

:$hat\{X\}approx\; X^\{p\}=X+CH^\{T\}left(\; HCH^\{T\}+R\; ight)\; ^\{-1\}(D-HX),$

where the perturbed data matrix $D$ is as above. Since $C$ can be written as

:

one can see that "the posterior ensemble consists of linear combinations of members of the prior ensemble". (This is, however, no longer the case when the covariance matrix of the ensemble is artificially modified, as in localized ensemble Kalman filters.)

Note that since $R$ is a covariance matrix, it is always positive semidefinite and usually positive definite, so the inverse above exists and the formula can be implemented by the Choleski decomposition. J. Mandel, "Efficient implementation of the ensemble Kalman filter". CCM Report 231, University of Colorado at Denver and Health Sciences Center. [http://www.math.cudenver.edu/ccm/reports/rep231.pdf link] , June 2006.

] In, $R$ is replaced by the sample covariance $DD^\{T\}/left(\; N-1\; ight)$ and the inverse is replaced by a pseudoinverse, computed using the Singular Value Decomposition (SVD) .

Since these formulas are matrix operations with dominant Level 3 operations, G. H. Golub and C. F. V. Loan, "Matrix Computations", Johns Hopkins Univ. Press, 1989. Second Edition.

] they are suitable for efficient implementation using software packages such as LAPACK (on serial and shared memory computers) and ScaLAPACK (on distributed memory computers). Instead of computing the inverse of a matrix and multiplying by it, it is much better (several times cheaper and also more accurate) to compute the Choleski decomposition of the matrix and treat the multiplication by the inverse as solution of a linear system with many simultaneous right-hand sides.

Observation matrix-free implementation

It is usually inconvenient to construct and operate with the matrix $H$ explicitly; instead, a function $h(mathbf\{x\})$ of the form

:$h(mathbf\{x\})=Hmathbf\{x\},$

is more natural to compute. The function $h$ is called the "observation function" or, in the inverse problems context, the "forward operator". The value of $h(mathbf\{x\})$ is what the value of the data would be for the state $mathbf\{x\}$ assuming the measurement is exact. Then the posterior ensemble can be rewritten as

:$X^\{p\}=X+frac\{1\}\{N-1\}Aleft(\; HA\; ight)\; ^\{T\}P^\{-1\}(D-HX)$

where

:$HA=HX-frac\{1\}\{N\}left(\; left(\; HX\; ight)\; mathbf\{e\}\_\{N\; imes1\}\; ight)\; mathbf\{e\}\_\{1\; imes\; N\},$

and

:$P=frac\{1\}\{N-1\}HAleft(\; HA\; ight)\; ^\{T\}+R,$

with

:

Consequently, the ensemble update can be computed by evaluating the observation function $h$ on each ensemble member once and the matrix $H$ does not need to be known explicitly. This formula holds also for an observation function $h(mathbf\{x\})=Hmathbf\{x+f\}$ with a fixed offset $mathbf\{f\}$, which also does not need to be known explicitly. The above formula has been commonly used for a nonlinear observation function $h$, such as the position of a hurricane vortex. Y. Chen and C. Snyder, "Assimilating vortex position with an ensemble Kalman filter". Monthly Weather Review, to appear, 2006. [http://www.mmm.ucar.edu/people/snyder/papers/ChenSnyder2006_draft.pdf preprint] .

] In that case, the observation function is essentially approximated by a linear function from its values at ensemble members.

Implementation for a large number of data points

For a large number $m$ of data points, the multiplication by $P^\{-1\}$ becomes a bottleneck. The following alternative formula is advantageous when the number of data points $m$ is large (such as when assimilating gridded or pixel data) and the data error covariance matrix $R$ is diagonal (which is the case when the data errors are uncorrelated), or cheap to decompose (such as banded due to limited covariance distance). Using the Sherman–Morrison–Woodbury formula W. W. Hager, "Updating the inverse of a matrix", SIAM Rev., 31 (1989), pp. 221--239.

]

:$(R+UV^\{T\})^\{-1\}=R^\{-1\}-R^\{-1\}U(I+V^\{T\}R^\{-1\}U)^\{-1\}V^\{T\}R^\{-1\},$

with

:$U=frac\{1\}\{N-1\}HA,quad\; V=HA,$

gives

:

which requires only the solution of systems with the matrix $R$ (assumed to be cheap) and of a system of size $N$ with $m$ right-hand sides. See for operation counts.

Further extensions

The EnKF version described here involves randomization of data. For filters without randomization of data, see. J. L. Anderson, "An ensemble adjustment Kalman filter for data assimilation", Monthly Weather Review, 129 (1999), pp. 2884--2903.

] G. Evensen, "Sampling strategies and square root analysis schemes for the EnKF", Ocean Dynamics, 54 (2004), pp. 539--560.

] M. K. Tippett, J. L. Anderson, C. H. Bishop, T. M. Hamill, and J. S. Whitaker, "Ensemble square root filters", Monthly Weather Review, 131 (2003), pp. 1485--1490.

]

Since the ensemble covariance is rank deficient (there are many more state variables, typically millions, than the ensemble members, typically less than a hundred), it has large terms for pairs of points that are spatially distant. Since in reality the values of physical fields at distant locations are not that much correlated, the covariance matrix is tapered off artificially based on the distance, which gives rise to localized EnKF algorithms. J. L. Anderson, "A local least squares framework for ensemble filtering", Monthly Weather Review, 131 (2003), pp. 634--642.] E. Ott, B. R. Hunt, I. Szunyogh, A. V. Zimin, E. J. Kostelich, M. Corazza, E. Kalnay, D. Patil, and J. A. Yorke, "A local ensemble Kalman filter for atmospheric data assimilation", Tellus A, 56 (2004), pp. 415--428.] These methods modify the covariance matrix used in the computations and, consequently, the posterior ensemble is no longer made only of linear combinations of the prior ensemble.

For nonlinear problems, EnKF can create posterior ensemble with non-physical states. This can be alleviated by regularization, such as penalization of states with large spatial gradients.

For problems with coherent features, such as firelines, squall lines, and rain fronts, there is a need to adjust the simulation state by distorting the state in space as well as by an additive correction to the state. The morphing EnKF J. D. Beezley and J. Mandel, "Morphing ensemble Kalman filters". Tellus A, submitted. CCM Report 240, University of Colorado at Denver and Health Sciences Center, February 2007, [http://www-math.cudenver.edu/ccm/reports/rep240.pdf report] .

] J. Mandel and J. D. Beezley, "Predictor-corrector and morphing ensemble filters for the assimilation of sparse data into high dimensional nonlinear systems". CCM Report 239, University of Colorado at Denver and Health Sciences Center. [http://www.math.cudenver.edu/ccm/reports/rep239.pdf report] , November 2006. 11th Symposium on Integrated Observing and Assimilation Systems for the Atmosphere, Oceans, and Land Surface (IOAS-AOLS), CD-ROM, Paper 4.12, 87th American Meteorological Society Annual Meeting, San Antonio, TX, January 2007, [http://www.ametsoc.org link] .

] employs intermediate states, obtained by techniques borrowed from image registration and morphing, instead of linear combinations of states.

EnKFs rely on the Gaussian assumption, though they are of course used in practice for nonlinear problems, where the Gaussian assumption is not satisfied. Related filters attempting to relax the Gaussian assumption in EnKF while preserving its advantages include filters that fit the state pdf with multiple Gaussian kernels, J. L. Anderson and S. L. Anderson, "A Monte Carlo implementation of the nonlinear filtering problem to produce ensemble assimilations and forecasts", Monthly Weather Review, 127 (1999), pp. 2741--2758.

] filters that approximate the state pdf by Gaussian mixtures, T. Bengtsson, C. Snyder, and D. Nychka, "Toward a nonlinear ensemble filter for high dimensional systems", Journal of Geophysical Research - Atmospheres, 108(D24) (2003), pp. STS 2--1--10. [http://www.image.ucar.edu/pub/nychka/manuscripts/bengtsson.pdf preprint] .

] a variant of the particle filter with computation of particle weights by density estimation, and a variant of the particle filter with thick tailed data pdf to alleviate particle filter degeneracy. P. van Leeuwen, "A variance-minimizing filter for large-scale applications", Monthly Weather Review, 131 (2003), pp. 2071--2084.]

References

See also

* Data assimilation
* Kalman filter
* Numerical weather prediction#Ensembles
* Particle filter
* Recursive Bayesian estimation

External links

* [http://www.cmascenter.org/conference/2006/abstracts/zubrow_session7.pdf Ensemble Adjusted Kalman Filter applied to CO measurements] (EAKF):"(EAKF-CMAQ: DEVELOPMENT AND INITIAL EVALUATION OF AN ENSEMBLE ADJUSTMENT KALMAN FILTER BASED DATA ASSIMILATION FOR CO)"
* [http://enkf.nersc.no EnKF webpage] (EnKF)
* [http://topaz.nersc.no TOPAZ, real-time forecasting of the North Atlantic ocean and Arctic sea-ice with the EnKF] (TOPAZ)