A NTISYMMETRIC RNN: A D YNAMICAL S YSTEM V IEW ON R ECURRENT N EURAL N ETWORKS
Anonymous authors
Paper under double-blind review
A BSTRACT
Recurrent neural networks have gained widespread use in modeling sequential data.
Learning long-term dependencies using these models remains difﬁcult though, due to exploding or vanishing gradients.
In this paper, we draw connections between recurrent networks and ordinary differential equations.
A special form of recurrent network called the AntisymmetricRNN is proposed under this theoretical framework, which is able to capture long-term dependencies thanks to the stability property of its underlying differential equation.
In comparison to existing approaches for improving RNN trainability, which often incur signiﬁcant computation overhead, AntisymmetricRNN achieves the same goal by design.
We showcase the advantage of this new architecture through extensive simulations and experiments.
AntisymmetricRNN exhibits much more predictable dynamics.
It outperforms regular LSTM models on tasks requiring long-term memory, and matches the performance on tasks where short-term dependencies dominate despite being much simpler.
I NTRODUCTION
Recurrent neural networks (RNNs) (Rumelhart et al., 1986; Elman, 1990) have found widespread use across a variety of domains from language modeling (Mikolov et al., 2010;
Kiros et al., 2015; Jozefowicz et al., 2016) and machine translation (Bahdanau et al., 2014) to speech recognition (Graves et al., 2013) and recommendation systems (Hidasi et al., 2015; Wu et al., 2017).
Modeling complex temporal dependencies in sequential data using RNNs, especially the long-term dependencies, remains an open challenge.
The main difﬁculty arises as the error signal back-propagated through time (BPTT) suffers from exponential growth or decay, a dilemma commonly referred to as exploding or vanishing gradient (Pascanu et al., 2012; Bengio et al., 1994).
Vanilla RNNs as originally proposed are particularly prone to these issues and are rarely used in practice.
Gated variants of RNNs, such as long short-term memory (LSTM) networks (Hochreiter & Schmidhuber, 1997) and gated recurrent units (GRU) (Cho et al., 2014) that feature various forms of “gating” are proposed to alleviate these issues.
The gates allow information to ﬂow from inputs at any previous time steps to the end of the sequence more easily, partially addressing the vanishing gradient problem (Collins et al., 2016).
In practice, these models must be paired with techniques such as normalization layers (Ioffe & Szegedy, 2015; Ba et al., 2016) and gradient clipping (Pascanu et al., 2013) to achieve good performance.
Identity and orthogonal initialization is another proposed solution to the exploding or vanishing gradient problem of deep neural networks (Le et al., 2015; Mishkin & Matas, 2015; Saxe et al., 2013; Chen et al., 2018a).
Recently, Arjovsky et al.
; Wisdom et al.
; Hyland & R¨atsch (2017); Xie et al.
; Zhang et al. (2018a) advocate going beyond initialization and forcing the weight matrices to be orthogonal throughout the entire learning process.
However, some of these approaches come with signiﬁcant computational overhead and reportedly hinder representation power of these models (Vorontsov et al., 2017).
Moreover, orthogonal weight matrices alone does not prevent exploding and vanishing gradients, due to the nonlinear nature of deep neural networks as shown in (Pennington et al., 2017).
Here we offer a new perspective on the trainability of RNNs from the dynamical system viewpoint.
While exploding gradient is a manifestation of instability of the underlying dynamical system, vanishing gradient results from a lossy system, properties that have been widely studied in the dynamical system literature (Haber & Ruthotto, 2017; Laurent & von Brecht, 2017).
The main contributions of the work are:
• We draw connections between RNNs and the ordinary differential equation theory, and design new recurrent architectures by discretizing ODEs.
• The stability of the ODE solutions and the numerical methods for solving ODEs lead us to design a special form of RNNs, which we name AntisymmetricRNN , that can capture longterm dependencies in the inputs.
The construction of the model is much simpler compared to the existing methods for improving RNN trainability.
• We conduct extensive simulations and experiments to demonstrate the beneﬁts of this new RNN architecture.
AntisymmetricRNN exhibits well-behaved dynamics and outperforms the regular LSTM model on tasks requiring long-term memory, and matches its performance on tasks where short-term dependencies dominate with much fewer parameters.
R ELATED WORK
Trainability of RNNs.
Capturing long-term dependencies using RNNs has been a long-standing research topic with various approaches proposed.
The ﬁrst group of approaches mitigates the exploding or vanishing gradient issues through introducing some forms of gating.
Long short-term memory networks (LSTM) (Hochreiter & Schmidhuber, 1997) and gated recurrent units (GRU) (Cho et al., 2014) are the most prominent models along this line of work.
Parallel efforts include designing special neural architectures, such as hierarchical RNNs (El Hihi & Bengio, 1996), recursive neural networks (Socher et al., 2011), attention networks (Bahdanau et al., 2014), dilated convolutions (Yu & Koltun, 2015), recurrent batch normalization (Cooijmans et al., 2017), residual RNNs (Yue et al., 2018), and Fourier recurrent units (Zhang et al., 2018b).
These, however, are fundamentally different networks with different tradeoffs.
Another direction is to constrain the weight matrices of an RNN so that the back-propagated error signal is well conditioned, in particular, the input-output Jacobian having unitary singular values.
Le et al. (2015); Mikolov et al.
; Mishkin & Matas (2015) propose the use of identity or orthogonal matrix to initialize the recurrent weight matrix.
Followup works further constrain the weight matrix throughout the entire learning process either through re-parametrization (Arjovsky et al., 2016), geodesic gradient descent on the Stiefel manifold (Wisdom et al., 2016; Vorontsov et al., 2017), or constraining the singular values (Jose et al., 2017; Kanai et al., 2017; Zhang et al., 2018a).
It is worth noting that, orthogonal weights by themselves do not guarantee unitary Jacobians.
Nonlinear activations still cause the gradients to explode or vanish.
Contractive maps such as sigmoid and hyperbolic tangent leads to vanishing gradients.
Chen et al.
(2018a) offer an initialization scheme taking into account the nonlinearity.
However, the theory developed relies heavily on the random matrix assumptions, which only hold at the initialization point of training.
Although it has been shown to predict trainability beyond initialization.
Dynamical systems view of recurrent networks.
Connections between dynamical systems and
RNNs have not been well explored.
Laurent & von Brecht (2017) study the behavior of dynamical systems induced by recurrent networks, and show that LSTMs and GRUs exhibit chaotic dynamics in the absence of input data.
They propose a simpliﬁed gated RNN named the chaos free network (CFN), that has non-chaotic dynamics and achieves comparable performance to LSTMs and GRUs on language modeling.
Tallec & Ollivier (2018) formulate RNNs as a time-discretized version of
ODE and show that time invariance leads to gate-like mechanisms in RNNs.
With this formulation, the authors propose an initialization scheme by setting the initial gate bias according to the range of time dependencies to capture.
Dynamical systems view of residual networks.
Another line of work that is closely related to ours is the dynamical systems view on residual networks (ResNets)
(He et al., 2016).
Haber & Ruthotto (2017); Chang et al.
(2018a;b);
Lu et al.
(2018) propose to interpret ResNets as ordinary differential equations (ODEs), under which learning the network parameters is equivalent to solve a parameter estimation problem involving the ODE.
Chen et al.
(2018b) parameterize the continuous dynamics of hidden units using an ODE speciﬁed by a neural network.
Stable and reversible architectures are developed (Haber & Ruthotto, 2017; Chang et al., 2018a) from this viewpoint, which form the basis of our proposed recurrent networks.
A NTISYMMETRIC RNN S 3. O RDINARY DIFFERENTIAL EQUATIONS
We ﬁrst give a brief overview of the ordinary differential equations (ODEs), a special kind of dynamical systems that involves a single variable, time t in this case.
Consider the ﬁrst-order ODE
[Equation]
for time t ≥ 0 , where h ( t ) ∈ R n and f : R n → R n .
Together with a given initial condition h (0) , the problem of solving for the function h ( t ) is called the initial value problem .
For most ODEs, it is impossible to ﬁnd an analytic solution.
Instead, numerical methods relying on discretization are commonly used to approximate the solution.
The forward Euler method is probably the best known and simplest numerical method for approximation.
One way to derive the forward Euler method is to approximate the derivative on the left-hand side of Equation 1 by a ﬁnite difference, and evaluate the right-hand side at h t − 1 :
[Equation]
Note that for the approximation to be valid, (cid:15) > 0 should be small by the deﬁnition of derivative.
One can easily prove that forward Euler method converges linearly w.r.t.
(cid:15) , assuming f ( h ) is Lipschitz continuous on h and that the eigenvalues of the Jacobian of f has negative real parts.
Rearranging it, we have the forward Euler method for a given initial value
[Equation]
Geometrically, each forward Euler step takes a small step along the tangential direction to the exact trajectory starting at h t − 1 .
As a result, (cid:15) is usually referred to as the step size.
As an example, consider the ODE
[Equation]
The forward Euler method approximates the solution to the ODE iteratively as
[Equation]
which can be regarded as a recurrent network without input data.
Here h t is the hidden state at the t -th step, W is a model parameter, and (cid:15) is a hyperparameter.
This provides a general framework of designing recurrent network architectures by discretizing ODEs.
As a result, we can design
ODEs that possess desirable properties by exploiting the theoretical successes of dynamical systems, and the resulting recurrent networks will inherit these properties.
Stability is one of the important properties to consider, which we will discuss in the next section.
It is worth mentioning that the “skip connection” in this architecture resembles the residual RNN (Yue et al., 2018) and the Fourier
RNN (Zhang et al., 2018b), which are proposed to mitigate the vanshing and exploding gradient issues.
3. S TABILITY OF ORDINARY DIFFERENTIAL EQUATIONS : A NTISYMMETRIC RNN S
In numerical analysis, stability theory addresses the stability of solutions of ODEs under small perturbations of initial conditions.
In this section, we are going to establish the connections between the stability of an ODE and the trainability of the RNNs by discretizing the ODE, and design a new
RNN architecture that is stable and capable of capturing long-term dependencies.
An ODE solution is stable if the long-term behavior of the system does not depend signiﬁcantly on the initial conditions.
A formal deﬁnition is given as follows.
Deﬁnition 1.
(Stability)
A solution h ( t ) of the ODE in Equation 1 with initial condition h (0) is stable if for any (cid:15) > 0 , there exists a δ > 0 such that any other solution ˜
h ( t ) of the ODE with initial condition ˜ h (0) satisfying | h (0) − ˜ h
| ≤ δ also satisﬁes | h ( t ) −
˜ h
( t ) | ≤ (cid:15) , for all t ≥ 0 .
In plain language, given a small perturbation of size δ of the initial state, the effect of the perturbation on the subsequent states is no bigger than (cid:15) .
The eigenvalues of the Jacobian matrix play a central role in stability analysis.
Let J ( t ) ∈
R n × n be the Jacobian matrix of f , and λ i ( · ) denotes the i -th eigenvalue.
Proposition 1.
The solution of an ODE is stable if
[Equation]
where Re ( · ) denotes the real part of a complex number.
A more precise proposition that involves the kinematic eigenvalues of J ( t ) is given in Ascher et al.
Stability alone however, does not sufﬁce to capture long-term dependencies.
As argued in Haber & Ruthotto (2017), Re ( λ i ( J ( t )))
(cid:28) 0 results in a lossy system; the energy or signal in the initial state is dissipated over time.
Using such an ODE as the underlying dynamical system of a recurrent network will lead to catastrophic forgetting of the past inputs during the forward propagation.
Ideally,
[Equation]
a condition we referred to as the critical criterion .
Under this condition, the system preserves the long-term dependencies of the inputs while being stable.
Stability and Trainability.
Here we connect the stability of the ODE to the trainability of the RNN produced by discretization.
Inherently, the stability analysis studies the sensitivity of a solution, i.e., how much a solution of the ODE would change w.r.t.
changes in the initial condition.
Differentiating
Equation 1 with respect to the initial state h (0) on both sides, we have the following sensitivity analysis (with chain rules):
[Equation]
For notation simplicity, let us deﬁne A ( t )
=
∂ h ( t ) /∂
h (0) , then we have
[Equation]
Note that this is a linear ODE with solution A ( t
[Equation] = e J [Equation]
J ) t P − 1 , assuming the Jacobian
J does not vary or vary slowly over time (We will later show this is a valid assumption).
Here Λ ( J ) denotes the eigenvalues of J , and the columns of P are the corresponding eigenvectors.
See
Appendix A for a more detailed derivation.
In the language of RNNs, A ( t ) is the Jacobian of a hidden state h t with respect to initial hidden state h 0 .
When the critical criterion is met, i.e. ,
Re ( Λ ( J ))
≈ 0 , the magnitude of A ( t ) is approximately constant in time, thus no exploding or vanishing gradient problems.
With the connection established, we next design ODEs that satisfy the critical criterion.
An antisymmetric matrix is a square matrix whose transpose equals its negative; i.e. , a matrix M ∈ R n × n is antisymmetric if M T = − M .
An interesting property of an antisymmetric matrix M is that, the eigenvalues of M are all imaginary:
[Equation]
making antisymmetric matrices a suitable building block of a stable recurrent architecture.
Consider the following ODE
[Equation]
[Equation]
where h ( t ) ∈
R n , x ( t ) ∈ R m , W h ∈ R n × n , V h ∈ R n × m and b h ∈ R n .
Note that W h − W Th is an antisymmetric matrix.
The Jacobian matrix of the right hand side is
[Equation]
whose eigenvalues are all imaginary, i.e. , Re ( λ i ( J ( t )))
= 0 , ∀ i = 1 , 2 , . . . , n .
In other words, it satisﬁes the critical criterion in Equation 5.
See Appendix B for a proof.
The entries of the diagonal matrix in Equation 9 are the derivatives of the activation function, which are bounded in [0 , 1] for sigmoid and hyperbolic tangent.
In other words, the Jacobian matrix J ( t ) changes smoothly over time.
Furthermore, since the input and bias term only affect the bounded diagonal matrix, their effect on the stability of the ODE is insigniﬁcant compared with the antisymmetric matrix.
A naive forward Euler discretization of the ODE in Equation 8 leads to the following recurrent network we refer to as the AntisymmetricRNN . [Equation]
[Equation]
where h t ∈ R n is the hidden state at time t ; x t ∈ R m is the input at time t ; W h ∈ R n × n ,
V h ∈ R n × m and b h ∈ R n are the parameters of the network; (cid:15) > 0 is a hyperparameter that represents the step size.
Note that the antisymmetric matrix W h − W Th only has n ( n − 1) / 2 degrees of freedom.
When implementing the model, W h can be parameterized as a strictly upper triangular matrix, i.e. , an upper triangular matrix of which the diagonal entries are all zero.
This makes the proposed model more parameter efﬁcient than an unstructured RNN model of the same size of hidden states.
3. S TABILITY OF FORWARD E ULER METHOD : D IFFUSION
Given a stable ODE, its forward Euler discretization can still be unstable, as illustrated in Section 4.
The stability condition of the forward Euler method has been well studied and summarized in the following proposition.
Proposition 2.
(Stability of forward Euler method)
The forward propagation in Equation 10 is stable if
[Equation]
(11) where | · | denote the absolute value or modulus of a complex number and J t is the Jacobian matrix evaluated at h t .
See Ascher & Petzold (1998) for a proof.
The ODE as deﬁned in Equation 8 is however incompatible with the stability condition of the forward Euler method.
Since λ i ( J t ) , the eigenvalues of the
Jacobian matrix, are all imaginary, | 1 + (cid:15)λ i ( J t ) | is always great than 1, which makes the AntisymmetricRNN deﬁned in Equation 10 unstable when solved using forward Euler.
One easy way to ﬁx it is to add diffusion to the system by subtracting a small number γ > 0 from the diagonal elements of the transition matrix.
The model thus becomes
[Equation]
where I is the identity matrix of size n and γ > 0 is a hyperparameter that controls the strength of diffusion.
By doing so, the eigenvalues of the Jacobian have slightly negative real parts.
This modiﬁcation improves the stability of the numerical method as demonstrated in Section 4.
3. G ATING MECHANISM
Gating is commonly employed in RNNs.
Each gate is often modeled as a single layer network taking the previous hidden state h t − 1 and data x t as inputs, followed by a sigmoid activation.
As an example, LSTM cells make use of three gates, a forget gate, an input gate, and an output gate.
A systematic ablation study suggests that some of the gate are crucial to the performance of
LSTM (Jozefowicz et al., 2015).
Gating can be incorporated into AntisymmetricRNN as well.
However, it should be done carefully so that the critical condition in Equation 5 still holds.
We propose the following modiﬁcation to
AntisymmetricRNN, which adds an input gate z t to control the ﬂow of information into the hidden states:
[Equation]
(13) where σ denotes the sigmoid function and ◦ denotes the Hadamard product.
The effect of z t resembles the input gate in LSTM and the update gate in GRU.
By sharing the antisymmetric weight matrix, the number of model parameters only increases slightly, instead of being doubled.
More importantly, the Jacobian matrix of this gated model has a similar form as that in Equation 9, that is, a diagonal matrix multiplied by an antisymmetric matrix (ignoring diffusion).
As a result, the real part of the eigenvalues of the Jacobian matrix are still close to zero, and the critical criterion remains satisﬁed. [Figure]
S IMULATION
Adopting the visualization technique used by Laurent & von Brecht (2017) and Haber & Ruthotto (2017) [Figure] we study the behavior of two-dimensional vanilla RNNs (left) and RNNs right
[Equation]
Here h t ∈ R 2 and 1 ≤ t ≤ T .
We arbitrarily choose three initial states: (0 , 0 . 5) , ( − 0 .
5 , − 0 .
and (0 . 5 , − 0 .
and apply the corresponding RNN recurrence.
Figure 1 plots the progression of the states h t of vanilla RNNs (ﬁrst row) and RNNs with feedback (second row) parameterized by different transition matrices W .
In each ﬁgure, more translucent points represent earlier time steps.
The number of total time steps T = 50 .
Figure 1(a) corresponds to a random weight matrix, where the entries are independent and standard
Gaussian.
The three initial points (shown in stars) converge to two ﬁxed points on the boundary.
Since the weight matrix is unstructured, the behavior is unpredictable as expected.
More examples with different weight matrices are shown in Appendix D. Figure 1(b) shows the behavior of the identity weight matrix.
Since tanh( · ) is a contractive mapping, the origin is the unique ﬁxed point.
Figure 1(c) and (d) correspond to orthogonal weight matrices that represent reﬂection and rotation transforms respectively.
A two-dimensional orthogonal matrix is either a reﬂection or a rotation transform; the determinant of the matrix is 1 or − 1 respectively.
Even though the weight matrix is orthogonal, the states all converge to the origin because of the derivative of the activation function 0 ≤ tanh (cid:48)
( W h t − 1 ) ≤ 1 .
In the case of RNNs with feedback, the trajectory of the hidden states is predictable based on the eigenvalues of the weight matrix W .We consider the following four weight matrices that correspond to Figure 1(e)-(f) respectively:
[Equation]
[-(cid:19) , W − = (cid:18) −  −(cid:19) , W 0 = (cid:18) 
-]
0 −  (cid:19) , W diﬀ = (cid:18) − 0 .− − 0 .(cid:19) .
We also overlay the vector ﬁeld that represent the underlying ODE.
The step size is set to (cid:15) = 0 .
For Figure 1(e) and (f), the eigenvalues are λ 1 ( W + ) =
λ 2 ( W + ) = 2 and λ 1 ( W − ) =
λ 2 ( W − ) = − 2 .
As a result, the hidden states are moving away from the origin and towards the origin re- spectively.
Figure 1(g) corresponds to the antisymmetric weight matrix W 0 , whose eigenvalues are purely imaginary: λ 1 ( W 0 )
=
2 i , λ 2 ( W 0 ) =
− 2 i .
In this case the vector ﬁeld is circular; a state moves around the origin without exponentially increasing or decreasing its norm.
However, on closer inspection, the trajectories are actually outward spirals.
This is because at each time step, the state moves along the tangential direction by a small step, which increases the distance from the origin and leads to numerical instability.
It is the behavior characterized by Proposition 2.
This issue can be mitigated by subtracting a small diffusion term γ from the diagonal elements of the weight matrix.
We choose γ = 0 .
15 and the weight matrix W diﬀ has eigenvalues of λ 1 ( W diﬀ )
= − 0 .
15 +
i , λ 2 ( W diﬀ ) = − 0 .
15 − 2 i .
Figure 1(h) shows the effect of the diffusion terms.
The vector ﬁeld is slightly tilting toward the origin and the trajectory maintains a constant distance from the origin.
These simulations show that the hidden states of an AntisymmetricRNN (Figure 1(g) and (h)) have predictable dynamics.
It achieves the desirable behavior, without the complication of maintaining an orthogonal or unitary matrix as in Figure 1(d), which still suffers from vanishing gradients due to the contraction of the activation function.
E XPERIMENTS
The performance of the proposed antisymmetric networks is evaluated on four image classiﬁcation tasks with long-range dependencies.
The classiﬁcation is done by feeding pixels of the images as a sequence to RNNs and sending the last hidden state h T of the RNNs into a full-connected layer and a softmax function.
We use the cross-entropy loss and SGD with momentum and Adagrad (Duchi et al., 2011) as optimizers.
More experimental details can be found in Appendix C.
In this section, AntisymmetricRNN denotes the model with diffusion in Equation 12, and AntisymmetricRNN w/ gating represents the model in Equation 13.
5. P IXEL - BY - PIXEL MNIST
In the ﬁrst task [Table] we learn to classify the MNIST digits by pixels (LeCun et al., 1998).
This task was proposed by Le et al.
and used as a benchmark for learning long term dependencies.
MNIST images are grayscale with 28 × 28 pixels.
The 784 pixels are presented sequentially to the recurrent net, one pixel at a time in scanline order (starting at the top left corner of the image and ending at the bottom right corner).
In other words, the input dimension m = 1 and number of time steps
T = 784 .
The pixel-by-pixel MNIST task is to predict the digit of the MNIST image after seeing all 784 pixels.
As a result, the network has to be able to learn the long-range dependencies in order to correctly classify the digit.
To make the task even harder, the MNIST pixels are shufﬂed using a ﬁxed random permutation.
It creates non-local long-range dependencies among pixels in an image.
This task is referred to as the permuted pixel-by-pixel MNIST
[Table] Table 1 summarizes the performance of our methods and the existing methods.
On both tasks, the proposed AntisymmetricRNNs outperform the regular LSTM model using only 1 / 7 of parameters.
Imposing orthogonal weights (Arjovsky et al., 2016) produces worse results, which corroborates with existing study showing that such constraints restrict the capacity of the learned model.
Softening the orthogonal weights constraints (Wisdom et al., 2016; Vorontsov et al., 2017; Jose et al., 2017) leads to slightly improved performance.
AntisymmetricRNNs outperform these methods by a large margin, without the computational overhead to enforce orthogonality.
5. P IXEL - BY - PIXEL CIFAR-
To test our methods on a larger dataset, we conduct experiments on pixel-by-pixel CIFAR-10 .
The
CIFAR-10 dataset contains 32 × 32 colour images in 10 classes (Krizhevsky & Hinton, 2009).
Similar to pixel-by-pixel MNIST, we feed the three channels of a pixel into the model at each time step.
The input dimension m = 3 and number of time steps T = 1024 .
The results are shown in Table 2.
The AntisymmetricRNN performance is on par with the LSTM, and AntisymmetricRNN with gating is slightly better than LSTM, both using only about half of the parameters of the LSTM model.
Further investigation shows that the task is mostly dominated by short term dependencies.
LSTM can achieve about 48.3% classiﬁcation accuracy by only seeing the last 8 rows of CIFAR-10 images 2 . [Table]
[Table]
padded CIFAR-10.
To introduce more long-range dependencies to the pixelby-pixel CIFAR-10 task, we deﬁne a more challenging task call the noise padded CIFAR-10 , inspired by the noise padded experiments in Chen et al.
(2018a).
Instead of feeding in one pixel at one time, we input each row of a CIFAR-10 image at every time step.
After the ﬁrst 32 time steps, we input independent standard Gaussian noise for the remaining time steps.
Since a CIFARimage is of size 32 with three RGB channels, the input dimension is m = 96 .
The total time steps is set to T = 1000 .
In other words, only the ﬁrst 32 time steps of input contain salient information, all remaining 968 time steps are merely random noise.
For a model to correctly classify an input image, it has to remember the information from long time ago.
This task is conceptually more difﬁcult than the pixel-by-pixel CIFAR-10, although the total amount of signal in the input sequence is the same.
The results are shown in Table 2.
LSTM fail to train at all on this task while our proposed methods perform reasonably well with fewer parameters.
To verify that AntisymmetricRNNs indeed mitigate the exploding/vanishing gradient issues, we conduct an additional set of experiments varying the length of noise padding so that the total time steps T ∈ { 100 , 200 , 400 , 800 } .
LSTMs and AntisymmetricRNNs with different diffusion constants γ ∈ { 0 . 001 , 0 . 01 , 0 .
1 , 1 , 5 , 10 } are trained on these tasks.
Figure 2 visualizes the mean and standard deviation of the eigenvalues of the end-to-end Jacobian matrices for these networks.
Unitary eigenvalues, i.e., mean close to 1 and standard deviation close to 0, indicate non-exploding and non-vanishing gradients. As shown in the ﬁgure, the eigenvalues for LSTMs quickly approaches [Table] zero as time steps increase, indicating vanishing gradients during back-propagation.
This explains why LSTMs fail to train at all on this task.
AntisymmetricRNNs with a broad range of diffusion constants γ , on the other hand have eigenvalues centered around 1.
It is worth noting though as the diffusion constant increases to large values, AntisymmetricRNNs run into vanishing gradients as well.
The diffusion constant γ plays an important role in striking a balance between the stability of discretization and capturing long-term dependencies.
For the CIFAR-10 experiments, we have also conducted an ablation study to further demonstrate the effect of antisymmetric weight matrix.
The ablation model in Table 2 refers to the model that replaces the antisymmetric weight matrices in Equation 13 with unstructured weight matrices.
As shown in Table 2, without the antisymmetric parametrization, the performance on both pixel-bypixel and noise padded CIFAR-10 is worse.
It is worth mentioning that the antisymmetric formulation is a sufﬁcient condition of stability, not necessary.
There are possibly other conditions that leads to stability as well as suggested in Chen et al.
(2018a), which could explain the modest degradation in the ablation results.
C ONCLUSION
In this paper, we present a new perspective on the trainability of RNNs from the dynamical system viewpoint.
We draw connections between RNNs and the ordinary differential equation theory, and design new recurrent architectures by discretizing ODEs.
This new view opens up possibilities to exploit the computational and theoretical success from dynamical systems to understand and improve the trainability of RNNs.
We also propose the AntisymmetricRNN, which is a discretization of ODEs that satisfy the critical criterion.
Besides its appealing theoretical properties, our models have demonstrated competitive performance over strong recurrent baselines on a comprehensive set of benchmark tasks.
By establishing a link between recurrent networks and ordinary differential equations, we anticipate that this work will inspire future research in both communities.
An important item of future work is to investigate other stable ODEs and numerical methods that lead to novel and well-conditioned recurrent architectures. R
A A N OVERVIEW OF STABILITY THEORY
In this section, we provide a brief overview of stability theory by examples.
Most of the materials are adapted from Ascher & Petzold (1998).
Consider the simple scalar ODE, often referred to as the test equation : y (cid:48)
( t ) = λy ( t ) , where λ ∈ C is a constant.
We allow λ to be complex because it later represents an eigenvalue of a system’s matrix.
The solution to this initial value problem is y ( t ) =
e λt y (0) .
If y ( t ) and ˜ y ( t ) are two solutions of the test equation, then their difference at any time t is
[Equation]
If Re ( λ ) > 0 , then a small perturbation of the initial states would cause an exponentially exploding difference.
If Re ( λ ) < 0 , the system is stable, but the perturbation decays exponentially.
The perturbation is preserved in the system only if Re ( λ ) = 0 .
We now consider the extension of the test equation to a matrix ODE y (cid:48) ( t ) =
Ay ( t ) .
The solution is y ( t ) =
e A t y (0) .
To simplify the analysis, we assume A is diagonalizable, i.e. , P − 1 AP = Λ , where Λ is a diagonal matrix of the eigenvalues of A and the columns of P are the corresponding eigenvectors.
If we deﬁne w ( t ) =
P − 1 y ( t ) , then w (cid:48) ( t ) =
Λ w ( t ) .
The system for w ( t ) is decoupled: for each component w i ( t ) of w ( t ) , we have a test equation w (cid:48)
i ( t ) =
λ i
w i ( t ) .
Therefore, the stability for w ( t ) , hence also for y ( t ) , is determined by the eigenvalues λ i .
B P ROOF OF A PROPOSITION
In this section, we provide a proof of a proposition which implies that the AntisymmetricRNN and
AntisymmetricRNN with gating satisfy the critical criterion, i.e. , the eigenvalues of the Jacobian matrix are imaginary.
The proof is adapted from Chang et al.
(2018a).
Proposition 3.
If W ∈ R n × n is an antisymmetric matrix and D ∈ R n × n is an invertible diagonal matrix, then the eigenvalues of DW are imaginary.
[Equation]
Proof.
Let λ and v be a pair of eigenvalue and eigenvector of DW , then
[Equation]
On one hand, v ∗ D − 1 v is real.
On the other hand,
[Equation]
where ∗ represents conjugate transpose.
It implies that v ∗ W v is imaginary.
Therefore, λ has to be imaginary.
As a result, all eigenvalues of DW are imaginary.
C E XPERIMENTAL DETAILS
Let m be the input dimension and n be the number of hidden units.
The input to hidden matrices are initialized to N (0 , 1 /m ) .
The hidden to hidden matrices are initialized to N (0 , σ 2 w /n ) , where σ w is chosen from σ w ∈ { 0 , 1 , 2 , 4 , 8 , 16 } .
The bias terms are initialized to zero, except the forget gate bias of LSTM is initialized to 1, as suggested by Jozefowicz et al.
For AntisymmetricRNNs, the step size (cid:15) ∈ { 0 . 01 , 0 . 1 , 1 } and diffusion γ ∈ { 0 . 001 , 0 . 01 , 0 . 1 , 1 . 0 } .
We use SGD with momentum and Adagrad (Duchi et al., 2011) as optimizers, with batch size of 128 and learning rate chosen from { 0 . 1 , 0 . 2 , 0 . 3 , 0 . 4 , 0 . 5 , 0 .
On MNIST and pixel-by-pixel CIFAR-10, all the models are trained for 50,000 iterations.
On noise padded CIFAR-10, models are trained 10,000 iterations.
We use the standard train/test split of MNIST and CIFAR-10.
The performance measure is the classiﬁcation accuracy evaluated on the test set.
D A DDITIONAL VISUALIZATIONS
In this section [Figure] we present additional visualizations that are related to the simulation study in Section 4.
Figure 3 and 4 show the dynamics of vanilla RNNs and AntisymmetricRNNs with standard
Gaussian random weights using different seeds.
These visualizations further illustrate the random behavior of a vanilla RNN and the predictable dynamics of an AntisymmetricRNN.
Figure 5 shows the the dynamics of AntisymmetricRNNs with independent standard Gaussian input.
This shows that the dynamics become noisier compared to Figure 1, but the trend remains the same
[Figure]
[Figure] [Figure]