State and Parameter Estimation from Observed Signal Increments
Nikolas N\"usken, Sebastian Reich, Paul J. Rozdeba

TL;DR
This paper develops ensemble Kalman-Bucy filter algorithms for simultaneous state and parameter estimation in continuous-time stochastic systems with correlated errors, demonstrated on complex multi-scale models.
Contribution
It introduces new ensemble Kalman-Bucy algorithms tailored for joint state and parameter estimation in correlated error settings.
Findings
Effective estimation in multi-scale stochastic models
Algorithms handle correlated model and measurement errors
Successful application to complex systems
Abstract
The success of the ensemble Kalman filter has triggered a strong interest in expanding its scope beyond classical state estimation problems. In this paper, we focus on continuous-time data assimilation where the model and measurement errors are correlated and both states and parameters need to be identified. Such scenarios arise from noisy and partial observations of Lagrangian particles which move under a stochastic velocity field involving unknown parameters. We take an appropriate class of McKean-Vlasov equations as the starting point to derive ensemble Kalman-Bucy filter algorithms for combined state and parameter estimation. We demonstrate their performance through a series of increasingly complex multi-scale model systems.
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Abstract
The success of the ensemble Kalman filter has triggered a strong interest in expanding its scope beyond classical state estimation problems. In this paper, we focus on continuous-time data assimilation where the model and measurement errors are correlated and both states and parameters need to be identified. Such scenarios arise from noisy and partial observations of Lagrangian particles which move under a stochastic velocity field involving unknown parameters. We take an appropriate class of McKean–Vlasov equations as the starting point to derive ensemble Kalman–Bucy filter algorithms for combined state and parameter estimation. We demonstrate their performance through a series of increasingly complex multi-scale model systems.
keywords:
parameter estimation, continuous-time data assimilation, ensemble Kalman filter, correlated noise, multi-scale diffusion processes
\pubvolume
xx \issuenum1 \articlenumber5
\historyReceived: date; Accepted: date; Published: date \TitleState and Parameter Estimation from Observed Signal Increments \AuthorNikolas Nüsken 1, Sebastian Reich 1* and Paul J. Rozdeba 1 \AuthorNamesNikolas Nüsken, Sebastian Reich and Paul J. Rozdeba
\corresCorrespondence: [email protected]; Tel.: +49-331-977-1859
1 Introduction
The research presented in this paper has been motivated by the state and parameter estimation problem for particles moving under a stochastic velocity field, with the measurements given by partial and noisy observations of their position increments. If the deterministic contributions to the velocity field are stationary, and the position increments of the moving particle are exactly observed, then one is led to a standard parameter estimation problem for stochastic differential equations (SDEs) (Kutoyants, 2004; Pavliotis, 2014). In Apte et al. (2007), this setting was extended to the case where the deterministic contributions to the velocity field themselves undergo a stochastic time evolution. Furthermore, while continuous-time observations of position increments are at the focus of the present study, the assimilation of discrete-time observations of particle positions has been investigated in Salman et al. (2006); Apte et al. (2008) under a so-called Lagrangian data assimilation setting for atmospheric fluid dynamics.
The assumption of exactly and fully observed position increments is not always realistic and the case of partial and noisy observations is at the center of the present study. Having access to partial and noisy observations of position increments leads to correlations between the measurement and model errors. The theoretical impact of such correlations on state and parameter estimation problems has been discussed, for example, in Simon (2006) in the context of linear systems, and in Bain and Crisan (2009) for nonlinear systems. One finds in particular that the appropriately adjusted data likelihood involves the gradient of log-densities, which is nontrivial from a computational perspective, and which prevents a straightforward application of standard Markov chain Monte Carlo (MCMC) or sequential Monte Carlo (SMC) methods Liu (2001).
In this paper, we instead follow an alternative Monte Carlo approach based on appropriately adjusted McKean–Vlasov filtering equations, an approach pioneered in Crisan and Xiong (2010) in the context of the standard state estimation problem for diffusion processes. We recall that the notion of McKean–Vlasov equations, first studied in McKean (1966), characterises a class of SDEs for which their right-hand side depends on the law of the process itself. We rely on a particular formulation of such McKean–Vlasov filtering equations, the so-called feedback particle filters (Yang et al., 2013), utilising stochastic innovation processes (Reich, 2019). Our proposed Monte Carlo formulation avoids the need for estimating log-densities, and can be implemented in a numerically robust manner relying on a generalised ensemble Kalman–Bucy filter approximation applied to an extended state space formulation (Majda and Harlim, 2012). The ensemble Kalman–Bucy filter (Bergemann and Reich, 2012; Taghvaei et al., 2017) has been introduced previously as an extension of the popular ensemble Kalman filter Majda and Harlim (2012); Law et al. (2015); Reich and Cotter (2015) to continuous-time data assimilation under the assumption of uncorrelated measurement and model errors.
We apply the proposed algorithms to a series of state and parameter estimation problems of increasing complexity. First, we study the state and parameter estimation problem for an Ornstein–Uhlenbeck process Pavliotis (2014). Two further experiments investigate the behaviour of the filters for reduced model equations, with the data being collected from underlying multi-scale models. There we distinguish between the averaging and homogenisation scenarios (Pavliotis and Stuart, 2008). Finally, we also look at nonparametric drift estimation Apte et al. (2007), and parameter estimation for the stochastic heat equation Altmeyer and Reiß (2019).
2 Mathematical problem formulation
We consider the time evolution of a random state variable in -dimensional state space, , as prescribed by an SDE of the form
[TABLE]
for time , with the drift function depending on unknown parameters . Model errors are represented through standard -dimensional Brownian motion , , and a matrix . We also introduce the associated model error covariance matrix . We will generally assume that the initial condition is fixed, that is, a.s. for given . In terms of a more specific example, one can think of denoting the position of a particle at time moving in dimensional space under the influence of a stochastic velocity field, with deterministic contributions given by and stochastic perturbations by . In the case , the SDE (1) reduces to an ordinary differential equation with given initial condition .
We assume throughout this paper that (1) possesses unique, strong solutions for all parameter values . See, for example, Pavliotis (2014) for necessary conditions on the drift function . The distribution of is denoted by , which we also abbreviate by . We use the same notation for measures and their Lebesgue densities, provided they exist.
{Example}
A wide class of drift functions can be written in the form
[TABLE]
where is a known drift function, the , , denote appropriate basis functions, and the vector contains the unknown parameters of the model. The family of basis functions, which we collect in a matrix-valued function , could arise from a finite-dimensional truncation of some appropriate Hilbert space . See, for example, Papaspiliopoulos et al. (2012) for computational approaches to nonparametric drift estimation using a Galerkin approximation in , where the become finite element basis functions. Furthermore, the expansion coefficients could be made time-dependent by letting them evolve according to some system of differential equations arising, for example, from the discretisation of an underlying partial differential equation with solutions in . See Apte et al. (2007) for specific examples of such a setting. While the present paper focuses on stationary drift functions, that is, the parameters are time-independent, the results from Sections 3 and 5, respectively, can easily be extended to the non-stationary case where the parameters themselves satisfy given evolution equations.
Data and an observation model are required in order to perform state and parameter estimation for SDEs of the form (1). In this paper, we assume that we observe partial and noisy increments of the signal , that is,
[TABLE]
for in the observation interval , , where is a given linear operator, denotes standard -dimensional Brownian motion with and is a covariance matrix. We introduce the observation map
[TABLE]
for later use. Unless , we find that the model error in (1) and the total observation error
[TABLE]
in (3) are correlated. The impact of correlations between the model and measurement errors on the state estimation problem have been discussed by Simon (2006); Bain and Crisan (2009). Furthermore, such correlations require adjustments to sequential estimation methods (Särkkä, 2013; Law et al., 2015; Reich and Cotter, 2015) which are the main focus of this paper. We assume throughout this paper that the covariance matrix
[TABLE]
of the observation error (5) is invertible.
The special case and leads to a pure parameter estimation problem, which has been extensively studied in the literature in the settings of maximum likelihood and Bayesian estimators (Kutoyants, 2004; Pavliotis, 2014). We will provide a reformulation of the Bayesian approach in the form of McKean–Vlasov equations in the parameters, based on the results in Crisan and Xiong (2010); Yang et al. (2013) in Section 3.
If , then (1) and (3) lead to a combined state and parameter estimation problem with correlated noise terms. We will first discuss the impact of this correlation on the pure state estimation problem in Section 4 assuming that the parameters of the problem are known. Again, we will derive appropriate McKean–Vlasov equations in the state variables. Our key contribution is a formulation that avoids the need for log-density estimates, and can be put into an appropriately generalised ensemble Kalman–Bucy filter approximation framework (Bergemann and Reich, 2012; Taghvaei et al., 2017). We also formally demonstrate that the McKean–Vlasov filter equation reduces to in the limit and , a property which is less straightforward to demonstrate for filter formulations involving log-densities.
These McKean–Vlasov equations can be generalised to the combined state and parameter estimation problem via an augmentation of state space (Majda and Harlim, 2012) in Section 5. Given the results from Section 4, such an extension is rather straightforward.
The numerical experiments in Section 6 rely exclusively on the generalised ensemble Kalman–Bucy filter approximation to the McKean–Vlasov equations, which are easy to implement and yield robust and accurate numerical results.
3 Parameter estimation from noiseless data
In this section, we treat the simpler Bayesian parameter estimation problem which arises from setting and in (3), that is, . This leads to and, furthermore, for all , provided which we assume throughout this paper. The requirement that is invertible requires that has rank , that is, in (1). The data likelihood
[TABLE]
thus follows from the observation model with additive Brownian noise in (3). Given a prior distribution for the parameters, the resulting posterior distribution at any time is
[TABLE]
according to Bayes’ theorem Bain and Crisan (2009). Here, we have introduced the shorthand
[TABLE]
for the expectation of with respect to . It is well-known that the posterior distributions satisfy the stochastic partial differential equation
[TABLE]
with time-dependent observation map
[TABLE]
where is a compactly supported smooth test function, and again denoting the expectation of with respect to . See Bain and Crisan (2009) for a detailed discussion. Equation (10) constitutes a special instance of the well-known Kushner–Stratonovitch equation from time-continuous filtering Bain and Crisan (2009).
3.1 Feedback particle filter
We now state a McKean–Vlasov reformulation of the Kushner–Stratonovitch equation (10) as a special instance of the feedback particle filter of Yang et al. (2013); Reich (2019). The key idea is to formulate a stochastic differential equation in the parameters in which they are treated as time-dependent random variables. We introduce the notation for these, and require that the law of coincide with (8) for , that is, with the solution to (10).
{Lemma}
[Feedback particle filter]
Consider the McKean–Vlasov equations
[TABLE]
where the matrix-valued Kalman gain satisfies
[TABLE]
the innovation process can be chosen to be given by either
[TABLE]
or
[TABLE]
and
[TABLE]
Then, the distribution coincides with the solution to (10), provided that the initial distributions agree. In other words, for all .
Throughout this paper, we write (12) in the more compact Stratonovitch form
[TABLE]
where the Stratonovitch interpretation is to be applied only to in , while the explicit time-dependence of remains in its Itô interpretation. It should be noted that the matrix-valued function is not uniquely defined by the PDE (44). Indeed, provided solves (44), is also a solution whenever . As discussed in Taghvaei et al. (2017), the minimiser over all suitable with respect to a kinetic energy-type functional is of the form
[TABLE]
for a vector of potential functions , . Inserting (18) into (44) leads to elliptic partial differential equations (often referred to as Poisson equations),
[TABLE]
understood componentwise, where the centering condition makes the solution unique under mild assumptions on , see Laugesen et al. (2015). Finally, (15) yields a particularly appealing formulation, since it is based on a direct comparison of with a random realisation of the right hand side of the SDE (1), given a parameter value and a realisation of the noise term . This fact will be explored further in Section 4.
{Remark}
For clarity, let us repeat equations (44) and (18) in their index forms:
[TABLE]
[TABLE]
3.2 Ensemble Kalman–Bucy filter
Let us now assume that the initial distribution is Gaussian, and that is linear in the unknown parameters such as in (2). Then, the distributions remain Gaussian for all times with mean and covariance matrix . The elliptic PDE (44) is solved by the parameter-independent Kalman gain matrix
[TABLE]
and one obtains the McKean–Vlasov formulation
[TABLE]
of the Kalman–Bucy filter, with the innovation process defined by either
[TABLE]
or
[TABLE]
Note that the Stratonovitch formulation (17) reduces to the standard Itô interpretation, since no longer depends explicitly on .
The McKean–Vlasov equations (23) can be extended to nonlinear, non-Gaussian parameter estimation problems by generalising the parameter-independent Kalman gain matrix (22) to
[TABLE]
Clearly, the gain (26) provides only an approximation to the solution of (44). However, such approximations have become popular in nonlinear state estimation in the form of the ensemble Kalman filter (Law et al., 2015; Reich and Cotter, 2015), and we will test its suitability for parameter estimation in Section 6.
Numerical implementations of the proposed McKean–Vlasov approaches rely on Monte–Carlo approximations. More specifically, given samples , , from the initial distribution , we introduce the interacting particle system
[TABLE]
where the innovation processes are defined by either
[TABLE]
or, alternatively,
[TABLE]
and , , denote independent -dimensional Brownian motions. For , we will use the parameter-independent empirical Kalman gain approximation
[TABLE]
in our numerical experiments, which leads to the so-called ensemble Kalman–Bucy filter (Bergemann and Reich, 2012; Taghvaei et al., 2017). Note that provides an unbiased estimator of .
Finally, a robust and efficient time-stepping procedure for approximating , , is provided in (Amezcua et al., 2014; de Wiljes et al., 2018; Blömker et al., 2018). Denoting the approximations at time by , , we obtain
[TABLE]
with step size , empirical covariance matrices
[TABLE]
and innovation increments given by either
[TABLE]
or
[TABLE]
Here we have used the abbreviations , , and .
While the feedback particle formulation (17) and its ensemble Kalman–Bucy filter approximation (31) are special cases of already available formulations, they provide the starting point for our novel McKean–Vlasov equations and their numerical approximation of the combined state and parameter estimation problem with correlated measurement and model errors, which we develop in the following two sections.
4 State estimation for noisy data
We return to the observation model (3) with and general . The pure state estimation problem is considered first, that is, in (1).
Using , given by (5), and defined by
[TABLE]
with the total measurement error covariance matrix given by (6), we find that
[TABLE]
and the covariations Pavliotis (2014) satisfy
[TABLE]
Hence (1) and (3) can be rewritten as follows:
[TABLE]
where and denote mutually independent standard Brownian motions of dimension and , respectively. These equations correspond exactly to the correlated noise example from (Bain and Crisan, 2009, Section 3.8). Furthermore, and lead to , , and, hence, .
A straightforward application of the results from (Bain and Crisan, 2009, Section 3.8) yields the following statement:
{Lemma}
[Generalised Kushner–Stratonovich equation] The conditional expectations satisfy
[TABLE]
where111We use the notation .
[TABLE]
is the generator of (1), denotes the observation map, and is a compactly supported smooth function.
For the convenience of the reader, we present an independent derivation in Appendix A. We note that (39) also arises as the Kushner–Stratonovitch equations for an SDE model (1) with observations satisfying the observation model
[TABLE]
where denotes -dimensional Brownian motion independent of the Brownian motion in (1). Here we have used that . This reinterpretation of our state estimation problem in terms of uncorrelated model and observation errors and modified observation map
[TABLE]
allows one to apply available MCMC and SMC methods for continuous-time filtering and smoothing problems. See, for example, Law et al. (2015). However, there are two major limitations of such an approach. First, it requires approximating the gradient of the log-density. Second, the modified observation model (41) is not well-defined in the limit and , since the density collapses to a Dirac delta function under the given initial condition a.s.
In order to circumvent these complications, we develop an alternative approach based on an appropriately modified feedback particle filter formulation in the following subsection.
4.1 Generalised feedback particle filter formulation
While it is clearly possible to apply the standard feedback particle filter formulations using (41), the following alternative formulation avoids the need for approximating the gradient of the log-density.
{Lemma}
[Feedback particle filter with correlated innovation] Consider the McKean–Vlasov equation
[TABLE]
where the gain solves
[TABLE]
with observation map . The function is given by
[TABLE]
and the innovation process by
[TABLE]
Here, and denote mutually independent -dimensional and -dimensional Brownian motions, respectively. Then, coincides with the solution to (39), provided that the initial distributions agree.
It should be stressed that in (43) and (46) denote the same Brownian motion, resulting in correlations between the innovation process and model noise.
Proof.
In this proof the Einstein summation convention over repeated indices is employed, noting that (44) takes the form
[TABLE]
We begin by writing (43) in its Itô-form,
[TABLE]
where
[TABLE]
Here we have used that the covariation between and satisfies
[TABLE]
and furthermore as well as .
For a smooth compactly supported test function , Itô’s formula implies
[TABLE]
where the covariation process is given by
[TABLE]
Our aim is to show that coincides with as defined by the Kushner–Stratonovich equation (39). To this end, we insert (48) and (52) into (51) and take the conditional expectation, arriving at
[TABLE]
recalling that the generator has been defined in (40). Under the assumption that satisfies (44), the two equations (39) and (53) coincide. Indeed,
[TABLE]
implies
[TABLE]
and the -contributions agree. To verify the same for the -contributions, we use (44) to obtain
[TABLE]
Finally, collecting terms in (53) and (56) and applying (55) to the remaining -contribution, i.e. , leads to the desired result. ∎
We note that the correlation between the innovation process and the model error leads to a correction term in (43) which cannot be subsumed into a Stratonovitch correction, in contrast to the standard feedback particle filter formulation (17).
{Remark}
Assuming that there exist potential functions , , solving the Poisson equation(s) (19) (with being replaced by ), (44) can be solved by requiring
[TABLE]
thus generalising (18).
{Remark}
If we set , , and in (43), then one obtains
[TABLE]
since vanishes, and all other terms in (43) cancel each other out. If, furthermore, a.s., then for all , which in turn justifies our assumption that the gain is independent of the state variable. Hence, the McKean–Vlasov formulation (43) reproduces the exact reference trajectory in the case of no measurement errors and perfectly known initial conditions.
We develop a simplified version of the feedback particle filter formulation (43) for linear SDEs and Gaussian distributions in the following subsection, which will form the basis of the generalised ensemble Kalman–Bucy filter put forward in the follow-up Section 4.3.
4.2 Generalised Kalman–Bucy filter
Let us assume that with , that is, equations (1) and (3) take the form
[TABLE]
with initial conditions drawn from a Gaussian distribution. In this case stays Gaussian for all , i.e. with , . Equations (19) can be solved uniquely by , and thus the McKean–Vlasov equations for the feedback particle filter (43) reduce to
[TABLE]
with the innovation process (46) leading to
[TABLE]
We take the expectation in (60)–(61) and end up with
[TABLE]
Defining , we see that
[TABLE]
Next we use
[TABLE]
and to obtain, after some calculations,
[TABLE]
Hence we have shown that our McKean–Vlasov formulation (60) agrees with the standard Kalman–Bucy filter equations for the mean and the covariance matrix in the correlated noise case Simon (2006).
4.3 Ensemble Kalman–Bucy filter
The McKean–Vlasov equations (60) for linear systems and Gaussian distributions suggest approximating the feedback particle filter formulation (43) for nonlinear systems by
[TABLE]
where the innovation process given by (46) as before. In other words, we approximate the gain matrix in (43) by the state independent term with the covariance matrix defined by
[TABLE]
where denotes the law of .
We can now generalise the ensemble Kalman–Bucy filter formulation (31) for the pure parameter estimation problem to the state estimation problem with correlated noise. We assume that initial state values have been sampled from an initial distribution or, alternatively, for all in case the initial condition is known exactly. These state values are then propagated under the time-stepping procedure
[TABLE]
with , step size , empirical covariance matrices
[TABLE]
and innovation increments given by
[TABLE]
The McKean–Vlasov equations of this section form the basis of the methods proposed for the combined state and parameter estimation problem to be considered next.
5 Combined state and parameter estimation
We now return to the combined state and parameter estimation problem and consider the augmented dynamics
[TABLE]
with observations (3) as before. The initial conditions satisfy a.s. and . Let us introduce the extended state-space variable . In terms of , the equations (71) and (3) take the form
[TABLE]
with
[TABLE]
Thus we end up with an augmented state estimation problem of the general structure considered in detail in Section 4 already. Below we provide details on some of the necessary modifications.
5.1 Feedback particle filter formulation
The appropriately extended feedback particle filter equation (43) leads to
[TABLE]
where (46) takes the form
[TABLE]
with observation map (4) and the correction is given by (45) with replaced by and by . In the Poisson equation(s) (19), is replaced by denoting the joint density of . We also stress that becomes a function of and and we distinguish between gradients with respect to and using the notation and , respectively.
Numerical implementations of the extended feedback particle filter are demanding due to the need of solving the Poisson equation(s) (19). Instead we again rely on the ensemble Kalman–Bucy filter approximation, which we describe next.
5.2 Ensemble Kalman–Bucy filter
We approximate the joint density of by an ensemble of particles
[TABLE]
that is,
[TABLE]
where denotes the Dirac delta function centred at . The initial ensemble satisfies for all , and the initial parameter values are independent draws from the prior distribution .
At the same time, we make the approximation when dealing with the Kalman gain of the feedback particle filter. Here the empirical mean has components
[TABLE]
and the joint empirical covariance matrix is given by
[TABLE]
As in Section 4.3, the solution to (19) can be approximated by
[TABLE]
where the covariance matrices and are finally estimated by their empirical counterparts
[TABLE]
with defined by
[TABLE]
Summing everything up, we obtain the following generalised ensemble Kalman–Bucy filter equations
[TABLE]
where the innovations are given by
[TABLE]
and and denote independent -dimensional and -dimensional, respectively, Brownian motions for .
The interacting particle equations (83) can be time-stepped along the lines discussed in Section 4.3 for the pure state estimation formulation of the ensemble Kalman–Bucy filter.
6 Numerical results
We now apply the generalised ensemble Kalman–Bucy filter formulation (83) with innovation (84) to five different model scenarios.
6.1 Parameter estimation for the Ornstein–Uhlenbeck process
Our first example is provided by the Ornstein–Uhlenbeck process
[TABLE]
with unknown parameter , and known initial condition . We assume an observation model of the form (3) with , and a measurement error taking values , , and . The model error variance is set to either or . Except for the case a combined state and parameter estimation problem is to be solved. We implement the ensemble Kalman–Bucy filter (83) with innovation (84), step size , and ensemble size . The data is generated using the Euler–Maruyama method applied to (85), with and integrated over a time-interval with the same step size. The prior distribution for the parameter is Gaussian with mean and variance . The results can be found in Figure 1. We find that the ensemble Kalman–Bucy filter is able to successfully identify the unknown parameter under all tested experimental settings, except for the largest measurement error case where . There, a small systematic offset of the estimated parameter value can be observed. One can also see that the variance in the parameter estimate monotonically decreases in time in all cases, while the variance in the state estimates approximately reaches a steady state.
6.2 Averaging
Consider the equations
[TABLE]
from Pavliotis and Stuart (2008) for , and initial condition , . The reduced equations in the limit are given by (85), with parameter value
[TABLE]
and initial condition . The reduced dynamics corresponds to a (stable) Ornstein–Uhlenbeck process for . We wish to estimate the parameter from observed increments
[TABLE]
where the sequence of is obtained by time-stepping (86) using the Euler–Maruyama method with a step size . We set , (so that ), , and in our experiments. The measurement noise is set to or (pure parameter estimation).
We implement the ensemble Kalman–Bucy filter (83) with innovation (84), step size , and ensemble size for the reduced equations (87). The data is generated from an Euler–Maruyama discretization of (86) with the same step size. We also investigate the effect of subsampling the observations for by solving (86) with step size and storing only every tenth solution , while the reduced equations and the ensemble Kalman–Bucy filter equations are integrated with . The results are shown in Figure 2. Figure 3 shows the results for the same experiments repeated with a smaller ensemble size of . We find that the smaller ensemble size leads to more noisy estimates for the variance in and a faster decay of the variance in , but the estimated parameter values are equally well converged. Subsampling does not lead to significant changes in the estimated parameter values. This is in contrast to the example considered next.
We finally mention Harlim (2017) for alternative approaches to sequential estimation in the context of averaging using however different assumptions on the data.
6.3 Homogenisation
In this example, the data is produced by integrating the multi-scale SDE
[TABLE]
with parameter values , , , and initial condition , . Here, denotes standard Brownian motion. The equations are discretised with step size , and the resulting increments (88) are stored over a time interval . See Krumscheid et al. (2011) for more details.
According to homogenisation theory, the reduced model is given by (85) with , and we wish to estimate the parameter from the data produced according to (88). It is known that a standard maximum likelihood estimator (MLE) given by
[TABLE]
leads to in the limit and the observation interval . This MLE corresponds to and in our extended state space formulation of the problem. Subsampling can be achieved by choosing an appropriate time-step in the ensemble Kalman–Bucy filter equations and a corresponding subsampling of the data points in (88). We used and , respectively. The results can be found in Figure 4. It can be seen that only the larger subsampling leads to a correct estimate of the parameter . This is in line with known results for the maximum likelihood estimator (90). See Krumscheid et al. (2011) and references therein.
6.4 Nonparametric drift and state estimation
We consider nonparametric drift estimation for one-dimensional SDEs over a periodic domain in the setting considered from a theoretical perspective in (van Waaij and van Zanten, 2016). There, a zero-mean Gaussian process prior is placed on the unknown drift function, with inverse covariance operator
[TABLE]
The integer parameter sets the regularity of the process, whereas control its characteristic correlation length and stationary variance.
Spatial discretization of the problem is carried out by first defining a grid of evenly spaced points on the domain, at locations , . The drift function is projected onto compactly supported functions centred at these points, which are piecewise linear with
[TABLE]
and linear interpolation is used to define a drift function for all , that is, it is of the form (2) with . In this example, we set . Sample realisations, as well as the reference drift , can be found in Figure 5(a).
Data is generated by integrating the SDE (1) with drift forward in time from initial condition and with noise level , using the Euler–Maruyama discretisation with step size over one million time-steps. The spatial distribution of the solutions is plotted in Figure 5(b). The data is then given by
[TABLE]
with . Data assimilation is performed using the time-discretised ensemble Kalman–Bucy filter equations (83) with innovation (84), ensemble size , and step size .
The final estimate of the drift function (ensemble mean) and the ensemble of drift functions can be found in Figure 5(c). Figure 5(d) displays the ensemble of state estimates and the value of the reference solution at the final time. We find that the ensemble Kalman–Bucy filter is able to successfully estimate the drift function and the model states. Further experiments reveal that the drift function can only be identified for sufficiently small measurement errors.
6.5 SPDE parameter estimation
Consider the stochastic heat equation on the periodic domain , given in conservative form by the stochastic partial differential equation (SPDE)
[TABLE]
where is space-time white noise. With constant , this SPDE reduces to
[TABLE]
In this example, we examine the estimation of from incremental measurements of a locally averaged quantity that arises naturally in a standard finite volume discretisation of (95).
To discretise the system, one first defines around grid points on a regular grid, separated by distances , as
[TABLE]
The conservative (drift) term in (94) reduces to
[TABLE]
where , etc. The following standard finite difference approximations
[TABLE]
yield the -dimensional SDE
[TABLE]
for constant , where are independent one-dimensional Brownian motions in time.
Following recent results from Altmeyer and Reiß (2019) we consider the case of estimation of a constant value from measurements at a fixed location/index . The data trajectory is thus given by
[TABLE]
where is a scalar and is a standard Brownian motion in one dimension. We perform numerical experiments in which the initial state is set to zero for all indices and the prior on the unknown parameter is uniform over the interval .
The increment data is generated by first integrating (95) forward in time from the known initial condition for all . The equation is discretised in time using the Euler-Maruyama method. It is known that is required for stability of the Euler–Maruyama discretisation; we use the much smaller time step . The solution is sampled with this same time step, and increment measurements are approximated at time by setting the measurement noise level to zero in (100), resulting in
[TABLE]
Note that the associated model error in (1) is given by and the matrix in (3) projects the vector of state increments onto a single component with index . Simulations are performed over the time-interval . The results can be found in Figure 6(a). We also compute the model evidence for a sequence of parameter values based on a standard Kalman–Bucy filter Simon (2006) for the associated linear state estimation problem. See Figure 6(b). Both approaches agree with the reference value .
6.6 Discussion
The presented results demonstrate that the proposed methodology can be applied to a broad range of continuous-time state and parameter estimation problems with correlated measurement and model errors. Alternatively, one could have employed standard SMC or MCMC methods utilising the modified observation model (41). However, such implementations require the approximation of the additional term which is nontrivial if only samples from are available. Furthermore, the limiting behaviour of such implementations in the limit and (pure parameter estimation problem) is unclear. The proposed generalised ensemble Kalman–Bucy filter avoids these issues and is easy to implement. In fact, the only differences to the standard ensemble Kalman–Bucy filter formulation of Bergemann and Reich (2012) consist in the additional term in the Kalman gain and a correlation between the stochastic innovation process and the model error.
7 Conclusions
In this paper, we have derived McKean–Vlasov equations for combined state and parameter estimation from continuously observed state increments. An approximate and robust implementation of these McKean-Vlassov equations in the form of a generalised ensemble Kalman–Bucy filter has been provided and applied to a range of increasingly complex model systems. Future work will address the treatment of temporally correlated measurement and model errors as well as a rigorous analysis of these McKean–Vlasov equations in a multi-scale context and in the context of nonparametric drift estimation.
\authorcontributions
Methodology, N.N. and S.R.; software, S.R. and P.R.; validation, N.N., S.R. and P.R.; writing–original draft preparation, N.N., S.R.; writing–review and editing, N.N., S.R. and P.R.
\funding
This research has been partially funded by Deutsche Forschungsgemeinschaft (DFG) through grants CRC 1294 ‘Data Assimilation’ (project A06) and CRC 1114 ‘Scaling Cascades’ (project A02).
\conflictsofinterest
The authors declare no conflict of interest. The funders had no role in the design of the study; in the collection, analyses, or interpretation of data; in the writing of the manuscript, or in the decision to publish the results.
\appendixtitles
no
Appendix A The filtering equations for correlated noise
In this appendix we outline a derivation of the Kushner-Stratonovich equation (39) for the signal-observation dynamics given by (38). In fact, we only compute the evolution equation (termed modified Zakai equation) for the unnormalised filtering distribution , where the likelihood is given by
[TABLE]
Obtaining the Kushner-Stratonovich formulation is then standard, applying Itô’s formula to the Kallianpur-Striebel formula , see (Bain and Crisan, 2009, Chapter 3). The following result is in agreement with the corollaries 3.39 and 3.40 in Bain and Crisan (2009).
{Lemma}
The modified Zakai equation is given by
[TABLE]
where the generator has been defined in (40).
Proof.
For convenience, let us define the process
[TABLE]
where satisfies (38b). From we see that
[TABLE]
hence the likelihood takes the form
[TABLE]
satisfying the SDE
[TABLE]
For an arbitrary smooth compactly supported test function Itô’s formula implies
[TABLE]
where satisfies (38a). For the covariation process we obtain
[TABLE]
using . Furthermore, , which follows from the definition of the stochastic contributions in (38a).
We now apply the conditional expectation to (108). Noticing that
[TABLE]
the result follows from (107). ∎
\reftitle
References
\externalbibliography
yes
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1Kutoyants (2004) Kutoyants, Y. Statistical inference for ergodic diffusion processes ; Springer–Verlag: New York, 2004.
- 2Pavliotis (2014) Pavliotis, G. Stochastic processes and applications ; Springer–Verlag: New York, 2014.
- 3Apte et al. (2007) Apte, A.; Hairer, M.; Stuart, A.; Voss, J. Sampling the posterior: An approach to non-Gaussian data assimilation. Physica D Nonlinear Phenomena 2007 , 230 , 50–64.
- 4Salman et al. (2006) Salman, H.; Kuznetsov, L.; Jones, C.; Ide, K. A method for assimilating Lagrangian data into a shallow-water-equation ocean model. Mon. Wea. Rev. 2006 , 134 , 1081–1101.
- 5Apte et al. (2008) Apte, A.; Jones, C.; Stuart, A. A Bayesian approach to Lagrangian data assimilation. Tellus A 2008 , 60 , 336–347.
- 6Simon (2006) Simon, D. Optimal State Estimation ; Wiley: Hoboken, New Jersey, 2006.
- 7Bain and Crisan (2009) Bain, A.; Crisan, D. Fundamentals of stochastic filtering ; Springer–Verlag: New York, 2009.
- 8Liu (2001) Liu, J. Monte Carlo strategies in scientific computing ; Springer-Verlag: New York, 2001.
