Simple RNN

The simple RNN architecture with just a single layer of neurons that receive the input $\mathbf{x}$ is shown below. Simple RNN layer with $n_{neurons}$ RNN neurones mapping the input to the hidden-state at each time step. A more practical simple RNN architecture is shown below. Simple RNN with recurrences between hidden units. This architecture can compute any computable function and therefore is a Universal Turing Machine. Notice how the path from input $\mathbf x_{t-1}$ affects the label $\mathbf y_{t}$ and also the conditional independence between $\mathbf y$ given $\mathbf x$.

​

Dimensioning Simple RNNs

In the table below $m$ is the number of examples in the mini-batch. Please note that there may be multiple layers that can be stacked on top of each other and they can individually keep a hidden state.

​

Forward Propagation

This network maps the input sequence to a sequence of the same length and implements the following forward pass: $\mathbf a_t = \mathbf W \mathbf h _{t-1} + \mathbf U \mathbf x_t + \mathbf b$ $\mathbf h_t = \tanh(\mathbf a_t)$ $\mathbf o_t = \mathbf V \mathbf h_t + \mathbf c$ $\mathbf{\hat y_t} = \mathtt{softmax}(\mathbf o_t)$ $L(\hat{\mathbf y}_1, \dots , \hat{\mathbf y}_{\tau}, \mathbf y_1, \dots , \mathbf y_{\tau}) = D_{KL}[\hat p_{data}(\mathbf y | \mathbf x) || p_{model}(\mathbf y | \mathbf x; \mathbf \theta)]$ $= - \mathbb E_{\mathbf y | \mathbf x \sim \hat{p}_{data}} \log p_{model}(\mathbf y | \mathbf x ; \mathbf \theta) = - \sum_t \log p_{model}(y_t | \mathbf x_1, \dots, \mathbf x_t ; \mathbf \theta)$ Notice that RNNs can model very generic distributions $\log p_{model}(\mathbf x, \mathbf y ; \mathbf \theta)$. The simple RNN architecture above, effectively models the posterior distribution $\log p_{model}(\mathbf y | \mathbf x ; \mathbf \theta)$ and based on a conditional independence assumption it factorizes into $\sum_t \log p_{model}(y_t | \mathbf x_1, \dots, \mathbf x_t ; \mathbf \theta)$. In many instances we have problems where it only matters the label $y_\tau$ at the end of the sequence. Lets say that you are classifying speech or video inside the cabin of a car to detect the psychological state of the driver. The same architecture shown above can also represent such problems - the only difference is the only the $\mathbf o_\tau$, $L_\tau$ and $y_\tau$ will be considered. Lets see an example to understand better the forward propagation equations. Example sentence as input to the RNN In the figure above you have a hypothetical document (a sentence) that is broken into tokens - lets say that a token is a word in this case. In the simpler case where we need a classification of the whole document, given that $\tau=6$, we are going to receive at t=1, the first token $\mathbf x_1$ and with an input hidden state $\mathbf h_0 = 0$ we will calculate the forward equations for $\mathbf h_1$, ignoring the output $\mathbf o_1$ and repeat the unrolling when the next input $\mathbf x_2$ comes in until we reach the end of sentence token $\mathbf x_6$ which in this case will calculate the output $y_6$ and loss $- \log p_{model} (y_6|\mathbf x_1, \dots , \mathbf x_6; \mathbf w)$ where $\mathbf \theta = { \mathbf W, \mathbf U, \mathbf V, \mathbf b, \mathbf c }$.

​

Back-Propagation Through Time (BPTT)

Lets now see how the training through backward propagation would work for RNNs. Understanding RNN memory through BPTT procedure Backpropagation is similar to that of other neural networks simply because the unrolled architecture resembles a feed forward arrangement. But there is an important difference and we can see it using the above computational graph for the unrolled recurrences $t$ and $t-1$. During computation of the variable $\mathbf h_t$ we use the value of the variable $\mathbf h_{t-1}$ calculated in the previous recurrence. So when we apply the chain rule in the backward phase of BP, for all nodes that involve such variables with recurrent dependencies, the end result is that non local gradients appear from previous backpropagation steps ($t$ in the figure). This is in contrast to other networks where during BP only local to each gate gradients where involved as we have seen earlier. This is effectively why we say that simple RNNs feature memory. The key point to notice in the backpropagation in recurrence $t-1$ is the junction between $\tanh$ and $\mathbf V \mathbf h_{t-1}$. This junction brings in the gradient $\nabla_{\mathbf h_{t-1}}L_t$ from the backpropagation of the $\mathbf W h_{t-1}$ node in recurrence $t$ and just because its a junction, it is added to the backpropagated gradient from above in the current recurrence $t-1$ i.e. $\nabla_{\mathbf h_{t-1}}L_{t-1} \leftarrow \nabla_{\mathbf h_{t-1}}L_{t-1} + \nabla_{\mathbf h_{t-1}}L_t$ Ian Goodfellow’s section 10.2.2 provides the exact equations - please note that you need to know only the intuition behind computational graphs for RNNs. In practice, BPTT is [truncated](https://proceedings.mlr.press/v115/aicher20a.html#:~:text=Truncated%20backpropagation%20through%20time%20(TBPTT,a%20fixed%20number%20of%20lags.) to avoid having to do one full forward pass and one full reverse pass through the training dataset of a e.g. language model that is usually very large, to do a single gradient update. When the truncation level is not sufficiently large, the bias introduced can cause SGD to not converge. In practice, a large truncation size is chosen heuristically (e.g. larger than the expected ‘memory’ of the system) or via cross-validation.

​

Vanishing or exploding gradients

In a similar fashion, the RNN hidden state recurrence, in the backwards pass that extends from the $t=\tau$ to $t=1$ can make the gradient, when $\tau$ is large, either vanish or explode. Instead of a scalar $w$ we saw in the IIR filter we have matrices $\mathbf W$ and instead of $h$ we have gradients $\nabla_{\mathbf h_{t}}L_{t}$. See this paper for details. The gradient of the $\tanh$ non-linearity shown below, is between 0 and 1 suppressing the gradients and slowing down training. Derivative of $\tanh$ non-linearity Similar the successive application of the $W$ matrix is causing explosive gradients as simplistically (ignoring the non-linearity) the hidden state can be written as: $\mathbf h_{t} = \mathbf W \mathbf h_{t-1}$ making after $\tau$ steps $\mathbf h_{t} = \mathbf W^\tau \mathbf h_{0}$ If the magnitude of the eigenvalues are less than 1.0 the matrix will create vanishing gradients as it is involved in the $\nabla_{\mathbf h_{t}}L_{t}$ expression (see equations in section 10.2.2). This issue is addressed using an evolved RNN architecture called Long Short Term Memory (LSTM).

​

PyTorch reference

Key references: (Graves et al., n.d.; Jaderberg et al., 2016; Parisotto et al., 2016)

​

References

• Graves, A., Wayne, G., Danihelka, I. (n.d.). Neural Turing Machines.
• Jaderberg, M., Czarnecki, W., Osindero, S., Vinyals, O., Graves, A., et al. (2016). Decoupled Neural Interfaces using Synthetic Gradients. arXiv [cs.LG].
• Parisotto, E., Mohamed, A., Singh, R., Li, L., Zhou, D., et al. (2016). Neuro-Symbolic Program Synthesis. arXiv [cs.AI].

---