N EURAL J UMP O RDINARY D IFFERENTIAL E QUATIONS
Anonymous authors
Paper under double-blind review
A BSTRACT
Combinations of neural ODEs with recurrent neural networks (RNN), like GRUODE-Bayes or ODE-RNN are well suited to model irregularly observed time series.
While those models outperform existing discrete-time approaches, no theoretical guarantees for their predictive capabilities are available.
Assuming that the irregularly-sampled time series data originates from a continuous stochastic process, the L 2 -optimal online prediction is the conditional expectation given the currently available information.
We introduce the neural jump ODE (NJ-ODE) that provides a data-driven approach to learn, continuously in time, the conditional expectation of a stochastic process.
Our approach models the conditional expectation between two observations with a neural ODE and jumps whenever a new observation is made.
We deﬁne a novel training framework, which allows us to prove theoretical guarantees for the ﬁrst time.
In particular, we show that the output of our model converges to the L 2 -optimal prediction.
We provide experiments showing that the theoretical results also hold empirically.
Moreover, we experimentally show that our model outperforms the baselines in more complex learning tasks and give comparisons on real-world datasets.
I NTRODUCTION
Stochastic processes are widely used in many ﬁelds to model time series that exhibit a random behaviour.
In this work, we focus on processes that can be expressed as solutions of stochastic differential equations (SDE) of the form
[Equation]
with certain assumptions on the drift µ and the diffusion σ .
With respect to the L 2 -norm, the best prediction of a future value of the process is provided by the conditional expectation given the current value.
If the drift and diffusion are known or a good estimation is available, the conditional expectation can be approximated by a Monte Carlo (MC) simulation.
However, since µ and σ are usually unknown, this approach strongly depends on the assumptions made on their parametric form.
A more ﬂexible approach is given by neural SDEs , where the drift µ and diffusion σ are modelled by neural networks (Tzen & Raginsky, 2019; Li et al., 2020; Jia & Benson, 2019).
Nevertheless, modelling the diffusion can be avoided if one is only interested in forecasting the behaviour instead of sampling new paths.
An alternative widely used approach is to use Recurrent Neural Networks (RNN), where a neural network dynamically updates a latent variable with the observations of a discrete input time-series.
RNNs are successfully applied to tasks for which time-series are regularly sampled, as for example speech or text recognition.
However, often observations are irregularly observed in time.
The standard approach of dividing the time-line into equally-sized intervals and imputing or aggregating observations might lead to a signiﬁcant loss of information (Rubanova et al., 2019).
Frameworks that overcome this issue are the GRU-ODE-Bayes (Brouwer et al., 2019) and the ODE-RNN (Rubanova et al., 2019), which combine a RNN with a neural ODE (Chen et al., 2018).
In standard RNNs, the hidden state is updated at each observation and constant in between.
Conversely, in the GRU-ODEBayes and ODE-RNN framework, a neural ODE is trained to model the continuous evolution of the hidden state of the RNN between two observations.
While GRU-ODE-Bayes and ODE-RNN both provide convincing empirical results, they lack thorough theoretical guarantees.
Contribution.
In this paper, we introduce a mathematical framework to precisely describe the problem statement of online prediction of a stochastic process with temporal irregular observations.
Based on this rigorous mathematical description, we introduce the Neural Jump ODE (NJ-ODE).
The model architecture is very similar to the one of GRU-ODE-Bayes and ODE-RNN, however we introduce a novel training framework, which in contrast to them allows us to prove convergence guarantees for the ﬁrst time.
Moreover, we demonstrate empirically the capabilities of our model.
Precise problem formulation.
We emphasize that a precise deﬁnition of all ingredients is needed, to be able to show theoretical convergence guarantees, which is the main purpose of this work.
Since the objects of interest are stochastic processes, we use tools from probability theory and stochastic calculus.
To make the paper more readable and comprehensible also for people without background in probability theory and stochastic calculus, the precise formulations and demonstrations of all claims are given in the appendix, while the main part of the paper focuses on giving well understandable heuristics.
P ROBLEM STATEMENT
The problem we consider in this work, is the online forecasting of temporal data.
In particular, we assume that we make observations of a stochastic process described by the stochastic differential equation (SDE)
[Equation]
at irregularly-sampled time points.
Between those observation times, we want to predict the stochastic process, based only on the observations that we made previously in time, excluding the possibility to interpolate observations.
In other words, online forecasting means, that after each observation we want to extrapolate the current and the previous observations into the future until we make the next observation.
The time at which the next observation will be made is random and assumed to be independent of the stochastic process itself.
More precisely, we suppose to observe N realisations of the R d X -dimensional stochastic process X deﬁned in (1).
Each realisation j is observed at n j random observation times t j 1 , . . .
, t jn j ∈ [0 , T ] with values x j 1 , . .
, x jn j ∈ R d X .
We assume that all coordinates of x ji are observed.
We are interested in forecasting how a new realization evolves in time, such that our predictions minimize the expected squared distance ( L 2 -metric) to the true unknown value.
The optimal prediction, i.e. the L 2 -minimizer, is the conditional expectation.
Given that the value of the realization at time t is x t , we are therefore interested in estimating the function
[Equation]
To estimate f ( x,t ) ( s ) we make use of the N realizations that were observed before.
The following example illustrates the considered problem.
Example.
Several blood values and hormone values of participants in a health study are measured at body checks multiple times during one year.
Each participant has his checks depending on the schedule of the study and his own schedule, hence the observation dates are irregular in time and exhibit some randomness which is independent of the measured values.
During the year, participant 1 has his n 1 = 4 checks at days ( t 11 , t 12 , t 13 , t 14 ) =
(7 , 142 , 227 , 355) where the values ( x 11 , x 12 , x 13 , x 14 ) =
78 , 0 . 81) are measured.
Participant 2 only has n 2 = 2 checks at days ( t 21 , t 22 ) =
(89 , 257) where the values ( x 21 , x 22 ) = (0 . 56 , 0 . 63) are measured.
Similarly, the j -th patient has his n j checks at the dates ( t j 1 , . . .
, t jn j ) and has the measured values ( x j 1 , . . .
, x jn j ) .
Based on this data, we want to forecast the blood and hormone values of patients in hospitals.
In particular, for a patient with measured values x 1 at time t 1 , we want to predict what his values will likely be at any time t 1 + s > t 1 .
Importantly, we do not only focus on predicting the values at some t 2 > t 1 , but we want to know the entire evolution of the values.
B ACKGROUND
Recurrent Neural Network.
The input to a RNN is a discrete time series of observations { x t 1 , · · · , x t n } .
At each time t i +1 , the latent variable h is updated using the previous latent variable h
i and the input x i +1 as
[Equation]
where RNNCell is a neural network.
Neural Ordinary Differential Equation.
Neural ODEs (Chen et al., 2018) are a family of continuous-time models deﬁning a latent variable h t :
= h ( t ) to be the solution to an ODE initial-value problem (IVP):
[Equation]
where f ( · , · , θ )
=
f θ is a neural network with weights θ .
Therefore, the latent variables can be updated continuously by solving this ODE (4).
We can emphasize the dependence of h t on a numerical ODE solver by rewriting (4) as
[Equation]
ODE-RNN.
ODE-RNN (Rubanova et al., 2019) is a mixture of a RNN and a neural ODE.
In contrast to a standard RNN, we are not only interested in an output at the observation times t i , but also in between those times.
In particular, we want to have an output stream that is generated continuously in time.
This is achieved by using a neural ODE to model the latent dynamics between two observation times, i.e. for t
i < t < t i +1 the latent variable is deﬁned as in (4) and (5), with h t 0 and t 0 replaced by h t i and t i .
At the next observation time t i +1 , the latent variable is then updated by a RNN.
Rubanova et al.
(2019) write this as
[Equation]
Therefore, ﬁxing h t 0 := h 0 , the entire latent process can be computed by iteratively solving an ODE followed by applying a RNN.
GRU-ODE-Bayes.
The model architecture describing the latent variable in GRU-ODE-Bayes (Brouwer et al., 2019) is deﬁned as the ODE-RNN but with the special choice of a gated recurrent unit (GRU) for the RNN-cell and the neural network f θ also being derived from a GRU.
P ROPOSED METHOD – N EURAL J UMP ODE
Markovian paths.
Our assumptions on the stochastic process X imply that it is a Markov process.
In particular, the optimal prediction of a future state of X only depends on the current state rather than on the full history.
Hence, the previous values do not provide any additional useful information for the prediction.
JumpNN instead of RNN.
Using a neural ODE between two observation has the advantage that it allows to continuously model the hidden state between two observations.
But since the underlying process is Markov, there is no need to use a RNN-cell to model the updates of the hidden state at each new observation.
Instead, whenever a new observation is made, the new hidden state can solely be deﬁned from this observation.
Therefore, we replace the RNN-cell used in ODE-RNN by a standard neural network mapping the observation to the hidden state.
We call this the JumpNN which can be interpreted as an encoder map.
The advantage over the ODE-RNN is that this architecture is easier to train.
Last observation and time increment as additional inputs for the neural ODE.
The neural network f θ used in the neural ODE takes two arguments as inputs, the hidden state h t and the current time t .
However, our theoretical problem analysis suggests, that instead of t the last observation time t i − 1 and the time increment t − t i − 1 should be used.
Additionally the last observation x i − 1 should also be part of the input.
Therefore, we change the input to f θ accordingly.
Use of hyperbolic tangent.
The observations x i and the hidden states h t are a priori not bounded.
However, it is well known that neural networks work best, if their inputs are bounded by 1 .
Moreover, bounded inputs are essential in our theoretical analysis to be able to apply universal approximation theorems.
Therefore, we apply the tanh function to all possibly unbounded inputs of the neural networks before passing them on.
For better readability, we use the same notation for the inputs x i and h t to which tanh was applied.
NJ-ODE.
Combine the ODE-RNN model architecture with the previous considerations, we introduce the modiﬁed model architecture of our neural jump ODE (NJ-ODE) model, that is given in the
Algorithm 1.
A neural ODE f θ transforms the hidden state between observations, and the hidden state jumps according to JumpNN when a new observation is available.
The outputNN, a standard neural network, maps any hidden state h t to the output y t .
In the training process, the weights of all three neural networks, JumpNN, the neural ODE f θ and outputNN are optimized.
Algorithm 1 The NJ-ODE.
A small step size ∆ t is ﬁxed and we denote t n +1 :
= T .
Input: Data points with timestamps { ( x i , t i ) } i =0 ...
n , for i = 0 to n do h t
i = JumpNN ( x i ) (cid:46) Update hidden state given next observation x i
y t
i = outputNN ( h t i ) (cid:46) compute output s ← t i while s + ∆ t ≤ t
i +1 do h s +∆ t
=
ODESolve ( f θ , h s , x i , t i , s − t
i , ( s, s + ∆ t ))
(cid:46) get next hidden state y s +∆ t =
outputNN ( h s +∆ t ) (cid:46) compute output s ← s + ∆ t end while end for
Objective function.
As described before, our goal is to train the NJ-ODE model such that its output approximates the conditional expectation (2), which is the optimal prediction of the target process X with respect to the L 2 -norm.
Therefore, we deﬁne a new objective function, with which we can prove convergence.
Let y i − denote the output of the NJ-ODE at t i before the jump and y i the output at t i after the jump.
We remark that the outputs depend on parameters θ
and the previously observed x i which are inputs to the model.
Then the objective function is deﬁned as
[Equation]
N
[Equation]
[Table]
[Equation]
paths
[Equation]
[Table]
[Equation]
[Table]
(cid:125) dates (cid:0) | x ji
[Equation]
y ji
[Equation]
jump part at observations
[Equation]
part between two observations
[Equation]
We give an intuitive explanation for this deﬁnition.
The “jump part” of the loss function forces the
JumpNN to produce good updates based on new observations, while the other part forces the jump size to be small in (the empirical)
L 2 -norm.
Since the conditional expectation minimizes the jump size with respect to the L 2 -norm, this forces the neural ODE f θ to continuously transform the hidden state such that the output approximates the conditional expectation.
Moreover, both parts of the loss function force the outputNN to reasonably transform the hidden state h t to the output y t .
M AIN RESULT – THEORETICAL CONVERGENCE GUARANTEE
In the following we informally state our main result.
To formally state and prove the theorem, precise deﬁnitions of all ingredients are needed.
This analysis is provided in the appendix where the following results is stated in Theorem E.2 and Theorem E.12.
Theorem 5.1 (informal) .
We assume that for each N and for every size of the neural networks, their weights are chosen optimally, as to minimize ˆΦ N ( θ ) .
Then, if the number of paths and the size of the neural networks tend to inﬁnity, the output of our model converges in mean ( L 1 -convergence) to the conditional exception of the stochastic process X given the current information.
An intuitive explanation for this theorem was given with the deﬁnition of the objective function.
In this result, the focus lies on the convergence analysis under the assumption that optimal weights are found.
Below we explain, why this assumption is not too restrictive, giving a glance at the research ﬁeld of ﬁnding global optima of neural networks.
Global versus local optima.
The assumption that the optimal weights are found, is typical for a convergence analysis of a neural network based algorithm, since the objective function is highly complex and non-convex with respect to the weights.
In particular, it is well known that the standard choice of (stochastic) gradient descent optimization methods do in general only ﬁnd local and not global minima.
Since the difference between any local minimum and the global minimum can not generally be bounded, it is unrealistic to hope for a theoretical proof of convergence with respect to such optimisation schemes.
On the other hand, global optimization methods as for example simulated annealing provably convergence (in probability) to a global optimum (Locatelli, 2000;
Lecchini-Visintini et al., 2008).
Hence, combining those with our result, convergence in probability of our model output to the conditional expectation can be established, without the assumption that the optimal weights are found.
However, these global optimization schemes come at the cost of much slower training compared to (stochastic) gradient descent methods when applied in practice.
Moreover, several works have focused on showing that most local optima of neural networks are nearly global, see for example (Feizi et al., 2017) and the related work therein.
Hence, using (stochastic) gradient descent optimization methods likely yield nearly globally optimal weights much more efﬁciently.
In our case, this is also supported by our empirical convergence studies in Section 6.3.
E XPERIMENTS
For further details and results for all experiments see Appendix F.
6. T RAINING ON SYNTHETIC DATASETS
Evaluation metric.
For synthetic datasets where an analytic formula for the conditional expectation exists, we can evaluate the distance of the model output to the target process (2).
We use a sampling time grid with equidistant step size ∆ t :
= TK , K ∈ N , on [0 , T ] .
On this grid, we compare for path j at time t , the true conditional expectation ˆ x jt with the predicted conditional expectation (the model output) y jt .
For N 2 test samples, the evaluation metric is deﬁned as
[Equation] [Equation]
Black-Scholes, Ornstein-Uhlenbeck and Heston.
We test our algorithm on three scalar stochastic models, Black-Scholes, Ornstein-Uhlenbeck and Heston, with ﬁxed parameters.
For each model, we generate a dataset by sampling N = 20 (cid:48)
000 paths on the time interval [0 , 1] using the Euler-scheme with 100 time steps.
Independently for each path, on average 10% of the grid points are randomly chosen as observation times.
The NJ-ODE is trained on 80% of the data.
On the remaining 20% the model is tested, by comparing the loss function (31) computed with the NJ-ODE to the loss function computed with the true conditional expectation (Figure 2).
During training, the relative difference becomes very small, hence the true conditional expectation is nearly replicated.
[Figure] [Figure]
6. F URTHER SYNTHETIC DATASETS
Heston model without Feller condition.
We also train our model on a Heston dataset for which the
Feller condition is not satisﬁed.
As explained by Andersen (2007); Jean-François et al.
(2015) this is a more delicate situation.
We see that our model is very robust, since even in this delicate case our model learns to replicate the true conditional expectation process.
In the Heston model, the variance of the stochastic process X t is also modelled by a stochastic process v t .
Here, we train our model to predict both processes at the same time. The training on this 2 -dimensional dataset is successful as can be seen in Figure 3.
The minimal evaluation metric after 200 epochs is 0 . 0983 .
[Figure]
Dataset with changing regime.
In this experiment we test, how well our model can deal with stochastic processes, that undergo an (abrupt) change of regime at a certain point in time.
Many real world time series might exhibit such a change of regime.
Some examples are listed below.
• Longitudinal patient health recordings might experience changes depending on seasonal or longer-term inﬂuences, as for example due to the seasonal ﬂue or currently the Covidpandemic.
• In many regions climate data has strong seasonal dependencies, that can lead to relatively abrupt changes as for example when the weather changes from dry to rain season.
• A stock market that suddenly changes from a bullish to a bearish market, for example due to a macro-economic event.
An example for this would be the start of the Covid-19 crisis in the ﬁrst quarter of 2020.
We test a change of regime by combining two synthetic datasets.
On the ﬁrst half of the time interval [0 , 0 .
5] we use the Ornstein-Uhlenbeck and on the second half [0 . 5 , 1] the Black-Scholes model.
In
Figure 4 we see that our model correctly learns the change of regime.
The minimal evaluation metric after 200 epochs is 0 . 0463 .
[Figure]
Dataset with explicit time dependence.
Many real world datasets, have an explicit time dependence, i [Figure] e. the drift and diffusion of (9) explicitly depend on t .
Examples are all datasets that have a certain periodicity, as for example weather data (seasonal and daily periodicity), intraday periodicity of stock prices (Andersen & Bollerslev, 1997) or prices of certain seasonal goods.
We incorporate an explicit time dependence into the Black-Scholes dataset, by replacing the drift constant µ with the time dependent constant α 2 (sin( βt ) + 1) , for α, β > 0 .
In Figure 5 we see that the model learns to adapt to the time-dependent coefﬁcients.
The minimal evaluation metric after 100 epochs is 0 .
( β = 2 π ) and 0 .
02805 ( β = 4 π ) respectively.
[Figure]
6. E MPIRICAL CONVERGENCE STUDY
We conﬁrm the theoretical results of Theorem 5.1 by an empirical convergence study for growing numbers of training samples N and network size M , where the performance is measured by the evaluation metric (8).
For each combination of the number of training samples N 1 and the neural network size M , the NJ-ODE is trained 5 times on the Black-Scholes, Ornstein-Uhlenbeck and
Heston datasets.
The means of the evaluation metric over the 5 trials with their standard deviations are shown in Figure 6 for Black-Scholes.
Already 200 training samples lead to a small evaluation metric if the network size is big enough, suggesting that the Monte Carlo approximation of the loss function is good, already with a few samples.
6. C OMPARISON TO GRU-ODE-B AYES ON SYNTHETIC DATA
On the Black-Scholes, Ornstein-Uhlenbeck and Heston datasets, we compare our model to GRUODE-Bayes (Brouwer et al., 2019), which is, to the best of our knowledge, the neural network based method with the most similar task to ours.
Results of our comparison are shown in Table 6.4.
On the
Black-Scholes and Ornstein-Uhlenbeck dataset, our model performs similarly to GRU-ODE-Bayes.
However, on the more complicated Heston dataset, the training of GRU-ODE-Bayes is unstable and does not converge.
On the other hand, our model converges during training.
Although the value of the evaluation metric is much higher than for Black-Scholes and Ornstein-Uhlebeck, the resulting model output is still meaningful, which is not the case for GRU-ODE-Bayes.
Hence we conclude, that GRU-ODE-Bayes cannot be applied reasonably for the Heston dataset, while our method works as the theoretical results suggest. [Table]
[Table]
R EAL WORLD DATASETS WITH INCOMPLETE OBSERVATIONS
Self-imputation for incomplete observations.
Until now we assumed that at each observation time all coordinates of the stochastic process X are observed.
However, in many real world applications, the observations are incomplete,
i
[Table]
e. only for some of the coordinates an observation is available.
To deal with such incomplete observations, we propose the following self-imputation method.
Whenever an observation x i ∈ R
d
X is made the mask m i ∈ { 0 , 1 } d X tells whether the j -th coordinate of x i is observed ( m ji = 1 ) or not ( m ji = 0 ).
Then we use the prediction y
i − of our model to generate the imputed observation vector
[Equation]
where (cid:12) is the element-wise multiplication (Hadamar product) and 1 d X ∈ R d X is the one-vector.
Instead of x i we use (˜ x i , m i ) as an input for the jump part jumpNN.
The intuition behind this deﬁnition is the following.
In the one dimensional case, if we do not make an observation, but input y t − as if it was an observation, we expect that this does not change the output y s for s ≥ t .
From this point of view, y t − does not provide any additional information for the model.
Similarly, we expect that imputing y t − for unobserved coordinates does not provide any information about this coordinate to the model.
However, since the model might learn to transfer the information about an observed coordinate to an unobserved one, we extend the input to also include the information which coordinates were observed.
For the ODE part, we use y i , which is the prediction after processing the input ˜ x
i , as input instead of x i .
Here, the intuition is that if the model learns how to best use the incomplete observation, i.e. if the jump is good, then this is the best approximation for x i .
Our objective function is adjusted by multiply each term in the sum with the mask.
Climate forecast.
We compare our model to GRU-ODE-Bayes on the USHCN daily dataset (Menne et al., 2016), using the same experimental setting as was used by Brouwer et al.
We train a small (S) and a large (L) version of NJ-ODE with different total number of parameters.
The validation set was used for early stopping after the ﬁrst 100 epochs, where we trained for a total of 200 epochs.
We see in Table 6.5 that our small version performs slightly worse than GRU-ODE-Bayes while our large version slightly outperforms it.
Physionet.
We compare our model to the latent ODE on their extrapolation task on the PhysioNet
Challenge 2012 dataset (Goldberger et al., 2000), using the same experimental setting as was used by Rubanova et al.
We see in Table 6.5 that in the ﬁrst (still ongoing) experiments our model slightly outperforms the latent ODE.
R ELATED WORK
Stochastic ﬁltering theory.
Our main theorem is similar to the theoretical results of neural ﬁltering (Lo, 2009), which is a neural network approach to stochastic ﬁltering theory in discrete time.
In stochastic ﬁltering, potentially incomplete and noisy observations of X are available continuously [Table] [Table] in time.
This observation process Y is usually described by the dynamics dY t
= h ( X t ) dt + dB t , where h is a measurable function and B is a Brownian motion.
Stochastic ﬁltering then estimates the conditional law of X t given the noisy observations ( Y s ) 0 ≤ s ≤ t (Bain & Crisan, 2008, Def. 3.2) and therefore can provide conditional expectations.
Comparably [Table] we directly compute the conditional expectation given the last observation.
Similar to our assumptions, in the neural ﬁltering approach of
GRU-ODE-Bayes (Brouwer et al., 2019) and (Ryder et al., 2018)
observations are only available at irregular discrete time points.
They approximate the conditional law of X given the last observation by a Gaussian distribution and learn its mean and variance parameters.
In particular, the conditional expectation is then given by this mean parameter.
In contrast, we do not make normality assumptions about the conditional distribution and we theoretically prove convergence to the true conditional expectation
[Table] with jumps.
Except for GRU-ODE-Bayes (Brouwer et al., 2019) and ODE-RNN (Rubanova et al., 2019) another work studying a neural ODE with jumps is neural jump SDE (NJSDE)
(Jia & Benson, 2019).
Similar to the NJ-ODE framework (29), the latent process of NJSDE is described by a neural ODE with jumps at random times.
This model is used to describe hybrid systems which evolve continuously in time but may also be interrupted by stochastic events.
In contrast to that, we model the conditional expectation of a continuous stochastic process.
C ONCLUSION
We presented the neural jump ODE, a data-driven framework for modelling the conditional expectation of a stochastic process given the previous observations.
We introduced a rigorous mathematical description of our model and more generally for the class of neural ODE based models. Moreover, this is the ﬁrst time that for a model falling in this category theoretical guarantees are given.
We evaluated our model empirically on six synthetic and two real world datasets.
In comparison to the baselines GRU-ODE-Bayes and latent ODE, we achieved better results especially on complex datasets.
A PPENDIX A S ETUP A. S TOCHASTIC PROCESS X
In this section in rigorously describe the process X and give the assumptions which are needed in order to derive the convergence results.
Let d X , d W ∈ N and T > 0 be the ﬁxed time horizon.
Consider a ﬁltered probability space (Ω , F , F :=
{F t } 0 ≤ t ≤ T , P ) , on which an adapted d W -dimensional
Brownian motion { W t } t ∈ [0 ,T ] is deﬁned.
We deﬁne the stochastic process 1 X :=
( X t ) t ∈ [0 ,T ] as the solution of the stochastic differential equation (SDE)
[Equation]
) for all 0 ≤ t ≤ T , where X 0 = x ∈ R
d
X is the starting point and the measurable functions µ : [0 , T ] × R d X → R d X and σ : [0 , T ] × R d X → R d X × d W are the drift and the diffusion respectively.
We impose the following assumptions:
• X is continuous and square integrable ,
i
e. for P -a.e.
ω ∈ Ω the map t (cid:55)→ X t ( ω ) is continuous and E [ X 2 t ] < ∞ for every t ∈ [0 , T ] .
• µ and σ are both globally Lipschitz continuous in their second component ,
i
e. for ϕ ∈ { µ, σ } there exists a constant M > 0 such that for all t ∈ [0 , T ] | ϕ
( t, x ) − ϕ
[Equation]
t, y )
|
[Equation]
x − y
[Equation]
and | ϕ ( t, x ) | 2 ≤ (1 + | x | 2 ) M.
[Equation] [Equation]
In particular, their growth is at most linear in the second component.
• µ is bounded and continuous in its ﬁrst component ( t ) uniformly in its second component ( x ) , i.e. for every t ∈ [0 , T ] and ε > 0 there exists a δ > 0 such that for all s ∈ [0 , T ] with | t − s | <
δ and all x ∈ R d
X we have | µ ( t, x ) − µ ( s, x ) | < (cid:15) .
• σ is càdlàg (right-continuous with existing left-limit) in the ﬁrst component and Lintegrable with respect to W , σ ∈ L 2 ( W ) , i.e
[Equation]
where |·| F denotes the Frobenius matrix norm.
This is in particular implied if σ is bounded.
A. R ANDOM OBSERVATION DATES
In this section we describe how we model the observation dates.
We want to treat irregularly observed time-series, i.e. without a ﬁxed number of observations and not necessarily equally distributed.
Therefore, we suppose that we have a possibly random number of n observations and that those observations are made at random times t 1 , . . . , t n .
For simplicity we make the assumption that the observation dates are independent of the stochastic process itself.
In future work, this will be generalized to observation dates correlated with X , that could be modelled by a point process on the same probability space.
In contrast, here we hard-code the independence assumption by considering a second probability space (˜Ω , ˜ F , ˜ P ) , on which the observation times of the stochastic process X are deﬁned.
More precisely, we assume that:
• n : ˜Ω → N ≥ 0 is a random variable with E ˜ P [ n ] < ∞ , the number of observations.
• t i : ˜Ω → [0 , T ] for 0 ≤ i ≤ n are sorted 2 random variables, the observation times.
We denote the joint pushforward measure of n and { t i } 0 ≤ i ≤ n as ˜ P t :=
( n, t 0 , . .
. , t n )#˜ P .
The random variable n can but bounded max
[Equation]
k ∈ N | ˜ P
[Equation]
n ≥ k ) > 0 (cid:111) to be the maximal value of n , which otherwise is inﬁnity.
We use the notation B ([0 , T ]) for the Borel σ -algebra of the set [0 , T ] .
Then we deﬁne for each 1 ≤ k < K +
[Equation]
which is a probability measure on the time interval as shown in the lemma below.
Here, ( t k − ) means the left-point of t k , for example, if t 0 = 0 , t 1 = 1 we have t 0 ∈ [0 , 1) , t 1 / ∈ [0 , 1) but t 0 − / ∈
[0 , 1) , t 1 − ∈ [0 , 1) .
Lemma A.1.
For 1 ≤ k < K + 1 the map λ k deﬁnes a probability measure.
Proof.
First we see that λ k
([0 , T ]) = 1 since t k maps to [0 , T ] .
Furthermore, for any disjoint sets
B i ∈ B ([0 , T ]) , i ∈ N , we have { n ≥ k, t k − ∈ ˙
∪ i ≥ 1 B i } = ˙ ∪ i ≥ 1 { n ≥ k, t k − ∈ B i } and these sets are ˜ F -measurable, since they are deﬁned through pre-images of random variables.
Therefore, the additivity of ˜ P implies that λ k ( ˙
∪
i ≥ 1 B i ) =
(cid:80) i ≥ 1 λ k
( B i ) .
Moreover, we deﬁne τ as the time of the last observation before a certain time t ,
[Equation]
→ [0 , T ] ,
[Equation]
≤ t } .
Example A.2.
We give two examples how the random observation dates could be deﬁned.
Let ( N t ) t ∈ [0 ,T ] be a homogeneous Poisson point process with rate r > 0 .
Hence, N (0) = 0 ,
t (cid:55)→ N ( t ) is constant except for jumps of size 1 at discrete random times and for any ﬁxed t ∈ [0 , T ] , N ( t ) is
Poisson distributed
, i
e.
[Equation]
Then the number of observations is deﬁned as n := N ( T ) and satisﬁes E ˜ P [ n ] = rT .
The observation dates t 1 , . .
. , t n are deﬁned as the discontinuity times of N , i.e. the times where
N increases by 1 .
In particular, for 0 ≤ k ≤ n , t k := inf { t | N ( t ) =
k } ≤ T .
Therefore, for 0 ≤ a ≤ b ≤ T we can rewrite
[Equation]
• We deﬁne n by one of the following options: – as a constant in N > 0 .
– as a Binomial random variable, n ∼ Binom( p, n max ) , for p ∈ (0 , 1) , n m ax ∈ N > 0 .
– as a Geometric random variable, n ∼ Geom( p ) , for p ∈ (0 , 1) .
– as a Poisson random variable, n ∼ Poi( r ) , for r > 0 .
and t 1 , . . . , t n are deﬁned by choosing n uniform random variables on [0 , T ] and sorting them.
A. I NFORMATION σ - ALGEBRA
In this section, we deﬁne a mathematical tool, the σ -algebra, that is essential to the description of the conditional expectation.
This object describes which information is available at a certain time t .
In the following, we leave away ˜ ω ∈ ˜Ω whenever the meaning is clear.
We deﬁne the ﬁltration of the currently available information A :
=
( A t ) t ∈ [0 ,T ] by
[Equation]
where t i are the observation times and σ ( · ) denotes the generated σ -algebra.
By the deﬁnition of τ we have A t =
A τ ( t ) for all t ∈ [0 , T ] .
(Ω × ˜Ω , F ⊗ ˜ F , F ⊗ ˜ F , P × ˜ P ) is the ﬁltered product probability space which, intuitively speaking, combines the randomness of the stochastic process with the randomness of the observations.
Here, F ⊗ ˜ F consists of the tensor-product σ -algebras ( F ⊗ ˜ F ) t :=
F t ⊗ ˜ F for t ∈ [0 , T ] .
As explained in the remark below, A t can be identiﬁed with a sub- σ -algebra of ( F ⊗ ˜ F ) t .
Remark A.3.
The σ -ﬁelds
A t depend on ˜ ω ∈ ˜Ω as well.
If we look at the product probability space (Ω × ˜Ω , F ⊗ ˜ F , F ⊗ ˜ F , P × ˜ P ) , where F ⊗ ˜ F consists of the tensor-product σ -algebras ( F ⊗ ˜ F ) t :=
F t ⊗ ˜ F for t ∈ [0 , T ] , then σ ( X t i | t i ≤ t )
[Equation]
is a well deﬁned sub- σ -algebra of ( F ⊗ ˜ F ) t .
Furthermore, we can recover the ˜ ω -wise deﬁned version of A t by intersecting each set in it with Ω × { ˜ ω } and subsequently projecting the intersection on its ﬁrst component.
We use the notation ˜
A t :=
˜
A t (˜ ω ) =
σ ( X t i (˜ ω )
| t
i (˜ ω ) ≤ t ) to distinguish this ˜ ω -wise deﬁnition from the deﬁnition as sub- σ -algebra of the product space given above.
However, Lemma
B.3 implies that for our considerations, both versions of this σ -algebra have the same effect, therefore we will simply write A t for both versions, by abuse of notation.
B O PTIMAL APPROXIMATION ˆ X OF THE STOCHASTIC PROCESS X
We are interested in the “best” approximation (or prediction)
ˆ X t of the process X that one can make at any time t ∈ [0 , T ] , given the currently available information A t .
For us “best” refers to the L 2 (Ω × ˜Ω , F ⊗ ˜ F , P × ˜ P ) -minimizer, therefore, this approximation is given by the conditional expectation.
Indeed, if we deﬁne ∆ := {
[Equation]
t
[Equation]
r ) ∈
[Equation]
T ] 2 | t + r ≤ T } , and the function
[Equation]
this is proven in the following proposition.
Proposition B.1.
The optimal (i.e. L 2 -norm minimizing)
A -adapted process in L 2 (Ω × ˜Ω , F ⊗ ˜ F , P × ˜ P ) approximating ( X t ) t ∈ [0 ,T ] is given by ˆ X :=
( ˆ X t ) t ∈ [0 ,T ] with ˆ X t :=
E P × ˜ P
[ X t |A t ] .
Moreover, this process is unique up to ( P × ˜ P ) -null-sets.
In addition we have ( ˜ ω -wise for ˜ ω ∈ ˜Ω ) that
[Equation]
The ﬁrst part of the result follows from the elementary Proposition below (which is proven for example in (Durrett, 2010, Thm. 5.1.8) for R -valued random variables and can easily be extended to R d -valued random variables when using the 2 -norm).
Proposition B.2.
Given a probability space (Ω , F , P ) and a sub- σ -algebra A ⊂ F , the orthogonal projection of X ∈ L 2 (Ω , F , P ) on L 2 (Ω , A , P ) is given by ˆ X := E [ X |A ] .
In particular, for every
Z ∈ L 2 (Ω , A , P ) with P ( Z (cid:54) =
ˆ X ) >
we have
[Equation]
Proof Proposition B.1.
In our case this means, that the optimal A -adapted process in L 2 (Ω × ˜ Ω , F ⊗ ˜ F , P × ˜ P ) approximating X is given by ( ˆ X t ) t ∈ [0 ,T ] with ˆ X t := E P × ˜ P
[ X t |A t ] .
Here, A t is meant as a sub- σ -algebra of F t × ˜ F .
This process is unique up to ( P × ˜ P ) -null-sets.
Moreover, the following lemma shows that it coincides ˜ ω -wise with E P
[ X t | ˜
A t ](˜ ω ) , where ˜
A t = ˜
A t (˜ ω ) is deﬁned in Remark
A.3.
Lemma B.3.
For ˜ P -almost-every ˜ ω ∈ ˜Ω we have E P × ˜ P
[ X t |A t ](˜ ω ) =
E P
[ X t | ˜
A t ](˜ ω ) P -almostsurely.
Proof.
Otherwise we have by Fubini’s theorem, Proposition B.2 and an argument similar to the one in Lemma E.
[Equation]
which is a contradiction to Proposition B.2.
This proves the ﬁrst part of Proposition B.1.
The second part of this Proposition, i.e. (12), should be understood ˜
ω -wise, for ˜ ω ∈ ˜Ω .
This is justiﬁed by Lemma B.3 and derived below.
In particular, for the remainder of this section, all statements are meant ˜ ω -wise.
With the assumption that µ and σ are Lipschitz, (Protter, 2005, Thm. 7, Chap.
V) implies that a unique solution of (9) exists.
Furthermore, this solution is a Markov process as soon as the starting point x is ﬁxed (Protter, 2005, Thm.
32, Chap.
V).
Hence, one can deﬁne a transition function P (compare (Protter, 2005, Chap. V.6)) such that for all s < t and φ bounded and measurable,
[Equation]
We have that X τ ( s ) is A τ ( s ) -measurable
and therefore, since A s =
A τ ( s ) ⊂
F τ ( s ) ,
[Equation]
By our additional assumption on σ it follows from (Protter, 2005, Lem.
before Thm.
28, Chap.
IV)
that
[Equation]
is a square integrable martingale, since the Browian motion W is square integrable.
In particular, for 0 ≤ s ≤ t ≤ T we have E [ (cid:82) t s σ ( r, X r ) dW r |F s ] =
E [ M t − M s |F s ] = 0 .
Moreover, the same is true when conditioning on A s 3 .
Using the martingale property of M , we have ( ˜ ω -wise) for every t ∈ [0 , T ]
[Equation]
where we used Fubini’s Theorem (for conditional expectations) in the last step.
This is justiﬁed because E P [ (cid:82) T 0 | µ ( r, X r ) | 2 dr ] < ∞ follows from µ being bounded.
Let us deﬁne ∆ := { ( t, r ) ∈ [0 , T ] 2 | t + r ≤ T } and the function ˜ µ
[Equation]
∆ × R d X → R d X ,
[Equation]
then we can use (13) to rewrite (14) as
[Equation]
This proves the second part of Proposition B.1.
Proposition B.4.
The function ˜ µ is (jointly) continuous.
Together with Proposition B.1 this implies that ˆ X t is càdlàg.
Proof of Proposition B.4.
For any ﬁxed s ∈ [0 , T ] , x ∈ R d X , we deﬁne
[Equation]
to
[Equation]
This solution exists and is unique by (Protter, 2005, Chap.
V, Thm. 7), therefore we have that
X t = ζ 0 ,t ( x ) P -almost surely.
Furthermore, (Gubinelli, 2016, Thm. 4) implies that for s ≤ t we have X t = ζ s,t ( ζ 0 ,s ( x )) .
Hence, for t = s + r , we have the identity ˜ µ ( s, r, ξ ) =
E [ µ ( t, ζ s,t ( ξ ))] .
Furthermore, by (Gubinelli, 2016, Thm.
8) we have for any ξ, ξ (cid:
) ∈ R d X and ( s, r ) , ( s (cid:48) , r (cid:48) ) ∈ ∆ with t := s + r, t (cid:48) :
= s (cid:48)
+ r (cid:48) ∈
[0 , T ] that there exists some constant C such that
[Equation]
Therefore, we have that
[Equation]
where we used Jensen’s inequality in the second last and (10) in the last step.
Hence, for ( s (cid:48) , r (cid:48) , ξ (cid:48) )
→ ( s, r, ξ ) we have that the ﬁrst term of (17) goes to zero due to continuity of µ in its ﬁrst component uniformly in the second component.
Moreover, the second term of (17) converges to zero by (16).
Together, this proves continuity of ˜ µ .
C R ECALL : RNN, NEURAL ODE AND ODE-RNN - EQUIVALENT WAYS OF WRITING
We describe our model as the solution of the SDE (29), which is a compact way of writing.
In the following, we ﬁrst shortly recall recurrent neural networks (RNN) and the neural ODE and then recall the ODE-RNN.
Furthermore, we give a step-by-step explanation, how the way ODE-RNN was formalized, can equivalently be written in terms of an SDE similar to (29).
Finally we give the alternative way of writing our model and a short comparison to ODE-RNN.
Recurrent Neural Network.
The input to a RNN is a discrete time series of observations { x t 1 , · · · , x t n } .
At each time t i +1 , the latent variable h is updated using the previous latent variable h t i and the input x t i +1 as
[Equation]
) where RNNCell is a neural network.
Neural Ordinary Differential Equation.
Neural ODEs (Chen et al., 2018) are a family of continuous-time models deﬁning a latent variable h t :
= h ( t ) to be the solution to an ODE initial-value problem (IVP):
[Equation]
where f ( · , · , θ )
=
f θ is a neural network with weights θ .
Therefore, the latent variables can be updated continuously by solving this ODE (19).
We can emphasize the dependence of h t on a numerical ODE solver by rewriting (19) as
[Equation]
ODE-RNN.
ODE-RNN (Rubanova et al., 2019) is a mixture of a RNN and a neural ODE.
In contrast to a standard RNN, we are not only interested in an output at the observation times t i , but also in between those times.
In particular, we want to have an output stream that is generated continuously in time.
This is achieved by using a neural ODE to model the latent dynamics between two observation times, i.e. for t
i < t < t i +1 the latent variable is deﬁned as in (19) and (20), with h t 0 and t 0 replaced by h t i and t i .
At the next observation time t i +1 , the latent variable is then updated by a RNN.
Rubanova et al.
(2019) write this as
[Equation]
Therefore, ﬁxing h t 0 := h 0 , the entire latent process can be computed by iteratively solving an ODE followed by applying a RNN.
GRU-ODE-Bayes.
The model architecture describing the latent variable in GRU-ODE-Bayes (Brouwer et al., 2019) is deﬁned as the ODE-RNN but with the special choice of a gated recurrent unit (GRU) for the RNN-cell and the neural network f θ also being derived from a GRU.
ODE-RNN as càdlàg process.
Thinking about the process h :
= ( h t ) t ≥ t 0 deﬁned in (21) as a (stochastic) process in time, it is deﬁned to evolve continuously for t i ≤ t <
t
i +1 according to the
ODE dynamics f θ and jumps at time t
i +1 according to the RNN cell.
In particular, it is deﬁned to be a càdlàg 4 process, for which h t i +1 − is the standard notation for the left limit, i.e. the last point before the jump at time t
i +1 .
According to this notation we have h t
i
[Equation]
1 − = h
[Equation]
t i +1 , hence, we can rewrite (21) as
[Equation]
C. R ESIDUAL ODE-RNN AS A SPECIAL CASE OF CONTROLLED ODE Residual
RNN.
We replace the standard RNN cell by a residual RNN cell (rRNN), as it was described e.g. in Yue et al.
In particular, instead of applying the RNN cell such that h t
i = RNNCell
( h t i − , x i ) we use a residual RNN cell to have h t i =
h t i − + rRNNCell( h t i − , x i ) .
The residual RNN is as expressive as the standard RNN and was empirically shown to perform very similarly or even better than the standard framework (Yue et al., 2018).
This way, the residual RNN cell models exactly the jump of the latent variable (i.e. the differences) that occurs at the time t
i + when taking into account the next observation x t
i +1 .
Therefore, we can rewrite the ODE-RNN (22) as
[Equation]
Controlled Ordinary Differential Equation.
We brieﬂy recall the deﬁnition of controlled ODEs as it was given in (Herrera et al., 2020, Section 4.1) and used in (Cuchiero et al., 2019;
Herrera et al., 2020) to describe neural networks.
We ﬁx (cid:96), d ∈ N and deﬁne for 1 ≤ i ≤ d the vector ﬁelds
[Equation]
which are càglàd 5 in t and Lipschitz continuous in x .
Furthermore, we deﬁne the scalar càdlàg control functions
[Equation]
which have ﬁnite variation and satisfy u i (0)
= 0 .
Then we deﬁne the process Z :=
( Z t ) t ≥ 0 as the solution of the controlled
[Equation]
where z ∈ R (cid:96) is some starting point.
(24) is written in Itô’s differential notation for (stochastic) integrals.
The solution of (24) exists and is unique under much more general assumptions than we made here (Protter, 2005, Chap.
V, Thm. 7).
Special Case: ODE-RNN.
To write the ODE-RNN as controlled ODE, we set d = 2 and deﬁne
V θ 1 :=
f θ , V 2 :=
rRNNCell and u 1 ( t ) :
= t .
As pointed out by Herrera et al.
(2020), one can take the u i to be semimartingales instead of deterministic functions.
In line with this, we deﬁne u 2 :=
u as the pure jump stochastic control process
[Equation]
We note that u is an adapted process starting at 0 with ﬁnite variation on the product probability space (Ω × ˜Ω , F ⊗ ˜ F , F ⊗ ˜ F , P × ˜ P ) , since the variation of u up to time T is n and E P × ˜ P [ n ] < ∞ .
The following result shows that the residual ODE-RNN can compactly be described as a controlled ODE.
Proposition C.1.
Using the vector ﬁelds and controls deﬁned above, the latent variable process h
= ( h t ) t ≥ t 0 of the residual ODE-RNN can equivalent be written as the solution of the controlled
ODE
[Equation]
Proof of Proposition C.1.
By (Protter, 2005, Chap.
II, Thm. 17), the stochastic integral is indistinguishable from the path-wise Lebesgue-Stieltjes integral if the integrator is of ﬁnite variation.
Hence, we can assume that some ˜ ω ∈ ˜Ω is ﬁxed and that the following expressions are evaluated at this ˜ ω whenever applicable.
First note, that u is constant except at the t i where it increases by 1 (cf.
Figure 7).
In particular, the Lebesgue-Stieltjes integral of some càdlàg function g with respect to u is a sum, i.e. (cid:82) t 0 g ( s − ) du s =
(cid:80) t i ≤ t g ( t i − )
∆ u t i , where ∆ u t i =
u t i − u t
i − = 1 .
Therefore, integrating (26) from t 0 to t we get
[Equation]
In particular, since the ﬁrst integral is continuous in t , we have for every 1 ≤ k ≤ n h t k
[Equation]
27) and for t k < t < t k +
[Equation]
Together, (27) and (28) prove that (26) is equivalent to (23).
We also emphasize that x t is used as input only for t
= t i , 1 ≤ i ≤ n .
D N EURAL J UMP ODE
We propose a model framework whose architecture can be interpreted as a simpliﬁcation of the ODERNN (Rubanova et al., 2019) and the GRU-ODE-Bayes (Brouwer et al., 2019) model architecture.
D. T HE MODEL FRAMEWORK : NEURAL ODE BETWEEN JUMPS
We deﬁne X ⊂ R d X and H ⊂ R d H to be the observation and latent space for d X , d H ∈ N .
Moreover, we deﬁne the following feed-forward neural networks with sigmoid activation functions:
• f θ 1 : [ − 1 , 1] d H × [ − 1 , 1] d X × [0 , T ] × [0 , T ] → R d H modelling the ODE dynamics, • ρ θ 2 : [ − 1 , 1] d X → R d H modelling the jumps when new observations are made, and • g θ 3 : R d H → R d Y the readout map, mapping into the target space Y ⊂
R d Y for d Y ∈ N .
The trainable parameters of the neural networks are θ :=
( θ 1 , θ 2 , θ
3 ) ∈ Θ := ∪ M ≥ 1 Θ M .
Here, for every M ∈ N , Θ M is deﬁned to be the set of all parameters such that f θ 1 and ρ θ 2 have hidden dimension M .
We deﬁne the pure jump stochastic process (cf.
Appendix C)
[Equation]
Then the neural jump ODE (NJ-ODE) is deﬁned by the latent process H :
= ( H t ) t ∈ [0 ,T ] and the output process Y :=
( Y t ) t ∈ [0 ,T ] deﬁned as solutions of the SDE system (cf.
Appendix C)
[Equation]
We apply the hyperbolic tangent function to each component of every input variable except for the time variables, before passing them to the network.
This ensures that every input to the neural networks is bounded in the compact sets where f θ 1 and ρ θ 2 are deﬁned.
Compactness of the domain is an assumption needed to apply the universal approximation theorem by Hornik et al.
Note that, only the values of X at the times t i , 0 ≤ i ≤ n are used as inputs.
We will use the notation
H θt ( X ) and Y θt ( X ) to emphasize the dependence of the latent process H and the output process Y on the model parameters θ and the input X .
Existence and uniqueness.
We note that u is an adapted process starting at 0 with ﬁnite variation on the product probability space (Ω × ˜Ω , F ⊗ ˜ F , F ⊗ ˜ F , P × ˜ P ) , since the total variation of u up to time T is n and E P × ˜ P [ n ] < ∞ .
Hence, since f θ 1 and ρ θ 2 are Lipschitz continuous (as composition of Lipschitz continuous functions) a unique càdlàg solution H θ exists, once an initial value is ﬁxed (Protter, 2005, Thm. 7, Chap.
V).
Moreover, the resulting process ( Y θt ) t ∈ [0 ,T ] is also càdlàg and A -adapted.
Equivalent way of writing NJ-ODE.
By applying the steps that were explained in Section C backwards to (29) (but without introducing the residual version of the RNN), it is easy to see that our model can equivalently be written (similar to (22)) as
[Equation]
In particular, the main difference to (22) is that we use a modiﬁed input for f θ 1 and that the neural network performing the jumps does not take h t i +1 − as an input, i.e. it is not a RNN.
D. O BJECTIVE FUNCTION
We introduce a loss function, such that the output Y θ of the NJ-ODE can be trained to approximate ˆ
X , i.e. to model the best on-line prediction of the stochastic process X .
We deﬁne the theoretical loss function and its Monte Carlo and Ergodic approximation.
Objective function.
Let us deﬁne D to be the set of all R d X -valued A -adapted processes on the probability space (Ω × ˜Ω , F ⊗ ˜ F , F ⊗ ˜ F , P × ˜ P ) .
Then we deﬁne our objective functions
[Equation]
where Φ will be our (theoretical) loss function.
Remark that from the deﬁnition of Y θ , it directly follows that it is an element of D , hence Φ is well-deﬁned.
Monte Carlo approximation of the objective function.
Let us assume, that we observe N ∈ N independent realisations of the path X at times (˜ t j 1 , · · · , ˜ t jn ( j ) ) , 1 ≤ j ≤ N , which are themselves independent realisations of the random vector ( n, t 1 , · · · , t n ) .
In particular, let us assume that X j ∼
X and ( n j , t j 1 , · · · , t jn j ) ∼ P t are i.i.d.
random processes (respectively variables) for 1 ≤ j ≤ N and that our training data is one realisation of them.
Then
[Equation]
converges ( P × ˜ P ) -a.s.
to Φ( θ ) as N → ∞ , by the law of large numbers (cf.
Theorem E.12).
Therefore, (33) is a good approximation of Φ (32) that can be used in practice.
However, our theoretical result makes use of the limiting loss function Φ , i.e. we assume to be in the case of having inﬁnitely many independent training samples.
Ergodic approximation of the objective function.
If we only observe one realization of the path
X at times (˜ t 1 , · · · , ˜ t N ) , we can still approximate the objective function by assuming that µ and σ are time-independent and that the stochastic process X is ergodic in the following sense.
We ﬁx n = 1 and assume that the time increments ∆˜ t j :=
˜ t j − ˜ t j − 1 are i.i.d.
realizations of the probability distribution λ 1 .
Furthermore, we consider each observation X ˜ t
j as one sample with initial condition
X ˜ t j − 1 for which Y θ,j is the realization of Y θ .
Then we approximate the objective function by
[Equation]
which is assumed to converge by the ergodicity assumption for N → ∞ to
[Equation]
Instead of setting the random variable n = 1 , one could similarly ﬁx the time horizon T , and take for each sample all subsequent observations that lie in the time interval [ t start , t start + T ] , where t start is the date of the ﬁrst observation of this sample.
The next sample would then start with the ﬁrst observation after t start + T .
E T HEORETICAL CONVERGENCE RESULTS
Our main results show, that the output of NJ-ODE converges to the conditional expectation when the size of the neural networks and the number of samples go to inﬁnity.
For each network size (and number of samples) we assume to have the weights minimizing the (Monte Carlo approximation of the) loss function.
In particular, we do not consider the problem of ﬁnding those optimal weights, and therefore also do not analyse backpropagation through our model.
In a similar setting, backpropagation was studied by Jia & Benson (2019).
For completeness we recall the deﬁnition of L p -convergence.
Deﬁnition E.1.
Let 1 ≤ p < ∞ .
Let (Ω , F , P ) be a probability space.
Then a sequence of random variables ( X n ) n ∈ N converges to a random variable X in L p (Ω , F , P ) (or simply L p ), if
[Equation]
We use the notation
X n
[Equation]
p
[Equation]
X.
L 1 -convergence is also denoted convergence in mean .
E. C ONVERGENCE WITH RESPECT TO THEORETICAL OBJECTIVE FUNCTION
Theorem E.2.
Let θ min M ∈ Θ min M := argmin θ ∈ Θ M { Φ( θ ) } for every M ∈ N .
Then, for M → ∞ , the value of the loss function Φ (32) converges to the minimal value of Ψ (31) which is uniquely achieved by ˆ X , i.e.
[Equation]
Furthermore, for every 1 ≤ k ≤ K we have that Y θ min M converges to ˆ X as random variable in
L 1 (Ω × [0 , T ] , P × λ k ) .
In particular, the limit process Y :=
lim M →∞ Y θ min M equals ˆ X ( P × λ k ) - almost surely as a random variable on Ω × [0 , T ] .
The idea of the proof is to split up the target jump process into its continuous-in-time parts and into the jumps.
Both parts are continuous functions of their inputs and can therefore be approximated by neural networks (Hornik et al., 1989).
More precisely, by Proposition B.1 and B.4 the continuous-intime part can be written as an integral over time of a function which is jointly continuous in its inputs.
This jointly continuous function is approximated by a neural network, which is itself integrated over time – that is the neural ODE part of the NJ-ODE.
The remainder of the proof shows L 1 -convergence of the output of NJ-ODE with optimal weights (with respect to the loss function) to the conditional expectation process.
For completeness we restate the straight forward generalization of the universal approximation result (Hornik et al., 1989, Thm. 2.4) to multidimensional output.
Theorem E.3 (Hornik) .
Let r, d ∈ N and σ be a sigmoid function.
Let NN σr,d be the set of all 1-hidden layer neural networks mapping R r to R d .
Then for every compact subset K ⊂ R r , every (cid:15) > 0 and every f ∈ C ( R r , R d ) there exists a neural network g ∈ NN σr,d such that sup x ∈ K | f
( x ) − g ( x ) | 2 < (cid:15) .
The following Lemmas are used in the Proof of Theorem E.2.
Lemma E.4.
Let 1 ≤ k ≤ K and let Z ∈ D be a process such that ( P × λ k )
[ ˆ X (cid:54) =
Z ] > 0 .
Then there exists an ε > 0 such that ˜ B := { t ∈ [0 , T ] | E P [ | X t − Z t − | 22 ] ≥ ε + E P [ | X t − ˆ X t − | 22 ] } satisﬁes λ k ( ˜ B ) > 0 .
Proof.
First remark that since X is continuous, we have X t =
X t − .
Let us deﬁne C :
= { ( ω, t ) ∈
Ω × [0 , T ] | ˆ X t − ( ω ) (cid:54) =
Z t − ( ω ) } and for each t ∈ [0 , T ] let C t :=
{ ω ∈ Ω | ( ω, t ) ∈ C } .
Then we have for B := { t ∈ [0 , T ] | P ( C t ) > 0 }
that λ k
( B ) > 0 , since otherwise by Fubini’s theorem
[Equation]
which is a contradiction.
Now Proposition B.2 yields that for each t ∈ B there exists some ε t > 0 such that E P [ | X t − − Z t − | 22 ] ≥ ε t + E P [ | X t − − ˆ X t − | 22 ] .
This implies the claim.
Indeed, assume no such ε > 0 exists, then we have for each n ∈ N that λ k (cid:16) { t ∈ [0 , T ] | E P [ | X t − − Z t − | 22 ] ≥ 1 n + E P [ | X t − − ˆ X t − | 22 ] } (cid:17) = 0 .
Therefore, λ k (cid:16) { t ∈ [0 , T ] | E P [ | X t
[Equation]
| 22 ]
[Equation] [Equation]
n
[Equation]
[Equation] [Equation]
E
[Equation]
[ | X
[Equation]
Z
[Equation]
| 22 ] ≥
[Equation]
E P [ | X
[Equation]
ˆ X t − | 22 ] } (cid:17)
[Equation]
0 , which is a contradiction to λ k ( B ) > 0 .
Lemma E.5.
For any A -adapted process Z it holds that
[Equation]
Proof.
First recall that by continuity X t i =
X t i − .
Then the statement is a consequence of Proposition
B.2, Lemma B.3 and Fubini’s theorem, which imply
[Equation]
Lemma E.6.
Let 1 ≤ p < ∞ .
Let ( Z n ) n ∈ N be a sequence of random variables
[Equation]
and Z and ˜
Z random variables deﬁned on a common probability space such that Z n L p −−→ Z and Z n L p −−→ ˜ Z .
Then
Z = ˜
Z almost surely.
Proof.
With the triangle inequality it follows that E [ | Z − ˜ Z | p ] 1 /p
=
0 , which implies the claim.
Proof of Theorem E.2.
We start by showing that ˆ X ∈ D is the unique minimizer of Ψ up to ( P × λ k ) - null-sets for any k ≤ K .
First, we recall that for every t i we have ˆ X t i =
X t i and by continuity of X that X t
i
=
X t i − .
Therefore,
[Equation]
where we used Fubini’s theorem for the second line, Proposition B.2 for the third line and the triangle inequality for the fourth line.
Hence, ˆ X is a minimizer of Ψ .
To see that it is unique ( P × λ k ) -a.s.
, let Z ∈ D be a process such that ( P × λ k )
[ ˆ X (cid:54) =
Z ] > 0 .
By Lemma E.4, this implies that there exists an ε > 0 such that B := { t ∈ [0 , T ] | E P [ | X t − − Z t − | 22 ] ≥ (cid:15) + E P
[ | X t − − ˆ X t − | 22 ] } satisﬁes λ k ( B ) > 0 .
Now recall that by deﬁnition of λ k we have λ k ( B ) =
˜ P ( ˙ ∪ j ≥ k { n = j, t k − ∈ B } ) / ˜ P ( n ≥ k ) > 0 .
This implies that there exists j ∈ N ≥ k such that ˜ P ( n = j, t k − ∈ B ) > 0 .
Therefore,
[Equation]
This inequality implies now that Z is not a minimizer of Ψ , because
[Equation]
>
[Equation]
Ψ
[Equation]
Z )
[Equation]
Next we show that (29) can approximate ˆ X arbitrarily well.
Since the dimension d H can be chosen freely
[Equation]
let us ﬁx it to d H :=
d X .
Furthermore, let us ﬁx θ ∗ 3 such that g θ ∗ 3 = id .
From Theorem E.3 it follows that for any ε > 0 there exist M ∈ N and θ ∗ 1 , θ ∗ 2 with ( θ ∗ 1 , θ ∗ 2 , θ ∗ 3 ) ∈
Θ M such that
[Equation]
and sup (
[Equation] [Equation]
where we used that ˜ µ is continuous by Proposition B.4 and substituted u = tanh(˜ u ) , v = tanh(˜ v ) .
Using this, we can bound the distance between Y θ ∗ M t and ˆ X .
In particular, if t ∈ { t 1 , · · · , t n } , (36) yields
[Equation]
and if t not in { t 1 , · · · , t n } , then (12), (37) and the previous bound yield
[Equation]
Y
[Equation]
M t
[Equation]
[Table]
Now we can show the convergence of Φ( θ min M ) using these two bounds and X t i
=
ˆ X t i .
Indeed, min Z
[Equation]
Ψ
[Equation]
≤
[Equation]
[Table]
[Equation]
[Table]
[Equation]
where we used Ψ( ˆ X )
=
min Z ∈ D Ψ( Z ) and that for a ∈ R , a ≤ a 2 + 1 in the last step.
It is clear from the construction before, that ε → 0 as M → ∞ .
In the last step we show that the limits lim M →∞ Y θ min M and lim M →∞ Y θ ∗ M exist as limits in the
Banach space L :
= L 1 (Ω × [0 , T ] , F ⊗ B ([0 , T ]) , P × λ k ) , for every k ≤ K , and that they are both equal to ˆ X .
Let us ﬁx k ≤ K .
First we note that for every B ∈ B ([0 , T ]) we have
[Equation]
Using “measure theoretic induction” (Durrett, 2010, Case 1-4 of Proof of Thm. 1.6.9) this yields for c :
= ( P ( n ≥ k )) − 1 and a B ([0 , T ]) -measurable function Z : [0 , T ] → R , t
(cid:55)→ Z t :=
Z ( t )
that
[Equation]
Moreover, the triangle inequality and Lemma E.5 yield
[Equation]
For any R d X -valued Z ∈ L the Hölder inequality, together with the fact that n ≥ 1 , yields
[Equation]
Together, this implies that lim M →∞ Y θ ∗ M = ˆ X as a L -limit.
Indeed, with ˜ c := E P × ˜ P [ n ] 1 / 2 < ∞ we have
[Equation]
where we used ﬁrst (38) and (40) followed by two simply upper bounds and (39) in the last step.
The same argument can be applied to show that lim M →∞ Y θ min M =
ˆ X as a L -limit.
In particular, this proves that the limit Y :=
lim M →∞ Y θ min M exists as L -limit and by Lemma E.6 it equals ˆ X ( P × λ k ) -almost surely, for any k ≤ K .
Remark E.7.
This result can be extended to any other neural network architecture for which a universal approximation theorem equivalent to (Hornik et al., 1989, Thm. 2.4) exists.
Moreover, the stochastic process X deﬁned in (9) can be chosen more general, in particular, the diffusion part (cid:82) σdW can be replaced by any martingale, as long as the resulting process still is a Markov process and ˜ µ stays continuous.
Remark E.8.
If we used the modiﬁed loss function ˜Ψ which is identical to Ψ except that we drop the factor 1 n , everything would work similarly and we could show L 2 -convergence instead of L 1convergence of Y θ min M to ˆ X .
However, we remark that there might exists Z ∈ D ,
such that Ψ( Z ) < ∞ while ˜Ψ( Z ) = ∞ .
In particular, if some moment of n does not exist, such a process can be constructed.
Remark E.9.
It follows directly from Theorem E.2 that lim M →∞ Y θ min M
=
ˆ X as random variables on Ω × [0 , T ] except on sets which are null sets with respect to every product measure P × λ k for 1 ≤ k ≤ K .
Remark E.10.
The result of Theorem E.2 does not imply that ˆ X and lim M →∞ Y θ min M are modiﬁcations or indistinguishable.
For example, if B ⊂ [0 , T ] is a subset such that no left-point of the observation times ( t k − ) lies in B with probability greater 0 , i.e. λ k ( B )
=
for 1 ≤ k ≤ K , then
Theorem E.2 does not tell us how close (lim M →∞ Y θ min M ) t is to ˆ X t for t ∈ B .
In particular, it does not tell us whether they are equal P -almost surely.
Furthermore, such a set B always exists, since there has to exists t ∈ [0 , T ] such that B := { t } has measure 0 for all k .
In the following corollary we show, that Theorem E.2 can be extended to show convergence to the conditional expectation of ϕ ( X ) , for some function ϕ ∈ C 2 ,b ( R d X , R ) , i.e. a function that is twice continuously differentiable with bounded derivatives.
Corollary E.11.
Let ϕ ∈ C 2 ,
b
( R d X , R ) , then the statement of Theorem E.2 holds equivalently, when replacing X in the loss functions Ψ and Φ by Γ = ϕ
( X ) and ˆ X by the conditional expectation ˆΓ , where ˆΓ t := E P × ˜ P [ ϕ ( X t ) |A
t ] .
Corollary E.11 combined with the monotone convergence theorem for conditional expectation theoretically enables us to make statements about the conditional law and conditional moments of X under some a priori integrability assumptions.
Proof of Corollary E.11.
We ﬁrst remark that Proposition B.2 and Lemma E.4, E.5, B.3 hold similarly for the conditional expectation ˆΓ .
Hence, the same argument as in the proof of Theorem E.2 implies that ˆΓ is the unique A -adapted minimizer of Ψ .
For simplicity of the notation we assume that d X = d Y = 1 , i.e. that the process X and the Brownian motion W are 1 -dimensional.
However, the following works as well in the general case, where the correlations of the Brownian motion components have to be taken into account.
By Itô’s Formula (Protter, 2005, Chap.
II, Thm. 32), Γ = ϕ ( X ) is the solution of the SDE
[Equation]
for α ( t, X t ) :
= ϕ (cid:48)
( X t ) µ ( t, X t )
+ 12 ϕ
(cid:48)(cid:48)
( X t ) σ ( t, X t ) 2 and β ( t, X t ) :
= ϕ (cid:48)
( X t ) σ ( t, X t ) .
Deﬁning ˜
α
similar to ˜ µ as
[Equation]
one can use the boundedness of of ϕ (cid:48) and ϕ (cid:48)(cid:48) to show that it is continuous and that
[Equation]
In particular, Proposition B.1 and B.4 hold equivalently for Γ .
Similar to (36) and (37), the neural network parameters can be chosen such that sup
[Equation]
and sup
[Equation]
which then implies the statement of the Corollary similar as in the proof of Theorem E.2.
E. C ONVERGENCE OF THE M ONTE C ARLO APPROXIMATION
In the following, we assume the size of the neural network M is ﬁxed and we study the convergence with respect to the number of samples N .
Moreover, we show that both types of convergence can be combined.
To do so, we deﬁne ˜Θ M := { θ ∈ Θ M | | θ | 2 ≤ M } , which is a compact subspace of Θ M .
It is straight forward to see, that Θ M in Theorem E.2 can be replaced by ˜Θ M .
Indeed, if the needed neural network weights for an ε -approximation have too large norm, then one can increase M until it is sufﬁciently big.
Furthermore, we make the additional assumption that E ˜ P [ n 2 ] < ∞ .
The following convergence analysis is based on (Lapeyre & Lelong, 2019, Chapter 4.3).
Theorem E.12.
Let θ min M,N ∈ Θ min M,N := arg inf θ ∈ ˜Θ M { ˆΦ N ( θ ) } for every M, N ∈ N .
Then, for every
M ∈ N , ( P × ˜ P ) -a.s.
ˆΦ N N →∞ −−−−→ Φ uniformly on ˜Θ M .
Moreover, for every M ∈ N , ( P × ˜ P ) -a.s.
Φ( θ min M,N ) N →∞ −−−−→ Φ( θ min M ) and ˆΦ N ( θ min M,N ) N →∞ −−−−→ Φ( θ min M ) .
In particular, one can deﬁne an increasing sequence ( N M ) M ∈ N in N such that for every 1 ≤ k ≤ K we have that Y θ min M,NM converges to ˆ X for M → ∞ as random variable in L 1 (Ω × [0 , T ] , P × λ k ) .
In particular, the limit process Y :=
lim M →∞ Y θ min M,NM equals ˆ X ( P × λ k ) -almost surely as a random variable on Ω × [0 , T ] .
The following Monte Carlo convergence analysis is based on (Lapeyre & Lelong, 2019, Section 4.3).
In comparison to them, we do not need the additional assumptions that were essential in (Lapeyre & Lelong, 2019, Section 4.3), i.e. that all minimizing neural network weights generate the same neural network output.
This assumption is not needed, because we do not aim to show that Y θ min M,N converges to Y θ min M .
We deﬁne the separable Banach space S :
= { x = ( x i ) ∈ N ∈ (cid:96) 1 ( R d X )
| ∃ n ∈
N s.t. ∀ i > n : x i = 0 } with the norm (cid:107) x (cid:107) (cid:96) 1 :
= (cid:80) i ∈ N | x i
| 2 .
E.2. C ONVERGENCE OF O PTIMIZATION P ROBLEMS
Consider a sequence of real valued function ( f n ) n deﬁned on a compact set K ⊂ R d .
Deﬁne, v n = inf x ∈ K f n ( x ) and let x n be a sequence of minimizers f n ( x n ) = inf x ∈ K f n ( x ) .
From (Rubinstein & Shapiro, 1993, Theorem A1 and discussion thereafter) we have the following
Lemma.
Lemma E.13.
Assume that the sequence ( f n ) n converges uniformly on K to a continuous function f .
Let v ∗ = inf x ∈ K f ( x ) and S ∗ = { x ∈ K : f ( x ) = v ∗ } .
Then v n → v ∗ and d ( x n , S ∗ ) → 0 a.s.
The following lemma is a consequence of (Ledoux & Talagrand, 1991, Corollary 7.10) and (Rubinstein & Shapiro, 1993, Lemma A1).
Lemma E.14.
Let ( ξ i )
i ≥ 1 be a sequence of i.i.d random variables with values in S and h : R d ×S → R be a measurable function.
Here (cid:96) 1 denotes the space of sequences with absolutely convergent series.
Assume that a.s, the function θ ∈ R (cid:55)→ h ( θ, ξ 1 ) is continuous and for all C > 0 ,
E (sup | θ | 2 ≤ C | h
( θ, ξ 1 ) | ) <
+ ∞ .
Then, a.s θ ∈ R d (cid:55)→ 1 N (cid:80) Ni =1 h ( θ, ξ i ) converges locally uniformly to the continuous function θ ∈ R d (cid:55)→ E ( h ( θ, ξ 1 )) .
[Equation]
a.s
[Equation]
E.2. S TRONG LAW OF LARGE NUMBERS
Let us deﬁne
F
[Equation]
θ, x, y, z
[Equation]
and ξ j :=
( X j t j 1 , . . .
, X j t j nj , 0 , . . . ) , where X j t ji are random variables describing the realizations of the training data, as deﬁned in Section D.2.
By this deﬁnition we have n j := n j ( ξ j ) :=
max i ∈ N { ξ j,i (cid:54) =
0 } P -almost-surely and we know that ξ j are i.i.d.
random variables taking values in S .
Furthermore, let us write Y θt ( ξ ) to make the dependence of Y on the input explicit.
Then we deﬁne
[Equation]
Lemma E.15.
The following properties are satisﬁed.
( P 1 )
There exists κ > 0 such that for all S ∈ S and θ ∈ ˜Θ M we have | Y θt ( S ) | 2 ≤ κ n ( S ) for all t ∈ [0 , T ] , where n ( S ) =
max i ∈ N { S i (cid:54) = 0 } .
( P 2 ) Almost-surely the random function θ ∈ ˜Θ M
(cid:55)→ Y θt is uniformly continuous for every t ∈ [0 , T ] .
( P 3 )
We have E P × ˜ P (cid:2) 1 n (cid:80) n i =1 |
X t i | 22 (cid:3) < ∞ and E P × ˜ P (cid:2) 1 n (cid:80) n i =1 | X t i − | 22 (cid:3) < ∞ .
Proof.
By deﬁnition of the neural networks with sigmoid activation functions (in particular having bounded outputs), all neural network outputs are bounded in terms of the norm of the network weights, which is assumed to be bounded, not depending on the norm of the input.
Since Y jumps at most n ( S ) times, we can ﬁnd κ depending on T , such that the claimed bound is satisﬁed for all t , proving ( P 1 ) .
Since the activation functions are continuous, also the neural networks are continuous with respect to their weights θ , which implies that also θ ∈ ˜Θ M (cid:55)→ Y θt is continuous.
Since ˜Θ M is compact, this automatically yields uniform continuity and therefore ﬁnishes the proof of ( P 2 ) .
From our assumption that X is square integrable on the closed interval [0 , T ] , it follows that
[Equation]
t
[Equation]
E P [ | X t | 2 ] < ∞ .
Hence, by independence we have
[Equation]
and the same arguments shows E P × ˜ P (cid:2) 1 n (cid:80)
ni
=1 | X t i − | 22 (cid:3) < ∞ , therefore also ( P 2 ) is proven.
Proof of Theorem E.12.
We apply Lemma E.14 to the sequence of i.i.d random function h ( θ, ξ j ) .
From ( P 1 ) of Lemma E.15 we have that
[Equation]
Hence
[Equation]
we obtain that
[Equation]
implying that
[Equation]
using ( P 3 ) of Lemma E.15 and the stronger assumption that n is square integrable.
By ( P 2 ) of
Lemma E.15, the function θ (cid:55)→ h ( θ ) is continuous.
Therefore, we can apply Lemma E.14, yielding that almost-surely for N → ∞ the function
[Equation]
converges uniformly on ˜Θ M to
[Equation]
We deduce from Lemma E.13 that d ( θ min M,N , Θ min M ) → 0 a.s. when N → ∞ .
Then there exists a sequence (ˆ θ min M,N ) N ∈ N in Θ min M such that | θ min M,N − ˆ θ min M,N | 2 → 0 a.s. for N → ∞ .
The uniform continuity of the random functions θ (cid:55)→ Y θt on ˜Θ M implies that | Y θ min M,N t
[Equation]
| 2 → 0 a.s. when
N
→ ∞ for all t ∈ [0 , T ] .
By continuity of F this yields | h
( θ min M,N , ξ 1 ) − h
(ˆ θ min M,N , ξ 1 ) | 2 → 0 a.s. as
N
→ ∞ .
With (43) we can apply dominated convergence which yields
[Equation]
Since for every integrable random variable Z we have 0 ≤ | E
[ Z ] | 2 ≤ E [ | Z | 2 ] and since ˆ θ min M,N ∈ Θ min M we can deduce
[Equation]
[Equation]
Now by triangle inequality,
[Equation]
(44), (45) and (46) imply that both terms on the right hand side converge to 0 when N → ∞ , which ﬁnishes the proof of the ﬁrst part of the Theorem.
We deﬁne N 0 := 0 and for every M ∈ N
[Equation]
which is possibly due to (46).
Then Theorem E.2 implies that
[Equation]
Therefore, we can apply the same arguments as in the proof of Theorem E.2 (starting from (39)) to show that
[Equation]
for every 1 ≤ k ≤ K .
Corollary E.16.
In the setting of Theorem E.12, we also have that ( P × ˜ P ) -a.s.
Φ( θ min M,N M ) N →∞ −−−−→ Ψ( ˆ X ) and ˆΦ ˜ N M ( θ min M, ˜ N M ) N →∞ −−−−→ Ψ( ˆ X ) , where ( ˜ N M ) M ∈ N is another increasing sequence in N .
Proof.
The ﬁrst convergence result was already shown in the proof of Theorem E.12 and the second one can be shown similarly, when deﬁning ˜ N M by ˜ N 0 :
= 0 and for every M ∈ N ˜
[Equation]
which is possibly due to (47).
F E XPERIMENTAL DETAILS
All implementations were done using PyTorch.
The code is available.
F. I MPLEMENTATION DETAILS
Dataset.
For each of the SDE models (Black-Scholes, Ornstein-Uhlenbeck, Heston) a dataset was generated by sampling N = 20 (cid:48)
000 paths of the SDE using the Euler-scheme.
We used an equidistant time grid of mesh 0 .
01 between time 0 and T = 1 .
Independently for each path, observation times were sampled from P t , by using each of the grid points with probability 0 .
1 as an observation time.
In particular, n ∼ Bin(100 , 0 . 1) , t
= 0 and the observation times { t i } 1 ≤ i ≤ n were chosen uniformly on the time grid.
Hence, 10% of the grid points were used on average.
This way, n and t i are deﬁned as a discretized version of those given in Example A.2 where n is binomially distributed and t i chosen uniformly on [0 , T ] .
For each of these datasets, the samples were used in a 80% / 20% split for training and testing.
The SDEs of the dataset models and the chosen parameters are described below.
• Black-Scholes: – SDE: dX t =
µX t dt + σX t dW t , where W is a 1-dimensional Brownian motion – conditional expectation: E ( X t + s | X t ) =
X t e µs – used parameters: µ = 2 , σ = 0 .
3 , X 0 =• Ornstein-Uhlenbeck: – SDE: dX t = − k
( X t − m ) dt + σdW t , where W is a 1-dimensional Brownian motion – conditional expectation: E ( X t + s | X t )
=
X t e − ks + m (cid:0) 1 − e − ks (cid:1) – used parameters: k
= 2 , m = 4 , σ = 0 .
3 , X 0 =• Heston: – SDE: for W and Z 1-dimensional Brownian motions dX t =
µX t
[Equation]
– conditional expectation 7 : E ( X t + s | X t )
=
X t e µs – used parameters: µ = 2 , σ = 0 .
3 , X
= 1 k
= 2 , m = 4 , v 0 = 4 , ρ =
Corr( W, Z ) = 0 .
Architecture.
In our experiments we choose the dimension of the latent variable to be d H = 10 .
For f θ 1 , ˜ g θ 3 and ˜ ρ θ 2 we use 2-hidden-layer feed-forward neural networks, with 50 nodes in each hidden layer and tanh activation functions.
Then the neural networks g θ
and ρ θ 2 are deﬁned as residual versions of ˜ g θ 3 and ˜ ρ θ 2 , by adding a residual shortcut between the input and the output of the neural networks.
Moreover, also to the input of g θ 3 a tanh was applied, for reasons of symmetry.
Dropout was applied after each non-linearity with a rate of 0 . 1 .
As we applied tanh to the possibly unbounded inputs x and h in every neural network, there was no further need to normalize or scale the data.
To solve the ODE of the neural ODE part, the simple Euler-method was used.
Training.
The neural networks were trained using the Adam optimizer (Kingma & Ba, 2014) with a learning rate of 0 .
001 and weight decay 0 .
0005 for 200 epochs using a batch size of 200 .
A random initialization was used and no hyper-parameter optimization was needed.
Further training results.
Further training results on test samples are shown in Figure 8, 9, 10.
F. E XPERIMENTS ON OTHER DATASETS
We test our framework on additional synthetic datasets.
[Figure]
[Figure] [Figure]
F.2. H ESTON MODEL WITHOUT F ELLER CONDITION
If the Feller condition
[Equation]
is satisﬁed in the Heston model, it is known that the variance process v t is always strictly bigger than 0 and that the Euler-scheme works well to sample from it.
However, if the Feller condition is not satisﬁed, the variance process can touch 0 , where the process is reﬂected deterministically.
Then
the
Euler-scheme can not longer be used to sample from the model.
Moreover, this situation is generally considered as more delicate.
When close to 0 the distribution of v t behaves differently than further away of 0 (Andersen, 2007, Section 3).
As explained by Andersen (2007), there are differently well performing sampling schemes for a Heston model where the Feller condition is not satisﬁed.
We use the simplest to implement, which is a slight extension of the Euler scheme, where values of v t below 0 are replaced by 0 (Andersen, 2007, Section 2.3).
Although there is empirical evidence, that the resulting sampling distribution of v t close to 0 is not correctly replicating the true distribution (Jean-François et al., 2015, Figure 2,3), this method already produces sufﬁciently good sample paths.
Dataset.
Heston model, sampled as in Section F.1 with the extension described above.
Used parameters: µ = 2 , σ = 3 , X 0 = 1 k = 2 , m = 1 , v 0 = 0 .
5 , ρ =
Corr( W, Z ) = 0 .
Hence the
Feller condition and also the weaker condition 4 k m ≥ σ 2 discussed in (Jean-François et al., 2015,
Section 3.2), are both not satisﬁed.
We produce two datasets, a 1 -dimensional one similar to before, where only X is stored and a 2 -dimensional,one where both X and v are stored, hence also v is a target for prediction.
Note that v has the same conditional expectation as the Ornstein-Uhlenbeck
SDE.
In the 2-dimensional dataset, X and v are always observed at the same time.
Architecture & Training.
Same as in Section F.1, but with batch size 100 .
Results.
The model learns to replicate the true conditional expectation process, which is analytic, hence not effected by the sampling scheme.
In particular, we see that our model is very robust, since even in the delicate case where the Feller condition is not satisﬁed and a sampling scheme is used, that does not perfectly replicate the true distribution, our model still works well.
Due to the very similar results, in Figure 11 we only show plots on test samples of the 2 -dimensional dataset, where
X and v are predicted.
[Figure]
F.2. D ATASET WITH CHANGING REGIME
In this experiment we test, how well our model can deal with stochastic processes, that undergo an (abrupt) change of regime at a certain point in time.
This constitutes a type of time dependence of the dynamics of the stochastic process.
Many real world time series might exhibit such a change of regime.
Some examples are:
• A stock market that suddenly changes from a bullish to a bearish market, for example due to a macro-economic event.
An example for this would be the start of the Covid-19 crisis in the ﬁrst quarter of 2020.
• Longitudinal patient health recordings might experience changes depending on seasonal or longer-term inﬂuences, as for example due to the seasonal ﬂue or currently the Covidpandemic.
• In many regions climate data has strong seasonal dependencies, that can lead to relatively abrupt changes as for example when the weather changes from dry to rain season.
Dataset.
To be able to evaluate the performance of our model, we test a change of regime by combining two synthetic datasets.
On the ﬁrst half of the time interval [0 , 0 .
5] we use the OrnsteinUhlenbeck and on the second half [0 . 5 , 1] the Black-Scholes model.
The Black-Scholes process takes as starting point, the last point of the Ornstein-Uhlenbeck process.
We use the same hyperparameters for the dataset generation as in Section F.1, except that we set the parameter m = 10 in the Ornstein-Uhlenbeck model to make the two parts act on similar scales.
Architecture & Training.
Same as in Section F.1, but with batch size 100 .
Moreover, we used 100 neurons in each hidden layer, to account for the more complicated setting, where also a time dependence has to be learnt.
Results.
In the plots on test samples of the dataset shown in Figure 12 we see that our model correctly learns the change of regime.
[Figure]
D ATASET WITH EXPLICIT
Many real world datasets, have an explicit time dependence, i.e. the drift and diffusion of (9) explicitly depend on t .
Examples are all datasets that have a certain periodicity, as for example weather data (seasonal and daily periodicity), intraday periodicity of stock prices (Andersen & Bollerslev, 1997) or prices of certain seasonal goods.
We incorporate an explicit time dependence into the Black-Scholes dataset of Section F.1 to test how well our model can deal with such a situation.
Dataset.
We use the Black-Scholes datasets of Section F.1, where we replace the constant µ by the time dependent constant α 2 (sin( βt ) + 1) , for α, β > 0 .
The conditional expectation changes accordingly.
For the data generation we use the same hyper-parameters as in Section F.1, except that we use instead of µ the parameter α = 2 and β ∈ { 2 π, 4 π } .
Architecture & Training.
Same as in Section F.1, but with batch size 100 .
Moreover, we usedneurons in each hidden layer, to account for the more complicated setting, where an explicit time dependence has to be learnt.
Results.
We show plots on test samples of the datasets in Figure 5.
We see that the model learns to adapt to the time-dependent coefﬁcients.
F. D ETAILS ON CONVERGENCE STUDY
Evaluation metric.
For the sampling time grid with equidistant step size ∆ t :=
Tν , ν ∈ N , on [0 , T ] and the true and predicted conditional expectation for path j ∈ N , ˆ X j and Y j respectively, we deﬁne the evaluation metric as
[Equation]
where N 2 is the number of test samples, and j accordingly iterates over the paths in the test set.
Increasingly big training sets.
We use the following procedure to create increasingly big training sets, while keeping the exactly same test set for evaluation.
Out of the initial 20 (cid:48) 000 paths, we take N 2 := 4 (cid:48) 000 paths, which are ﬁxed as the testing set.
Out of the remaining 16 (cid:48) 000 paths, we randomly choose N 1 training paths for N 1 ∈ { 200 , 400 , 800 , 1 (cid:48) 600 , 3 (cid:48) 200 , 6 (cid:48) 400 , 12 (cid:48) 800 } .
Increasing neural network sizes.
The increasingly big neural networks are deﬁned as follows.
For all involved networks, we use the feed-forward 2-layer architecture with tanh activations (cf.
Appedix F.1), where each hidden layer has the same size M for M ∈ { 10 , 20 , 40 , 80 , 160 , 320 } .
Results on Black-Scholes dataset.
In accordance with the theoretical results in Theorem E.2 and
E.12, we see that the evaluation metric decreases when N 1 and M increase (Figure 13).
It is important to notice, that already a quite small number of samples can lead to a good approximation of the conditional expectation, if the network is big enough.
In particular, the Monte Carlo approximation of the theoretical loss function is good already with a few samples.
On the other hand, even with a large number of samples, the evaluation metric does not become so small, if the network size is not big enough.
From a practitioners point of view, this is good news, since increasing the network size is often much easier than collecting more training data.
Results on Ornstein-Uhlenbeck and Heston dataset.
We get very similar results on the Ornstein
Uhlenbeck (Figure 14) and Heston dataset (Figure 15).
Similar as for Black Scholes, also for
Ornstein-Uhlenbeck the training size N 1 is not as important as the network size M .
For all network sizes, increasing the training size further than 1600 hardly changes the performance, while increasing
M is crucial to get better performance.
In contrast to this, for the Heston dataset we see that a large number of training samples is more important to get a smaller convergence metric.
This reﬂects the fact, that the Heston dataset is more complex and more difﬁcult to learn.
F. D ETAILS ON COMPARISON TO GRU-ODE-B AYES
GRU-ODE-Bayes (Brouwer et al., 2019).
To the best of our knowledge, this is the neural network based method with the most similar task to ours.
In particular, this continuous time model is trained to learn the unknown temporal parameters of a normal distribution, best describing the conditional distribution of X given the previous observations.
This distribution is given by the Fokker-Planck equation.
Brouwer et al.
(2019) outlined, that their model can exactly represent the Fokker-Planck dynamics of the Ornstein-Uhlenbeck process, since the corresponding distribution is Gaussian.
Implementation of GRU-ODE-Bayes.
We use the code of the ofﬁcial implementation of (Brouwer et al., 2019) 8 and slightly adjust it for our purpose.
In particular, we do not use incomplete observations, hence the input mask used for this task has always only 1 -entries.
Furthermore, we slightly changed the scheme how the time steps are taken, to be the same as in our implementation, so that comparisons can be made.
Besides these minor changes, the original model is used and trained on all 3 datasets (Black-Scholes, Ornstein-Uhlenbeck and Heston).
We tried out all combinations of the following parameters and always chose the best performing one for comparison to our model: impute ∈ { True , False } , logvar ∈ { True , False } , mixing ∈ { 0 . 0001 , 0 . 5 } and hiddensize ∈ { 50 , 100 } , whereby phidden and prephidden were chosen to be equal to hiddensize .
The ﬁrst parameter choices were the ones used in the ofﬁcial implementation.
The model was always trained using the Euler-method for solving ODEs and with dropout = 0 . 1 , to be comparable to our implementation.
Furthermore, we always used the full GRU-ODE-Bayes implementation, since it should be the more powerful one.
For the comparison, we only use the estimated mean of the normal distribution (the estimated variance is not used), which is precisely the estimated conditional expectation of the model.
Implementation of NJ-ODE.
Same as described in Section F.1.
For each dataset only one model was trained, since we already saw the convergence properties before and wanted to have a qualitative comparison to GRU-ODE-Bayes.
In particular, we did not try to optimize our hyper-parameters for best performance (e.g. by choosing different number of layers or neurons or different activation functions) and used about 10 K trainable parameters compared to the best performing GRU-ODEBayes models of our study which used
112 K trainable parameters.
Datasets.
Same as described in Section F.1.
The train and test sets were ﬁxed to be the same for all trained models.
Training.
All models were trained using the Adam optimizer (Kingma & Ba, 2014) with a learning rate of 0 .
001 and weight decay 0 .
0005 for 100 epochs using a batch size of 20 .
A random initialization was used.
F.4. F URTHER RESULTS AND DISCUSSION OF THE COMPARISON
Evolution of losses and evaluation metric during training.
In Figure 16 and 17 we see, that already after a few epochs the NJ-ODE model ﬁnds close to optimal weights for the given network size on the Black-Scholes and Orstein-Uhlenbeck dataset and oscillates around this optimum.
The
Heston dataset is considerably more difﬁcult and the model slowly converges to close-to-optimal weights for the given network size.
In comparison to this, it takes the GRU-ODE-Bayes model longer to converge to its close-to-optimal weights on the Black-Scholes and Orstein-Uhlenbeck dataset.
Moreover, the model does not converge to close-to-optimal weights on the Heston dataset, but rather oscillates between bad weights.
Due to some very large outliers, this is not directly visible in the plot, but can be deduced from Table 6.4.
[Figure] [Figure]
Comparison of predicted paths.
For each dataset we show 5 paths that were predicted with NJODE and with GRU-ODE-Bayes, ﬁrst at the optimal epoch, i.e. where the test loss was minimal during training (Figures 18, 19, 20), and then at the last epoch (Figures 21, 22, 23).
For the sake of comparison, in each row the performance on the same test sample is shown.
The results are very similar for NJ-ODE and GRU-ODE-Bayes on the Black-Scholes and Ornstein-Uhlenbeck dataset, with sometimes the one and sometimes the other having a slightly better prediction.
On the Heston dataset, the results of NJ-ODE are good predictions, being correct whenever a new observation is made and not too far away even after longer periods without observations.
On the other hand,
GRU-ODE-Bayes is not even correct at times of new observations and learns an incorrect behaviour in between observations (e.g. making kinks where there should not be any).
For the last epoch, we see that this malfunction is ampliﬁed.
F.4. F URTHER EXPERIMENTS
Since the results of GRU-ODE-Bayes were unexpectedly bad on the Heston dataset
[Table]
we performed additional tests.
In particular, we retrained the best combinations of GRU-ODE-Bayes with the larger batch size 100 .
For smaller versions of the network (S)
, i
[Table]
e. with hiddensize =
prephidden = phidden = 50 this stabilized the training, such that the models converged.
Still, even at the best epoch, the models suffered from the same difﬁculties as explained in Section F.4.1.
Increasing the hiddensize by factor 2 to 100 (L), made the training unstable again, and the models did not converge any more.
This gives more empirical evidence, that GRU-ODE-Bayes can not be reliably trained on more complex datasets, where the target conditional distributions differs to much from a normal distribution.
In contrast to this, retraining NJ-ODE with batch size 100 , once for the smaller version described in Section F (S), once for a larger version with 100 instead of 50 neurons in all hidden layers (M) and once with 200 neurons (L), yielded the expected results of better performance with larger networks.
In particular, there is no instability for larger networks.
In Table F.4.1 we show results of our model and of the best GRU-ODE-model for the given size.
[Table]
F. R EAL WORLD DATASETS WITH INCOMPLETE OBSERVATIONS F.5. S ELF - IMPUTATION FOR INCOMPLETE OBSERVATIONS
In the theoretical description of our model, we always assumed that at each observation time all coordinates of X are observed.
However, in many real world dataset, the observations are incomplete, i.e. only for some of the coordinates an observation is available.
To deal with this, we propose the following self-imputation method.
Whenever an observation X t ∈ R
d
X is made the mask m t ∈ { 0 , 1 } d X tells whether the i -th coordinate of X t is observed ( m it = 1 ) or not ( m it = 0 ).
Then we use the prediction Y t − of our model to generate the imputed observation vector
[Equation]
where (cid:12) is the element-wise multiplication (Hadamar product) and 1 d X ∈ R d X is the one-vector.
Instead of X t we use ( ˜ X t , m t ) as an input for the jump part ρ θ 2 .
The intuition behind this deﬁnition is the following.
In the one dimensional case, if we do not make an observation, but input Y t − as if it was an observation, we expect that this does not change the output Y s for s ≥ t .
From this point of view, Y t − does not provide any additional information for the model.
Similarly, we expect that imputing Y t − for unobserved coordinates does not provide any information about this coordinate to the model.
However, since the model might learn to transfer the information about an observed coordinate to an unobserved one, we extend the input to also include the information which coordinates were observed.
For the ODE part, we use Y τ ( t ) , which is the prediction after processing the input ˜ X t , instead of X τ ( t ) .
Here the intuition is that if the model learns how to best use the incomplete observation, i.e. if the jump is good, then this is the best approximation for X τ ( t ) .
In accordance, the loss function is adjusted to only use the non-imputed coordinates, i.e. if m is the random process in { 0 , 1 } d X describing which coordinates are observed we have
[Equation]
C LIMATE FORECASTING DETAILS
Dataset.
We use the publicly available United State Historical Climatology Network (USHCN) daily dataset (Menne et al., 2016) together with all pre-processing steps as they were provided by Brouwer et al.
In particular, there are 5 sporadically observed (i.e. incomplete observations) climate variables (daily temperatures, precipitation, and snow) measured at 1 (cid:48) 114 stations scattered over the
United States during an observation window of 4 years (between 1996 and 2000) where each station has an average of 346 observations over those 4 years.
For 5 folds, the data is split into train ( 70% ), validation ( 20% ) and test ( 10% ) sets.
The task is to predict the next 3 measurements after the ﬁrst 3 years of observation.
The mean squared error between the prediction and the correct values is computed on the validation and test set.
Baselines.
We use the results reported in (Brouwer et al., 2019, Table 1) as baselines for our comparison and perform the exact same 5 -fold cross validation using the same folds with the same train, validation and test sets.
For completeness, we give all results of the table together with our results in Table 6.5.
We only show the mean squared error (MSE), since our model does not provide the negative log-likelihood.
Implementation of NJ-ODE.
We once use the architecture described in Section F.1 (S) and once use the exact same architecture, but with hidden size d H = 50 and 400 instead of 50 nodes in each hidden layer (L).
To deal with the incomplete observations, the self-imputation scheme described in
Section F.5 is used.
In particular, the data is self-imputed and passed together with the observation mask as input to the network ˜ ρ θ 2 .
Moreover, the network f θ 1 uses the NJ-ODE output at the last observation time instead of the last observation as input.
Training.
Both versions of NJ-ODE were trained using the Adam optimizer (Kingma & Ba, 2014) with a learning rate of 0 .
001 and weight decay 0 .
0005 for 100 epochs using a batch size of 100 .
A random initialization was used.
The performance on the validation set was used for early stopping after the ﬁrst 100 epochs, i.e. early stopping was possible at any epoch between { 101 , . .
Results.
The results of our model and all models reported in (Brouwer et al., 2019, Table 1) are shown in Table 6.5.
F.5. P HYSIONET PREDICTION DETAILS
Dataset.
We use the publicly available PhysioNet Challenge 2012 dataset (Goldberger et al., 2000) together with all pre-processing steps as they were described and provided by Rubanova et al.
In particular, there are 41 features of 8000 patients that are observed irregularly over a 48 hour time period.
The observation are put on a time grid with step size 0 .
016 hours leading to 3000 grid points.
While in the paper Rubanova et al.
(2019) say that they use 2880 grid points (i.e. minute wise) in their implementation they used 3000 .
Moreover, in contrast to what was written in the paper, theconstant features were not excluded in their implementation, hence we also keep them.
Furthermore, we also rescale the time-grid to [0 , 1] and normalize each feature as Rubanova et al.
(2019) did it for training the model.
The dataset is split with the same ﬁxed seed into 80% training and 20% test set.
In particular, no cross validation but only multiple runs with new random initializations are performed to be exactly comparable to the results reported by Rubanova et al.
On the test set, the observation paths are split in half.
The ﬁrst half (ﬁrst 1500 time steps) is used as input for the model, from which the second half (second 1500 time steps) should be forecast.
The masked mean squared error (MSE) with a certain balancing procedure between the predictions and the true values is reported.
For comparability we use the exact same implementation of the MSE as Rubanova et al.
Baselines.
We compare the performance of our model to latent ODE on the extrapolation task (as described above) on physionet.
As baseline for our comparison we use the results reported in (Rubanova et al., 2019, Table 5).
For completeness, we show all extrapolation results of the table together with our results in Table 6.5.
We shortly outline the different approach of latent ODE compared to our model for the given extrapolation task.
Latent ODE also splits the training samples similar to the test samples in half, using the ﬁrst half as input and the second half as target.
It is trained as an encoder-decoder, encoding the observations in the ﬁrst half and reconstructing (decoding) the second half.
This falls in the standard supervised learning framework.
In particular, this approach cannot straight forward be extended for online forecasting.
Moreover, this approach might learn certain path dependencies.
On the other hand, our model is trained as always, online forecasting after each observation until the next observation is made.
Instead of splitting the training samples we use the entire path as input for our unsupervised training framework.
Our model is based on the assumption that paths are Markov, therefore it cannot learn path dependencies, i.e. dependencies on more than just the last observation.
However, by training the model also on the second half of the training samples, it learns the underlying behaviour there, which should be helpful for the extrapolation task.
Implementation of NJ-ODE.
We use the architecture described in Section F.1, but with hidden size d H = 41 and 200 nodes in each hidden layer.
To deal with the incomplete observations, the self-imputation scheme described in Section F.5 is used.
In particular, the data is self-imputed and passed together with the observation mask as input to the network ˜ ρ θ 2 .
Moreover, the network f θuses the NJ-ODE output at the last observation time instead of the last observation as input.
Training.
All versions of NJ-ODE were trained using the Adam optimizer (Kingma & Ba, 2014) with a learning rate of 0 .
001 and weight decay 0 .
0005 for 400 epochs using a batch size of 50 .
A random initialization was used.
Results.
The results of our model and all models reported in (Rubanova et al., 2019, Table 5, extrapolation) are shown in Table 6.5.
[Figure] [Figure]
[Figure] [Figure]
[Figure] [Figure]