Uniform error bounds of
the ensemble square root filter for
chaotic dynamics with
multiplicative covariance inflation
The 1st MMS Workshop for Young Researchers 2024/11/20-21 @Kyoto University
The first author is supported by RIKEN Junior Research Associate Program and JST SPRING JPMJSP2110.
The second author is supported by JST MIRAI JPMJMI22G1.
Kota Takeda , Takashi Sakajo
Kyoto University, RIKEN R-CCS
- What's Data Assimilation (DA)?
- Setting: model, observation
- Problem: Sequential state estimation (filtering)
- Approach & Algorithms: EnKF
- Results (Error analysis): prev. & our
- Discussion/Further analysis
Contents
Data Assimilation (DA)
Seamless integration of data and models
Model
Dynamical systems and its numerical simulations
ODE, PDE, SDE,
Numerical analysis, HPC,
Fluid mechanics, Physics
Data
Obtained from experiments or measurements
Statistics, Data analysis,
Radiosonde/Radar/Satellite/ observation
×
Numerical Weather Prediction
typical application
×
Data Assimilation (DA)
-
Control
- Nudging/Continuous Data Assimilation
-
Bayesian
- Particle Filter (PF)
- Ensemble Kalman filter (EnKF)
-
Variational
- Four-dimensional Variation method (4DVar)
Several formulations & algorithms of DA:
focus on
Setup: Model
Model dynamics in
Semi-group in
generates
(assume)
: separable Hilbert space
: separable Hilbert space (state space)
The known model generates
the unknown true chaotic orbit.
A small perturbation can
grow exponentially fast.
unknown orbit
unknown
Chaotic dynamical systems
The orbits of the Lorenz 63 equation.
Setup: Observation
Discrete-time model (at obs. time)
We have noisy observations in
Noisy observation
observe
...
bounded linear
(another Hilbert space).
Sequential state estimation
...
given
Problem
estimate
known:
obs. noise distribution.
- What's DA? ✓
- Setting: model, observation ✓
- Problem: Sequential state estimation (filtering) ✓
-
Approach & Algorithms:
- Bayesian approach → Gaussian approximation (KF) → Monte Carlo (EnKF)
- EnKF: PO (Stoc.), ETKF (Det.) + inflation (technique)
- Results (Error analysis)
- Discussion/Further analysis
Contents
Bayesian Data Assimilation
Approach: Bayesian Data Assimilation
construct a random variable so that
given
Problem
estimate
Sequential Data Assimilation
(I) Prediction
(II)Analysis
We can construct it iteratively (2 steps at each time step)
(I) Prediction
(II) Analysis
Prediction
...
by model
by Bayes' rule
with auxiliary variable
Repeat (I) & (II)
Proposition
Need approximations!
→ next slide
Data Assimilation Algorithms
Only focus on Approximate Gaussian Algorithms
KF & EnKF
Kalman filter (KF) [Kalman 1960]
- All Gaussian distribution
- Update of mean and covariance.
Assume: linear F, Gaussian noises
Ensemble Kalman filter (EnKF)
Non-linear extension by Monte Carlo method
- Update of ensemble
(Reich+2015) S. Reich and C. Cotter (2015), Probabilistic Forecasting and Bayesian Data Assimilation, Cambridge University Press, Cambridge.
(Law+2015) K. J. H. Law, A. M. Stuart, and K. C. Zygalakis (2015), Data Assimilation: A Mathematical Introduction, Springer.
KF & EnKF
Kalman filter(KF)
Assume: linear F, Gaussian noises
Linear F
Bayes' rule
predict
analysis
Ensemble Kalman filter (EnKF)
Non-linear extension by Monte Carlo method
approx.
ensemble
Two variants
→ next slide
- Two major variants of EnKF:
- Perturbed Observation (PO): stochastic → sampling error
- Ensemble Transform Kalman Filter (ETKF): deterministic
Two EnKF variants
...
Repeat (I) & (II)
(II)Analysis
(I) Prediction
(Burgers+1998)
(Bishop+2001)
Two variants
PO (stochastic EnKF)
(Burgers+1998) G. Burgers, P. J. van Leeuwen, and G. Evensen (1998), Analysis Scheme in the Ensemble Kalman Filter, Mon. Weather Rev., 126, 1719–1724.
Stochastic!
Perturbed Observation
: A transpose (adjoint) of a matrix (linear operator)
ETKF (deterministic EnKF)
(Bishop+2001) C. H. Bishop, B. J. Etherton, and S. J. Majumdar (2001), Adaptive Sampling with the Ensemble Transform Kalman Filter. Part I: Theoretical Aspects, Mon. Weather Rev., 129, 420–436.
Deterministic!
The explicit representation exists!
Ensemble Transform Kalman Filter
ETKF (deterministic EnKF)
Ensemble Transform Kalman Filter
Lemma
ETKF is one of a class of deterministic EnKF known as ESRF (Ensemble Square Root Filters).
※ESRF (deterministic EnKF)
ESRF
ETKF
EAKF (Ensemble Adjustment Kalman Filter)
: bounded linear
(Anderson 2001) J. L. Anderson (2001), An Ensemble Adjustment Kalman Filter for Data Assimilation, Mon. Weather Rev., 129, 2884–2903.
Key perspective of EnKF (KF)
The analysis state is the weighted average of
the prediction and observation
The weighted is determined by estimated uncertainties
: uncertainty in observation (prescribed)
: uncertainty in prediction (approx. by ensemble)
Inflation (numerical technique)
Issue: Underestimation of covariance through the DA cycle
→ Overconfident in the prediction → Poor state estimation
(I) Prediction → inflation → (II) Analysis → ...
Rank is improved
Rank is not improved
Idea: inflating the covariance before (II)
Two basic methods of inflation
※depending on the implementation
※
PO
Two EnKF variants
Simple
Stochastic
Require many samples
ETKF
Complicated
Deterministic
Require few samples
Practical !
Inflation: add., multi.
Inflation: only multi.
※depending on the implementation
※
- What's DA? ✓
- Setting: model, observation ✓
- Problem: Sequential state estimation (filtering) ✓
-
Approach & Algorithms: ✓
- Bayesian approach → Gaussian approximation (KF) → Monte Carlo (EnKF)
- EnKF: PO (Stoc.), ETKF (Det.) + inflation (technique)
- Results (Error analysis)
- Discussion/Further analysis
Contents
Error analysis of EnKF
Assumptions on model
(Reasonable)
Assumption on observation
(Strong)
We focus on finite-dimensional state spaces
(T.+2024) K. T. & T. Sakajo (2024), SIAM/ASA Journal on Uncertainty Quantification, 12(4), 1315–1335.
Assumptions for analysis
Assumption (obs-1)
"full observation"
Strong!
Assumption (model-1)
Assumption (model-2)
Assumption (model-2')
Reasonable
"one-sided Lipschitz"
"Lipschitz"
"bounded"
or
Dissipative dynamical system
Dissipative dynamical system
Dissipative dynamical systems
Lorenz 63, 96 equations, widely used in geoscience,
Example (Lorenz 96)
non-linear conserving
linear dissipating
forcing
Assumptions (model-1, 2, 2')
hold
incompressible 2D Navier-Stokes equations (2D-NSE) in infinite-dim. setting
Lorenz 63 eq. can be written in the same form.
→ We can discuss ODE and PDE in a unified framework.
can be generalized
bilinear operator
linear operator
dissipative dynamical systems
Mimics chaotic variation of
physical quantities at equal latitudes.
(Lorenz1996) (Lorenz+1998)
Dissipative dynamical systems
The orbits of the Lorenz 63 equation.
Literature review
-
Consistency (Mandel+2011, Kwiatkowski+2015)
-
Sampling errors (Al-Ghattas+2024)
-
Stability (Tong+2016a,b)
-
Error analysis (full observation )
-
PO + add. inflation (Kelly+2014)
-
ETKF + multi. inflation (T.+2024) → current
-
EAKF + multi. inflation (T.)
-
-
Error analysis (partial observation) → future (partially solved, T.)
How EnKF approximate KF?
How EnKF approximate
the true state?
※ Continuous time formulation (EnKBF)
-
EnKBF with full-obs. (de Wiljes+2018)
Error analysis of PO (prev.)
(Kelly+2014) D. Kelly et al. (2014), Nonlinearity, 27(10), 2579–2603.
Strong effect! (improve rank)
Theorem 1 (Kelly+2014)
exponential decay
initial error
Error analysis of PO (prev.)
variance of observation noise
uniformly in time
(Kelly+2014) D. Kelly et al. (2014), Nonlinearity, 27(10), 2579–2603.
Strong effect! (improve rank)
Corollary 1 (Kelly+2014)
Error analysis of ETKF (our)
(T.+2024) K. T. & T. Sakajo (2024), SIAM/ASA Journal on Uncertainty Quantification, 12(4), 1315–1335.
Theorem 2 (T.+2024)
Error analysis of ETKF (our)
(T.+2024) K. T. & T. Sakajo (2024), SIAM/ASA Journal on Uncertainty Quantification, 12(4), 1315–1335.
Corollary 2 (T.+2024)
Sketch: Error analysis of ETKF (our)
Strategy: estimate the changes in DA cycle
expanded
by dynamics
contract &
noise added
Applying Gronwall's lemma to the model dynamics,
+ {additional term}.
①
subtracting yields
From the Kalman update relation
prediction error
② contraction
③ obs. noise effect
estimated by op.-norm
Sketch: Error analysis of ETKF (our)
Key estimates in
Using Woodbury's lemma, we have
② contraction
③ obs. noise
Contraction (see the next slide)
assumption
Sketch: Error analysis of ETKF (our)
①
②
Key result
Sketch: Error analysis of ETKF (our)
estimate the change
(inflation & analysis)
(prediction)
Comparison
ETKF with multiplicative inflation (our)
PO with additive inflation (prev.)
accurate observation limit
Two approximate Gaussian filters (EnKF) are assessed in terms of the state estimation error.
due to
due to
Numerical result
Discussion
(T.+2024)
1. Partial observation
2. Small ensemble
3. Infinite-dimension
→ Issue 1, 2
→ Issue 1
→ Issue 1
partially solved (T.)
Discussion
Issue 1 (rank deficient)
② Error contraction in (II)
1. Partial observation
2. Small ensemble
Discussion
Issue 2 (non-symmetric)
③ bound of obs. noise
If non-symmetric, it doesn't hold in general.
Example
Discussion
Infinite-dimension
Theorem (Kelly+2014) for PO + add. infl.
Theorem (T.+2024) for ETKF + multi. infl.
( improve rank)
still holds
doesn't hold
(due to Issue 1: rank deficient)
※with full-observations
(ex: 2D-NSE with the Sobolev space)
Further result for partial obs.
The projected PO + add. infl. with partial obs. for Lorenz 96
Illustration (partial observations for Lorenz 96 equation)
○
○
×
○
○
×
○
○
×
○
○
×
○: observed
×: unobserved
the special partial obs. for Lorenz 96
Further result for partial obs.
The projected PO + add. infl. with partial obs. for Lorenz 96
Theorem 3 (T.)
The projected add. infl.:
key 2
key 1
Summary
What's DA?
Sequential state estimation using model and data.
Results
Error analysis of a practical DA algorithm (ETKF)
under an ideal setting (T.+2024).
Discussion
Two issues with generalizing our analysis.
(rank, non-symmetric)
References
- (T.+2024) K. T. & T. Sakajo, SIAM/ASA Journal on Uncertainty Quantification, 12(4), 1315–1335.
- (T.) Kota Takeda, Error Analysis of the Ensemble Square Root Filter for Dissipative Dynamical Systems, PhD Thesis, Kyoto University, 2025.
- (Kelly+2014) D. T. B. Kelly, K. J. H. Law, and A. M. Stuart (2014), Well-posedness and accuracy of the ensemble Kalman filter in discrete and continuous time, Nonlinearity, 27, pp. 2579–260.
- (Al-Ghattas+2024) O. Al-Ghattas and D. Sanz-Alonso (2024), Non-asymptotic analysis of ensemble Kalman updates: Effective dimension and localization, Information and Inference: A Journal of the IMA, 13.
- (Tong+2016a) X. T. Tong, A. J. Majda, and D. Kelly (2016), Nonlinear stability and ergodicity of ensemble based Kalman filters, Nonlinearity, 29, pp. 657–691.
- (Tong+2016b) X. T. Tong, A. J. Majda, and D. Kelly (2016), Nonlinear stability of the ensemble Kalman filter with adaptive covariance inflation, Comm. Math. Sci., 14, pp. 1283–1313.
- (Kwiatkowski+2015) E. Kwiatkowski and J. Mandel (2015), Convergence of the square root ensemble Kalman filter in the large ensemble limit, Siam-Asa J. Uncertain. Quantif., 3, pp. 1–17.
- (Mandel+2011) J. Mandel, L. Cobb, and J. D. Beezley (2011), On the convergence of the ensemble Kalman filter, Appl.739 Math., 56, pp. 533–541.
References
-
(de Wiljes+2018) J. de Wiljes, S. Reich, and W. Stannat (2018), Long-Time Stability and Accuracy of the Ensemble Kalman-Bucy Filter for Fully Observed Processes and Small Measurement Noise, Siam J. Appl. Dyn. Syst., 17, pp. 1152–1181.
-
(Burgers+1998) G. Burgers, P. J. van Leeuwen, and G. Evensen (1998), Analysis Scheme in the Ensemble Kalman Filter, Mon. Weather Rev., 126, 1719–1724.
-
(Bishop+2001) C. H. Bishop, B. J. Etherton, and S. J. Majumdar (2001), Adaptive Sampling with the Ensemble Transform Kalman Filter. Part I: Theoretical Aspects, Mon. Weather Rev., 129, 420–436.
-
(Anderson 2001) J. L. Anderson (2001), An Ensemble Adjustment Kalman Filter for Data Assimilation, Mon. Weather Rev., 129, 2884–2903.
-
(Dieci+1999) L. Dieci and T. Eirola (1999), On smooth decompositions of matrices, Siam J. Matrix Anal. Appl., 20, pp. 800–819.
-
(Reich+2015) S. Reich and C. Cotter (2015), Probabilistic Forecasting and Bayesian Data Assimilation, Cambridge University Press, Cambridge.
-
(Law+2015) K. J. H. Law, A. M. Stuart, and K. C. Zygalakis (2015), Data Assimilation: A Mathematical Introduction, Springer.
Appendix
Problems on numerical analysis
- Discretization/Model errors
- Dimension-reduction for dissipative dynamics
- Numerics for Uncertainty Quantification (UQ)
- Imbalance problems
1. Discretization/Model errors
Motivation
- Including discretization errors in the error analysis.
- Consider more general model errors and ML models.
- Error analysis with ML/surrogate models.
cf.: S. Reich & C. Cotter (2015), Probabilistic Forecasting and Bayesian Data Assimilation, Cambridge University Press.
Question
- Can we introduce the errors as random noises in models?
- Can minor modifications to the existing error analysis work?
2. Dimension-reduction for dissipative dynamics
Motivation
- Defining "essential dimension" and relaxing the condition:
- should be independent of discretization.
cf.: C. González-Tokman & B. R. Hunt (2013), Ensemble data assimilation for hyperbolic systems, Physica D: Nonlinear Phenomena, 243(1), 128-142.
Question
- Dissipativeness, dimension of the attractor?
- Does the dimension depend on time and state? (hetero chaos)
- Related to the SVD and the low-rank approximation?
Attractor
3. Numerics for UQ
Motivation
- Evaluating physical parameters in theory.
- Computing the Lyapunov spectrum of atmospheric models.
- Estimating unstable modes (e.g., singular vectors of Jacobian).
cf.: T. J. Bridges & S. Reich (2001), Computing Lyapunov exponents on a Stiefel manifold. Physica D, 156, 219–238.
F. Ginelli et al. (2007), Characterizing Dynamics with Covariant Lyapunov Vectors, Phys. Rev. Lett. 99(13), 130601.
Y. Saiki & M. U. Kobayashi (2010), Numerical identification of nonhyperbolicity of the Lorenz system through Lyapunov vectors, JSIAM Letters, 2, 107-110.
Question
- Can we approximate them only using ensemble simulations?
- Can we estimate their approximation errors?
4. Imbalance problems
Motivation
- Assimilating observations can break the balance relations inherited from dynamics. → numerical divergence
(e.g., the geostrophic balance and gravity waves)
- Explaining the mechanism (sparsity or noise of observations?).
- Only ad hoc remedies (Initialization, IAU).
cf.: E. Kalney (2003), Atmospheric modeling, data assimilation and predictability, Cambridge University Press. Hastermann et al. (2021), Balanced data assimilation for highly oscillatory mechanical systems, Communications in Applied Mathematics and Computational Science, 16(1), 119–154. |
Question
- Need to define a space of balanced states?
- Need to discuss the analysis state belonging to the space?
- Can we utilize structure-preserving schemes?
balanced states
Uniform error bounds of the ensemble square root filter for chaotic dynamics with multiplicative covariance inflation
By kotatakeda
Uniform error bounds of the ensemble square root filter for chaotic dynamics with multiplicative covariance inflation
The 1st MMS Workshop for Young Researchers (1h) https://www.math.kyoto-u.ac.jp/ja/event/conference/5525
- 5