Infinite-dimensional reservoir computing
Lukas Gonon, Lyudmila Grigoryeva, Juan-Pablo Ortega
Introduction
Reservoir computing (RC) [Jaeg 10, Maas 02, Jaeg 04, Maas 11] and in particular echo state networks (ESNs) [Matt 92, Matt 93, Jaeg 04] have gained much popularity in recent years due to their excellent performance in the forecasting of dynamical systems [Grig 14, Jaeg 04, Path 17, Path 18, Lu 18, Wikn 21, Arco 22] and due to the ease of their implementation. RC aims at approximating nonlinear input/output systems using randomly generated state-space systems (called reservoirs) in which only a linear readout is estimated. It has been theoretically established that this is indeed possible in a variety of deterministic and stochastic contexts [Grig 18b, Grig 18a, Gono 20c, Gono 21b, Gono 23] in which RC systems have been shown to have universal approximation properties.
In this paper, we focus on deriving error bounds for a variant of the architectures that we just cited and consider as approximants randomly generated linear systems with readouts given by randomly generated neural networks in which only the output layer is trained. Thus, from a learning perspective, we combine linear echo state networks and what is referred to in the literature as random features [Rahi 07] /extreme learning machines (ELMs) [Huan 06]. We develop explicit and readily computable approximation and estimation bounds for a newly introduced concept class whose elements we refer to as recurrent (generalized) Barron functionals since they can be viewed as a dynamical analog of the (generalized) Barron functions introduced in [Barr 92, Barr 93] and extended later in [E 20b, E 20a, E 19]. The main novelty in this concept class with respect to others already available in the literature, like the fading memory class [Boyd 85] in deterministic setups or functionals in the stochastic case, is that it consists of elements that admit an explicit infinite-dimensional state-space representation, which makes them analytically tractable. As we shall see later on, many interesting families of input/output systems belong to this class that, additionally, is universal in the -sense.
From an approximation-theoretical point of view, the universality properties of linear systems with readouts belonging to dense families have been known for a long time both in deterministic [Boyd 85, Grig 18b] and in stochastic [Gono 20c] setups. Their corresponding dynamical and memory properties have been extensively studied (see, for instance, [Herm 10, Coui 16, Tino 18, Tino 20, Gono 20a, Li 22] and references therein). Our contribution in this paper can hence be considered as an extension of those works to the recurrent Barron class.
All the dynamic learning works cited so far use exclusively finite-dimensional state spaces. Hence, one of the main novelties of our contribution is the infinite-dimensional component in the concept class that we propose. It is worth mentioning that, even though in static setups, there exist numerous neural functional approximation results (see, for instance, [Chen 95, Stin 99, Krat 20, Bent 22, Cuch 22, Neuf 22], in addition to the works on Barron functions cited above), the use of infinite-dimensional state-space systems has not been much exploited, and it is only very recently that it is being seriously developed. See [Herm 12, Bouv 17b, Bouv 17a, Kira 19, Li 20, Kova 21, Acci 22, Gali 22, Hu 22, Salv 22] for a few examples. Dedicated hardware realizations of RC systems using quantum systems are a potential application of these extensions for which, in the absence of adequate tools, most of the theoretical developments have been carried out so far in exclusively finite-dimensional setups [Chen 19, Chen 20, Tran 20, Tran 21, Chen 22, Mart 23].
In order to introduce our contributions in more detail, we start by recalling that an echo state network (ESN) is an input/output system defined by the two equations:
The second step consists in obtaining generalization error bounds that match the parameter restrictions emerging from the approximation results for these systems. The key challenge here is that the observational data is non-IID and so classical statistical learning techniques [Bouc 13] can not be employed directly, see [Gono 20b]. By combining these bounds, we then obtain learning error bounds for such random recurrent neural networks for learning a general class of maps . As a by-product, we obtain new universality results for reservoir systems, see [Grig 18b, Grig 18a, Gono 20c, Gono 23, Gono 21b], with generically randomly generated coefficients. It is worth emphasizing that the construction that we describe in this paper yields a fully implementable recurrent neural network-based learning algorithm with provable convergence guarantees not suffering from the curse of dimensionality.
A dynamic analog of the generalized Barron functions
2 Definition and properties of recurrent Barron functionals
In this section we define the class of recurrent generalized Barron functionals and study some properties of these functionals.
and, writing ,
We denote by the class of all recurrent generalized Barron functionals or just recurrent Barron functionals.
Note that the readout (2.2) is built in such a way that the output is constructed out of and instead of exclusively , like it is customary in reservoir computing (see (1.2)). The reason for this choice is that it will allow us later on in Section 3.6 to recover the static situation as a particular case of the recurrent setup in a straightforward manner.
The unique solution property (USP). The existence and uniqueness of solutions property of state equations of the type (2.1), which is part of the definition of the recurrent Barron functionals, is a well-studied problem. This property is referred in the literature as the echo state property (ESP) [Jaeg 10, Yild 12, Manj 13] or the unique solution property (USP) [G Ma 20, Manj 22a] and it has been much tackled in deterministic (see references above and [Grig 18a], for instance), and stochastic setups [Manj 22b]. The following proposition is an infinite-dimensional adaptation of a commonly used sufficient condition for the USP to hold in the case of (2.1), as well as a full characterization of it when the activation function is the identity, and hence (2.1) is a linear system. In both cases, the inputs are assumed to be bounded; possible generalizations to the unbounded case can be found in [Grig 19].
Consider the state equation given by (2.1) and the two following cases:
The proof of part (i) is a straightforward generalization of [Grig 18a, Theorem 3.1(ii)] together with the observation that the hypothesis implies that the state equation is a contraction on the state entry. Part (ii) can be obtained by mimicking the proof of [Grig 21b, Proposition 4.2(i)]; a key element in that proof is the use of Gelfand’s formula for the spectral radius, which holds for operators between infinite-dimensional Banach spaces [Lax 02, Theorem 4, page 195]. ∎
The concept class of recurrent Barron functionals is a vector space.
Next, define as for and set , where denotes the pushforward of under . Then the integral transformation theorem shows that
and, by (2.5), for all ,
Hence, , as claimed. ∎
and, writing ,
We shall refer to the elements in as finite-dimensional recurrent Barron functionals. In the following proposition, we show that the set of finite-dimensional recurrent Barron functionals can be naturally seen as a subspace of . In the statement, we use the notion and properties of system morphisms that are recalled, for instance, in [Grig 21b].
that is, every finite-dimensional recurrent Barron functional admits a realization as a recurrent Barron functional.
First, we notice that is system equivariant because
which proves that in (2.2) defines an element in . Finally, we show that the map is readout invariant using that the first components of each solution are given by and applying again the change-of-variables theorem:
for , as required. In particular, and are in fact the same functional, which proves the inclusion (2.8). ∎
3 Examples of recurrent Barron functionals
We now show that large classes of input/output systems naturally belong to the recurrent Barron class.
Finite-memory functionals. The following proposition shows that finite-memory causal and time-invariant systems that have certain regularity properties can be written as finite-dimensional recurrent Barron functionals with a linear state equation.
But is continuous in by [Evan 10, Theorem 6(ii) in Chapter 5.6] and also the right-hand side in (2.11) is continuous in , hence the representation (2.11) holds for all .
which shows that is a finite-dimensional recurrent Barron functional, that is . ∎
Linear functionals. The following propositions show that certain linear functionals are also in the recurrent Barron class .
Let be as in (i) or, in case (ii) holds, let .
since for and for . Furthermore, for any ,
which proves that in the case .
For the case , let be defined by and consider . Then the change-of-variables theorem and the fact that (as established above) show that and by (2.14) and we obtain
hence and, by Proposition 2.3, also . ∎
The next proposition also covers the linear case under slightly different hypotheses.
As a further example, we now see that any functional associated with a linear state-space system on a separable Hilbert space admits a realization as an element in for .
This shows that the functional can be represented as a readout (2.2) of a system (2.1), hence . It also proves that is a system isomorphism between the system determined by , , , and , and the system in determined by , , , and . ∎
The next proposition is a generalization of a universality result in [Gono 20c, Theorem III.13] to the context of recurrent Barron functionals.
This shows that and . Combining this with Proposition 2.4 yields the claim.
In the case where is bounded, continuous and non-constant we may directly work with (instead of ) and proceed similarly. ∎
Generalized convolutional functionals. We conclude by showing that the class contains functionals that generalize convolutional filters, an important family of transforms.
Notice that standard convolutional filters are a special case of the functional in (2.21) when is a Dirac measure and is the identity.
Approximation
In this section, we establish approximation results for the concept class of recurrent Barron functionals. As approximating systems, we use a modification of the finite-dimensional echo state networks (1.1)–(1.2) in which the linear readout in (1.2) is replaced by an extreme learning machine (ELM) / random features neural network, that is, a neural network of the type (1.3) whose parameters are all randomly generated, except for the output layer given by the weights which are trainable. To derive such bounds, we proceed in several approximation steps and (a) approximate the infinite system (2.1) by a finite system of type (1.1), (b) apply a Monte Carlo approximation of the integral in (2.2), and (c) use an importance sampling procedure to guarantee that the neural network weights can be generated from a generic distribution rather than the (unknown) measure .
For each the system
This can now be used to estimate the difference between and its truncation:
and, writing ,
This proves that . ∎
2 Normalized realizations
The next result shows that a large class of elements in the concept class can be transformed into what we call a normalized realization in which the solution map for (2.1) is a multiplicative scaling operator.
since and is bounded. The second term in (3.12) can be rewritten as
Combining (3.13) and (3.12) and inserting in (2.2) then yields
Let be given as and denote by the pushforward measure of under . Then the change-of-variables formula and (3.11) show that
and similarly we obtain from (3.14) that for any ,
3 Echo state network approximation of normalized, truncated functionals
Consider now the setting of reservoir computing, where we aim to approximate an unknown element in our class using a finite-dimensional ELM-ESN pair, that is, a randomly generated echo state network-like state equation with a neural network as readout in which only the output layer is trained. As a first step, we consider in this section the situation when the target functional is in the class of finite-dimensional recurrent Barron functionals.
More specifically, consider the echo state network
Let and define the Kalman controlability matrix
where, in case , the map removes the last columns.
In order to obtain the last inequality in the previous expression, we decomposed , with , which allowed us to write
4 Main approximation result
This section contains our main approximation result, in which we extend the possibility of approximating the elements in the class of finite-dimensional recurrent Barron functionals, using finite-dimensional ELM-ESN pairs, to the full class of recurrent Barron functionals.
Using this notation, we shall now formulate a result that shows that the elements in that admit a representation in which the state equation (2.4) is linear and has the unique solution property, can be approximated arbitrarily well by randomly generated ESN-ELM pairs in which only the output layer of the ELM is trained. We provide an approximation bound where the dependence on the dimensions of the ESN state space and of the input is explicitly spelled out.
holds. Notice that from the proof of Proposition 2.6 we obtain that , where is the unique solution to
satisfies and that
for all and by (3.11)
Choose such that for all . Furthermore, note that and . Hence, using and for all we obtain from (3.33)
satisfies for all the bound (3.22), that is,
Combining this with (3.40), (3.34), (3.37) and (3.38) then yields
A bound similar to the one in (3.27) can be obtained if the ESN-ELM used as approximant satisfies the more general condition . The theorem is proved in that case by replacing in (3.37) the use of the inequality (3.22) by its more general version (3.21).
Implementation. From a practical point of view, the bounds obtained in Theorem 3.7 and in Corollary 3.9 apply to two different learning procedures: in the first case, when the chosen ELM weight distribution satisfies the absolute continuity assumption in Theorem 3.7, only the ELM output layer needs to be trained. In the second case, when that condition is not available, all the ELM weights also need to be trained. However, note that these are not recurrent weights; consequently, the vanishing gradient problem does not occur here. In contrast, this problem sometimes occurs for standard recurrent neural networks in which all weights are trained by backpropagation through time.
We conclude this section by considering another implementation scenario in which we impose the measure that we use for the sampling of the ELM, and show that it is still possible to sample using measures that are arbitrarily close to it in the sense of the -Wasserstein metric in exchange for potentially increasing an error term that can be compensated by increasing the dimension of the ESN. Let be the -Wasserstein metric. Suppose that we use the measure to generate the hidden layer weights. The next corollary shows how the error increases when we sample “almost” from the given measure .
We now derive an alternative bound (regardless of whether is finite or not). Combining both bounds then yields the bound (3.45).
From (3.39) and the estimate we obtain
5 Universality
As an additional result, we obtain the following universality result, which shows that any square-integrable functional (not necessarily in the recurrent Barron class) can be well approximated by the ESN-ELM (3.16)-(3.17) with suitably chosen readout and hidden weights generated from a distribution arbitrarily close to a prescribed measure .
Firstly, by Proposition 2.10 there exists such that . From the proof of Proposition 2.10 and the construction in [Gono 20c] it follows that is of the form
Let be defined by and denote by the pushforward measure of under . Then
Thus, has a representation of the type (2.2) with . Furthermore, since is atomic it follows directly that and .
which does not depend on . Similarly, the bound
the assumptions on the ESN matrices, and (3.47) can be used to obtain an upper bound on that only depends on , , , , , , and , but not on . Set .
We now apply Corollary 3.12 to . We select with and
6 Special case: static situation
In this situation, we show that such static elements of the recurrent Barron class can be approximated by ELMs, that is, by feedforward neural networks
As noted above, is a recurrent generalized Barron functional with
for any for which the integrand in the last line is integrable with respect to . This shows that in (3.36) coincides with the map
and furthermore . Therefore, denoting by also the functional , in the first step of the error estimate (3.42) only the last term is non-zero.
Furthermore, and thus . Therefore, the hypothesis implies .
Choosing , from (3.40) we thus obtain
The expectation in (3.57) can be estimated by
with . ∎
Learning from a single trajectory
where we denote .
The problem (4.2) admits an explicit solution (see, for instance, [Gono 21a, Section 4.3]). Once computed, the learned functional is . The learning performance of the algorithm can now be evaluated by assessing its learning error (or generalization error)
where is an IID copy of and is independent of all other random variables introduced so far. An analysis of the learning error is carried out below in Theorem 4.2, in which we assume, inspired by the developments in [Gono 20b], that the input and the output processes are of the type introduced in the following definition.
where we used in the last step that (4.2) implies .
Altogether, the hypotheses of [Gono 20b, Corollary 8(ii)] are satisfied. Writing for the idealized empirical risk and applying this result proves that
Recall that was considered fixed; now we take expectation also with respect to its distribution and obtain in the same way as we obtained (4.12)
The expectations in (4.14) can be estimated using Jensen’s inequality and (3.23):
Combining this with (4.6), (4.7) and (4.8) we thus obtain
and inserting (3.28) then shows (4.3) with (4.4) and (4.5), as claimed. ∎
The bound in Theorem 4.2 can be directly extended to the multivariate case as follows: if and each satisfies the hypotheses formulated in Theorem 4.2, then
Conclusions
In this paper we have provided reservoir computing approximation and generalization bounds for a new concept class of input/output systems that extends to a dynamical context the so-called generalized Barron functionals. We call the elements of the new class recurrent generalized Barron functionals. Its elements are characterized by the availability of readouts with a certain integral representation built on infinite-dimensional state-space systems. We have also shown that they form a vector space and that the class contains as a subset systems of this type with finite-dimensional state-space representations. This class has been shown to be very rich: it contains all sufficiently smooth finite-memory functionals, linear and convolutional filters, as well as any input-output system built using separable Hilbert state spaces with readouts of integral Barron type. Moreover, this class is dense in the -sense.
The reservoir architectures used for the approximation/estimation of elements in the new class are randomly generated echo state networks with either linear (for approximation) or ReLU (estimation) activation functions, and with readouts built using randomly generated neural networks, in which only the output layer is trained (extreme learning machines or random feature neural networks). The results in the paper yield a fully implementable recurrent neural network-based learning algorithm with provable convergence guarantees that does not suffer from the curse of dimensionality.
Acknowledgments. The authors acknowledge partial financial support coming from the Swiss National Science Foundation (grant number 200021_175801/1). L. Grigoryeva and J.-P. Ortega wish to thank the hospitality of the University of Munich and L. Gonon that of the Nanyang Technological University during the academic visits in which some of this work was developed. We thank Christa Cuchiero and Josef Teichmann for interesting exchanges that inspired some of the results in this paper.