5.1: The vanishing gradient problem
- Page ID
- 3766
\( \newcommand{\vecs}[1]{\overset { \scriptstyle \rightharpoonup} {\mathbf{#1}} } \)
\( \newcommand{\vecd}[1]{\overset{-\!-\!\rightharpoonup}{\vphantom{a}\smash {#1}}} \)
\( \newcommand{\id}{\mathrm{id}}\) \( \newcommand{\Span}{\mathrm{span}}\)
( \newcommand{\kernel}{\mathrm{null}\,}\) \( \newcommand{\range}{\mathrm{range}\,}\)
\( \newcommand{\RealPart}{\mathrm{Re}}\) \( \newcommand{\ImaginaryPart}{\mathrm{Im}}\)
\( \newcommand{\Argument}{\mathrm{Arg}}\) \( \newcommand{\norm}[1]{\| #1 \|}\)
\( \newcommand{\inner}[2]{\langle #1, #2 \rangle}\)
\( \newcommand{\Span}{\mathrm{span}}\)
\( \newcommand{\id}{\mathrm{id}}\)
\( \newcommand{\Span}{\mathrm{span}}\)
\( \newcommand{\kernel}{\mathrm{null}\,}\)
\( \newcommand{\range}{\mathrm{range}\,}\)
\( \newcommand{\RealPart}{\mathrm{Re}}\)
\( \newcommand{\ImaginaryPart}{\mathrm{Im}}\)
\( \newcommand{\Argument}{\mathrm{Arg}}\)
\( \newcommand{\norm}[1]{\| #1 \|}\)
\( \newcommand{\inner}[2]{\langle #1, #2 \rangle}\)
\( \newcommand{\Span}{\mathrm{span}}\) \( \newcommand{\AA}{\unicode[.8,0]{x212B}}\)
\( \newcommand{\vectorA}[1]{\vec{#1}} % arrow\)
\( \newcommand{\vectorAt}[1]{\vec{\text{#1}}} % arrow\)
\( \newcommand{\vectorB}[1]{\overset { \scriptstyle \rightharpoonup} {\mathbf{#1}} } \)
\( \newcommand{\vectorC}[1]{\textbf{#1}} \)
\( \newcommand{\vectorD}[1]{\overrightarrow{#1}} \)
\( \newcommand{\vectorDt}[1]{\overrightarrow{\text{#1}}} \)
\( \newcommand{\vectE}[1]{\overset{-\!-\!\rightharpoonup}{\vphantom{a}\smash{\mathbf {#1}}}} \)
\( \newcommand{\vecs}[1]{\overset { \scriptstyle \rightharpoonup} {\mathbf{#1}} } \)
\( \newcommand{\vecd}[1]{\overset{-\!-\!\rightharpoonup}{\vphantom{a}\smash {#1}}} \)
\(\newcommand{\avec}{\mathbf a}\) \(\newcommand{\bvec}{\mathbf b}\) \(\newcommand{\cvec}{\mathbf c}\) \(\newcommand{\dvec}{\mathbf d}\) \(\newcommand{\dtil}{\widetilde{\mathbf d}}\) \(\newcommand{\evec}{\mathbf e}\) \(\newcommand{\fvec}{\mathbf f}\) \(\newcommand{\nvec}{\mathbf n}\) \(\newcommand{\pvec}{\mathbf p}\) \(\newcommand{\qvec}{\mathbf q}\) \(\newcommand{\svec}{\mathbf s}\) \(\newcommand{\tvec}{\mathbf t}\) \(\newcommand{\uvec}{\mathbf u}\) \(\newcommand{\vvec}{\mathbf v}\) \(\newcommand{\wvec}{\mathbf w}\) \(\newcommand{\xvec}{\mathbf x}\) \(\newcommand{\yvec}{\mathbf y}\) \(\newcommand{\zvec}{\mathbf z}\) \(\newcommand{\rvec}{\mathbf r}\) \(\newcommand{\mvec}{\mathbf m}\) \(\newcommand{\zerovec}{\mathbf 0}\) \(\newcommand{\onevec}{\mathbf 1}\) \(\newcommand{\real}{\mathbb R}\) \(\newcommand{\twovec}[2]{\left[\begin{array}{r}#1 \\ #2 \end{array}\right]}\) \(\newcommand{\ctwovec}[2]{\left[\begin{array}{c}#1 \\ #2 \end{array}\right]}\) \(\newcommand{\threevec}[3]{\left[\begin{array}{r}#1 \\ #2 \\ #3 \end{array}\right]}\) \(\newcommand{\cthreevec}[3]{\left[\begin{array}{c}#1 \\ #2 \\ #3 \end{array}\right]}\) \(\newcommand{\fourvec}[4]{\left[\begin{array}{r}#1 \\ #2 \\ #3 \\ #4 \end{array}\right]}\) \(\newcommand{\cfourvec}[4]{\left[\begin{array}{c}#1 \\ #2 \\ #3 \\ #4 \end{array}\right]}\) \(\newcommand{\fivevec}[5]{\left[\begin{array}{r}#1 \\ #2 \\ #3 \\ #4 \\ #5 \\ \end{array}\right]}\) \(\newcommand{\cfivevec}[5]{\left[\begin{array}{c}#1 \\ #2 \\ #3 \\ #4 \\ #5 \\ \end{array}\right]}\) \(\newcommand{\mattwo}[4]{\left[\begin{array}{rr}#1 \amp #2 \\ #3 \amp #4 \\ \end{array}\right]}\) \(\newcommand{\laspan}[1]{\text{Span}\{#1\}}\) \(\newcommand{\bcal}{\cal B}\) \(\newcommand{\ccal}{\cal C}\) \(\newcommand{\scal}{\cal S}\) \(\newcommand{\wcal}{\cal W}\) \(\newcommand{\ecal}{\cal E}\) \(\newcommand{\coords}[2]{\left\{#1\right\}_{#2}}\) \(\newcommand{\gray}[1]{\color{gray}{#1}}\) \(\newcommand{\lgray}[1]{\color{lightgray}{#1}}\) \(\newcommand{\rank}{\operatorname{rank}}\) \(\newcommand{\row}{\text{Row}}\) \(\newcommand{\col}{\text{Col}}\) \(\renewcommand{\row}{\text{Row}}\) \(\newcommand{\nul}{\text{Nul}}\) \(\newcommand{\var}{\text{Var}}\) \(\newcommand{\corr}{\text{corr}}\) \(\newcommand{\len}[1]{\left|#1\right|}\) \(\newcommand{\bbar}{\overline{\bvec}}\) \(\newcommand{\bhat}{\widehat{\bvec}}\) \(\newcommand{\bperp}{\bvec^\perp}\) \(\newcommand{\xhat}{\widehat{\xvec}}\) \(\newcommand{\vhat}{\widehat{\vvec}}\) \(\newcommand{\uhat}{\widehat{\uvec}}\) \(\newcommand{\what}{\widehat{\wvec}}\) \(\newcommand{\Sighat}{\widehat{\Sigma}}\) \(\newcommand{\lt}{<}\) \(\newcommand{\gt}{>}\) \(\newcommand{\amp}{&}\) \(\definecolor{fillinmathshade}{gray}{0.9}\)Imagine you're an engineer who has been asked to design a computer from scratch. One day you're working away in your office, designing logical circuits, setting out AND
gates, OR
gates, and so on, when your boss walks in with bad news. The customer has just added a surprising design requirement: the circuit for the entire computer must be just two layers deep:
You're dumbfounded, and tell your boss: "The customer is crazy!"
Your boss replies: "I think they're crazy, too. But what the customer wants, they get."
In fact, there's a limited sense in which the customer isn't crazy. Suppose you're allowed to use a special logical gate which lets youAND
together as many inputs as you want. And you're also allowed a many-input NAND
gate, that is, a gate which can AND
multiple inputs and then negate the output. With these special gates it turns out to be possible to compute any function at all using a circuit that's just two layers deep.
But just because something is possible doesn't make it a good idea. In practice, when solving circuit design problems (or most any kind of algorithmic problem), we usually start by figuring out how to solve sub-problems, and then gradually integrate the solutions. In other words, we build up to a solution through multiple layers of abstraction.
For instance, suppose we're designing a logical circuit to multiply two numbers. Chances are we want to build it up out of sub-circuits doing operations like adding two numbers. The sub-circuits for adding two numbers will, in turn, be built up out of sub-sub-circuits for adding two bits. Very roughly speaking our circuit will look like:
That is, our final circuit contains at least three layers of circuit elements. In fact, it'll probably contain more than three layers, as we break the sub-tasks down into smaller units than I've described. But you get the general idea.
So deep circuits make the process of design easier. But they're not just helpful for design. There are, in fact, mathematical proofs showing that for some functions very shallow circuits require exponentially more circuit elements to compute than do deep circuits. For instance, a famous series of papers in the early 1980s*
*The history is somewhat complex, so I won't give detailed references. See Johan Håstad's 2012 paper On the correlation of parity and small-depth circuits for an account of the early history and references.showed that computing the parity of a set of bits requires exponentially many gates, if done with a shallow circuit. On the other hand, if you use deeper circuits it's easy to compute the parity using a small circuit: you just compute the parity of pairs of bits, then use those results to compute the parity of pairs of pairs of bits, and so on, building up quickly to the overall parity. Deep circuits thus can be intrinsically much more powerful than shallow circuits.
Up to now, this book has approached neural networks like the crazy customer. Almost all the networks we've worked with have just a single hidden layer of neurons (plus the input and output layers):
These simple networks have been remarkably useful: in earlier chapters we used networks like this to classify handwritten digits with better than 98 percent accuracy! Nonetheless, intuitively we'd expect networks with many more hidden layers to be more powerful:
Such networks could use the intermediate layers to build up multiple layers of abstraction, just as we do in Boolean circuits. For instance, if we're doing visual pattern recognition, then the neurons in the first layer might learn to recognize edges, the neurons in the second layer could learn to recognize more complex shapes, say triangle or rectangles, built up from edges. The third layer would then recognize still more complex shapes. And so on. These multiple layers of abstraction seem likely to give deep networks a compelling advantage in learning to solve complex pattern recognition problems. Moreover, just as in the case of circuits, there are theoretical results suggesting that deep networks are intrinsically more powerful than shallow networks**For certain problems and network architectures this is proved in On the number of response regions of deep feed forward networks with piece-wise linear activations, by Razvan Pascanu, Guido Montúfar, and Yoshua Bengio (2014). See also the more informal discussion in section 2 of Learning deep architectures for AI, by Yoshua Bengio (2009)..
How can we train such deep networks? In this chapter, we'll try training deep networks using our workhorse learning algorithm -stochastic gradient descent by backpropagation. But we'll run into trouble, with our deep networks not performing much (if at all) better than shallow networks.
That failure seems surprising in the light of the discussion above. Rather than give up on deep networks, we'll dig down and try to understand what's making our deep networks hard to train. When we look closely, we'll discover that the different layers in our deep network are learning at vastly different speeds. In particular, when later layers in the network are learning well, early layers often get stuck during training, learning almost nothing at all. This stuckness isn't simply due to bad luck. Rather, we'll discover there are fundamental reasons the learning slowdown occurs, connected to our use of gradient-based learning techniques.
As we delve into the problem more deeply, we'll learn that the opposite phenomenon can also occur: the early layers may be learning well, but later layers can become stuck. In fact, we'll find that there's an intrinsic instability associated to learning by gradient descent in deep, many-layer neural networks. This instability tends to result in either the early or the later layers getting stuck during training.
This all sounds like bad news. But by delving into these difficulties, we can begin to gain insight into what's required to train deep networks effectively. And so these investigations are good preparation for the next chapter, where we'll use deep learning to attack image recognition problems.
The vanishing gradient problem
So, what goes wrong when we try to train a deep network?
To answer that question, let's first revisit the case of a network with just a single hidden layer. As per usual, we'll use the MNIST digit classification problem as our playground for learning and experimentation*
*I introduced the MNIST problem and data here and here.
If you wish, you can follow along by training networks on your computer. It is also, of course, fine to just read along. If you do wish to follow live, then you'll need Python 2.7, Numpy, and a copy of the code, which you can get by cloning the relevant repository from the command line:
git clone https://github.com/mnielsen/neural-networks-and-deep-learning.git
If you don't use git then you can download the data and code here. You'll need to change into the src subdirectory.
Then, from a Python shell we load the MNIST data:
>>> import mnist_loader >>> training_data, validation_data, test_data = \ ... mnist_loader.load_data_wrapper()
We set up our network:
>>> import network2 >>> net = network2.Network([784, 30, 10])
This network has 784 neurons in the input layer, corresponding to the \(28×28=784\) 784 pixels in the input image. We use \(30\) hidden neurons, as well as \(10\) output neurons, corresponding to the \(10\) possible classifications for the MNIST digits \(('0', '1', '2', ……, '9')\).
Let's try training our network for \(30\) complete epochs, using mini-batches of \(10\) training examples at a time, a learning rate \(η=0.1\), and regularization parameter \(λ=5.0\). As we train we'll monitor the classification accuracy on the validation_data*
*Note that the networks is likely to take some minutes to train, depending on the speed of your machine. So if you're running the code you may wish to continue reading and return later, not wait for the code to finish executing.:
>>> net.SGD(training_data, 30, 10, 0.1, lmbda=5.0, ... evaluation_data=validation_data, monitor_evaluation_accuracy=True)
We get a classification accuracy of 96.48 percent (or thereabouts - it'll vary a bit from run to run), comparable to our earlier results with a similar configuration.
Now, let's add another hidden layer, also with 30 neurons in it, and try training with the same hyper-parameters:
>>> net = network2.Network([784, 30, 30, 10]) >>> net.SGD(training_data, 30, 10, 0.1, lmbda=5.0, ... evaluation_data=validation_data, monitor_evaluation_accuracy=True)
This gives an improved classification accuracy, \(96.90\) percent. That's encouraging: a little more depth is helping. Let's add another 30-neuron hidden layer:
>>> net = network2.Network([784, 30, 30, 30, 10]) >>> net.SGD(training_data, 30, 10, 0.1, lmbda=5.0, ... evaluation_data=validation_data, monitor_evaluation_accuracy=True)
That doesn't help at all. In fact, the result drops back down to \(96.57\) percent, close to our original shallow network. And suppose we insert one further hidden layer:
>>> net = network2.Network([784, 30, 30, 30, 30, 10]) >>> net.SGD(training_data, 30, 10, 0.1, lmbda=5.0, ... evaluation_data=validation_data, monitor_evaluation_accuracy=True)
The classification accuracy drops again, to \(96.53\) percent. That's probably not a statistically significant drop, but it's not encouraging, either.
This behaviour seems strange. Intuitively, extra hidden layers ought to make the network able to learn more complex classification functions, and thus do a better job classifying. Certainly, things shouldn't get worse, since the extra layers can, in the worst case, simply do nothing*
*See this later problem to understand how to build a hidden layer that does nothing.. But that's not what's going on.
So what is going on? Let's assume that the extra hidden layers really could help in principle, and the problem is that our learning algorithm isn't finding the right weights and biases. We'd like to figure out what's going wrong in our learning algorithm, and how to do better.
To get some insight into what's going wrong, let's visualize how the network learns. Below, I've plotted part of a \([784,30,30,10]\) network, i.e., a network with two hidden layers, each containing \(30\) hidden neurons. Each neuron in the diagram has a little bar on it, representing how quickly that neuron is changing as the network learns. A big bar means the neuron's weights and bias are changing rapidly, while a small bar means the weights and bias are changing slowly. More precisely, the bars denote the gradient \(∂C/∂b\) for each neuron, i.e., the rate of change of the cost with respect to the neuron's bias. Back in Chapter 2 we saw that this gradient quantity controlled not just how rapidly the bias changes during learning, but also how rapidly the weights input to the neuron change, too. Don't worry if you don't recall the details: the thing to keep in mind is simply that these bars show how quickly each neuron's weights and bias are changing as the network learns.
To keep the diagram simple, I've shown just the top six neurons in the two hidden layers. I've omitted the input neurons, since they've got no weights or biases to learn. I've also omitted the output neurons, since we're doing layer-wise comparisons, and it makes most sense to compare layers with the same number of neurons. The results are plotted at the very beginning of training, i.e., immediately after the network is initialized. Here they are*
*The data plotted is generated using the program generate_gradient.py. The same program is also used to generate the results quoted later in this section:
The network was initialized randomly, and so it's not surprising that there's a lot of variation in how rapidly the neurons learn. Still, one thing that jumps out is that the bars in the second hidden layer are mostly much larger than the bars in the first hidden layer. As a result, the neurons in the second hidden layer will learn quite a bit faster than the neurons in the first hidden layer. Is this merely a coincidence, or are the neurons in the second hidden layer likely to learn faster than neurons in the first hidden layer in general?
To determine whether this is the case, it helps to have a global way of comparing the speed of learning in the first and second hidden layers. To do this, let's denote the gradient as \(δ^l_j=∂C/∂b^l_j\),i.e., the gradient for the \(j\)th neuron in the \(l\)th layer*
*Back in Chapter 2 we referred to this as the error, but here we'll adopt the informal term "gradient". I say "informal" because of course this doesn't explicitly include the partial derivatives of the cost with respect to the weights, \(∂C/∂w\).. We can think of the gradient \(δ^1\) as a vector whose entries determine how quickly the first hidden layer learns, and \(δ^2\) as a vector whose entries determine how quickly the second hidden layer learns. We'll then use the lengths of these vectors as (rough!) global measures of the speed at which the layers are learning. So, for instance, the length \(∥δ^1∥\) measures the speed at which the first hidden layer is learning, while the length \(‖δ^2‖\) measures the speed at which the second hidden layer is learning.
With these definitions, and in the same configuration as was plotted above, we find \(∥δ^1∥=0.07\) and \(∥δ^2∥=0.31\). So this confirms our earlier suspicion: the neurons in the second hidden layer really are learning much faster than the neurons in the first hidden layer.
What happens if we add more hidden layers? If we have three hidden layers, in a \([784,30,30,30,10]\) network, then the respective speeds of learning turn out to be \(0.012\), \(0.060\), and \(0.283\). Again, earlier hidden layers are learning much slower than later hidden layers. Suppose we add yet another layer with \(30\) hidden neurons. In that case, the respective speeds of learning are \(0.003\), \(0.017\), \(0.070\), and \(0.285\). The pattern holds: early layers learn slower than later layers.
We've been looking at the speed of learning at the start of training, that is, just after the networks are initialized. How does the speed of learning change as we train our networks? Let's return to look at the network with just two hidden layers. The speed of learning changes as follows:
To generate these results, I used batch gradient descent with just \(1,000\) training images, trained over \(500\) epochs. This is a bit different than the way we usually train - I've used no mini-batches, and just \(1,000\) training images, rather than the full \(50,000\) image training set. I'm not trying to do anything sneaky, or pull the wool over your eyes, but it turns out that using mini-batch stochastic gradient descent gives much noisier (albeit very similar, when you average away the noise) results. Using the parameters I've chosen is an easy way of smoothing the results out, so we can see what's going on.
In any case, as you can see the two layers start out learning at very different speeds (as we already know). The speed in both layers then drops very quickly, before rebounding. But through it all, the first hidden layer learns much more slowly than the second hidden layer.
What about more complex networks? Here's the results of a similar experiment, but this time with three hidden layers (a \([784,30,30,30,10]\) network):
Again, early hidden layers learn much more slowly than later hidden layers. Finally, let's add a fourth hidden layer (a \([784,30,30,30,30,10]\) network), and see what happens when we train:
Again, early hidden layers learn much more slowly than later hidden layers. In this case, the first hidden layer is learning roughly 100 times slower than the final hidden layer. No wonder we were having trouble training these networks earlier!
We have here an important observation: in at least some deep neural networks, the gradient tends to get smaller as we move backward through the hidden layers. This means that neurons in the earlier layers learn much more slowly than neurons in later layers. And while we've seen this in just a single network, there are fundamental reasons why this happens in many neural networks. The phenomenon is known as the vanishing gradient problem*
*See Gradient flow in recurrent nets: the difficulty of learning long-term dependencies, by Sepp Hochreiter, Yoshua Bengio, Paolo Frasconi, and Jürgen Schmidhuber (2001). This paper studied recurrent neural nets, but the essential phenomenon is the same as in the feedforward networks we are studying. See also Sepp Hochreiter's earlier Diploma Thesis,Untersuchungen zu dynamischen neuronalen Netzen (1991, in German).
Why does the vanishing gradient problem occur? Are there ways we can avoid it? And how should we deal with it in training deep neural networks? In fact, we'll learn shortly that it's not inevitable, although the alternative is not very attractive, either: sometimes the gradient gets much larger in earlier layers! This is the exploding gradient problem, and it's not much better news than the vanishing gradient problem. More generally, it turns out that the gradient in deep neural networks is unstable, tending to either explode or vanish in earlier layers. This instability is a fundamental problem for gradient-based learning in deep neural networks. It's something we need to understand, and, if possible, take steps to address.
One response to vanishing (or unstable) gradients is to wonder if they're really such a problem. Momentarily stepping away from neural nets, imagine we were trying to numerically minimize a function \(f(x)\) of a single variable. Wouldn't it be good news if the derivative \(f′(x)\) was small? Wouldn't that mean we were already near an extremum? In a similar way, might the small gradient in early layers of a deep network mean that we don't need to do much adjustment of the weights and biases?
Of course, this isn't the case. Recall that we randomly initialized the weight and biases in the network. It is extremely unlikely our initial weights and biases will do a good job at whatever it is we want our network to do. To be concrete, consider the first layer of weights in a \([784,30,30,30,10]\) network for the MNIST problem. The random initialization means the first layer throws away most information about the input image. Even if later layers have been extensively trained, they will still find it extremely difficult to identify the input image, simply because they don't have enough information. And so it can't possibly be the case that not much learning needs to be done in the first layer. If we're going to train deep networks, we need to figure out how to address the vanishing gradient problem.
What's causing the vanishing gradient problem? Unstable gradients in deep neural nets
To get insight into why the vanishing gradient problem occurs, let's consider the simplest deep neural network: one with just a single neuron in each layer. Here's a network with three hidden layers:
Here, \(w_1,w_2,…\) are the weights, \(b_1,b_2,…\) are the biases, and \(C\) is some cost function. Just to remind you how this works, the output \(a_j\) from the \(j\)th neuron is \(σ(z_j)\), where σσ is the usual sigmoid activation function, and \(z_j=w_ja_{j−1}+b_j\) is the weighted input to the neuron. I've drawn the cost \(C\) at the end to emphasize that the cost is a function of the network's output, \(a_4\): if the actual output from the network is close to the desired output, then the cost will be low, while if it's far away, the cost will be high.
We're going to study the gradient \(∂C/∂b_1\) associated to the first hidden neuron. We'll figure out an expression for \(∂C/∂b_1\), and by studying that expression we'll understand why the vanishing gradient problem occurs.
I'll start by simply showing you the expression for \(∂C/∂b_1\). It looks forbidding, but it's actually got a simple structure, which I'll describe in a moment. Here's the expression (ignore the network, for now, and note that σ′σ′ is just the derivative of the \(σ\) function):
The structure in the expression is as follows: there is a \(σ′(z_j)\) term in the product for each neuron in the network; a weight \(w_j\) term for each weight in the network; and a final \(∂C/∂a_4\) term, corresponding to the cost function at the end. Notice that I've placed each term in the expression above the corresponding part of the network. So the network itself is a mnemonic for the expression.
You're welcome to take this expression for granted, and skip to the discussion of how it relates to the vanishing gradient problem. There's no harm in doing this, since the expression is a special case of our earlier discussion of backpropagation. But there's also a simple explanation of why the expression is true, and so it's fun (and perhaps enlightening) to take a look at that explanation.
Imagine we make a small change \(Δb_1\) in the bias \(b_1\). That will set off a cascading series of changes in the rest of the network. First, it causes a change \(Δa_1\) in the output from the first hidden neuron. That, in turn, will cause a change \(Δz_2\) in the weighted input to the second hidden neuron. Then a change \(Δa_2\) in the output from the second hidden neuron. And so on, all the way through to a change \(ΔC\) in the cost at the output. We have
\[ \frac{∂C}{∂b_1}≈\frac{ΔC}{Δb_1}.\label{114}\tag{114} \]
This suggests that we can figure out an expression for the gradient \(∂C/∂b_1\) by carefully tracking the effect of each step in this cascade.
To do this, let's think about how \(Δb_1\) causes the output \(a_1\) from the first hidden neuron to change. We have \(a_1=σ(z_1)=σ(w_1a_0+b_1)\), so
\[ \begin{align} Δa_1 &≈ \frac{∂σ(w_1a_0+b_1)}{∂b_1}Δb_1\label{115}\tag{115} \\ &=σ′(z_1)Δb_1\label{116}\tag{116} \end{align}$$.
That \(σ′(z_1)\) term should look familiar: it's the first term in our claimed expression for the gradient \(∂C/∂b_1\). Intuitively, this term converts a change \(Δb_1\) in the bias into a change \(Δa_1\) in the output activation. That change \(Δa_1\) in turn causes a change in the weighted input \(z_2=w_2a_1+b_2\) to the second hidden neuron:
\[ \begin{align} Δz_2 &≈ \frac{∂z_2}{∂a_1}Δa_1\label{117}\tag{117} \\ &= w_2Δa_1.\label{118}\tag{118} \end{align} \]
Combining our expressions for \(Δz_2\) and \(Δa_1\), we see how the change in the bias \(b_1\) propagates along the network to affect \(z_2\):
\[ Δz_2 ≈ σ′(z1)w_2Δb_1.\label{119}\tag{119} \]
Again, that should look familiar: we've now got the first two terms in our claimed expression for the gradient \(∂C/∂b_1\).
We can keep going in this fashion, tracking the way changes propagate through the rest of the network. At each neuron we pick up a \(σ′(z_j)\) term, and through each weight we pick up a \(w_j\) term. The end result is an expression relating the final change \(ΔC\) in cost to the initial change \(Δb_1\) in the bias:
\[ ΔC≈σ′(z_1)w_2σ′(z_2)…σ′(z_4)\frac{∂C}{∂a_4}Δb_1.\label{120}\tag{120} \]
Dividing by \(Δb_1\) we do indeed get the desired expression for the gradient:
\[ \frac{∂C}{∂b_1}=σ′(z_1)w_2σ′(z_2)…σ′(z_4)\frac{∂C}{∂a_4}.\label{121}\tag{121} \]
Why the vanishing gradient problem occurs: To understand why the vanishing gradient problem occurs, let's explicitly write out the entire expression for the gradient:
\[ \frac{∂C}{∂b_1}=σ′(z_1)w_2σ′(z_2)w_3σ′(z_3)w_4σ′(z_4)\frac{∂C}{∂a_4}.\label{122}\tag{122} \]
Excepting the very last term, this expression is a product of terms of the form \(w_jσ′(z_j)\). To understand how each of those terms behave, let's look at a plot of the function \(σ′\):
The derivative reaches a maximum at \(σ′(0)=1/4\). Now, if we use our standard approach to initializing the weights in the network, then we'll choose the weights using a Gaussian with mean \(0\) and standard deviation \(1\). So the weights will usually satisfy \(|w_j|<1\). Putting these observations together, we see that the terms \(w_jσ′(z_j)\) will usually satisfy \(|w_jσ′(z_j)|<1/4\). And when we take a product of many such terms, the product will tend to exponentially decrease: the more terms, the smaller the product will be. This is starting to smell like a possible explanation for the vanishing gradient problem.
To make this all a bit more explicit, let's compare the expression for \(∂C/∂b_1\) to an expression for the gradient with respect to a later bias, say \(∂C/∂b_3\). Of course, we haven't explicitly worked out an expression for \(∂C/∂b_3\), but it follows the same pattern described above for \(∂C/∂b_1\). Here's the comparison of the two expressions:
The two expressions share many terms. But the gradient \(∂C/∂b_1\) includes two extra terms each of the form \(w_jσ′(z_j)\). As we've seen, such terms are typically less than \(1/4\) in magnitude. And so the gradient \(∂C/∂b_1\) will usually be a factor of \(16\) (or more) smaller than \(∂C/∂b_3\). This is the essential origin of the vanishing gradient problem.
Of course, this is an informal argument, not a rigorous proof that the vanishing gradient problem will occur. There are several possible escape clauses. In particular, we might wonder whether the weights \(w_j\) could grow during training. If they do, it's possible the terms \(w_jσ′(z_j)\) in the product will no longer satisfy \(|w_jσ′(z_j)|<1/4\). Indeed, if the terms get large enough - greater than \(1\) - then we will no longer have a vanishing gradient problem. Instead, the gradient will actually grow exponentially as we move backward through the layers. Instead of a vanishing gradient problem, we'll have an exploding gradient problem.
The exploding gradient problem: Let's look at an explicit example where exploding gradients occur. The example is somewhat contrived: I'm going to fix parameters in the network in just the right way to ensure we get an exploding gradient. But even though the example is contrived, it has the virtue of firmly establishing that exploding gradients aren't merely a hypothetical possibility, they really can happen.
There are two steps to getting an exploding gradient. First, we choose all the weights in the network to be large, say \(w_1=w_2=w_3=w_4=100\). Second, we'll choose the biases so that the \(σ′(z_j)\) terms are not too small. That's actually pretty easy to do: all we need do is choose the biases to ensure that the weighted input to each neuron is \(z_j=0\) (and so \(σ′(z_j)=1/4\). So, for instance, we want \(z_1=w_1a_0+b_1=0\). We can achieve this by setting \(b_1=−100∗a_0\). We can use the same idea to select the other biases. When we do this, we see that all the terms \(w_jσ′(z_j)\) are equal to \(100∗1/4=25\). With these choices we get an exploding gradient.
The unstable gradient problem: The fundamental problem here isn't so much the vanishing gradient problem or the exploding gradient problem. It's that the gradient in early layers is the product of terms from all the later layers. When there are many layers, that's an intrinsically unstable situation. The only way all layers can learn at close to the same speed is if all those products of terms come close to balancing out. Without some mechanism or underlying reason for that balancing to occur, it's highly unlikely to happen simply by chance. In short, the real problem here is that neural networks suffer from an unstable gradient problem. As a result, if we use standard gradient-based learning techniques, different layers in the network will tend to learn at wildly different speeds.
Exercise
- In our discussion of the vanishing gradient problem, we made use of the fact that \(|σ′(z)|<1/4\). Suppose we used a different activation function, one whose derivative could be much larger. Would that help us avoid the unstable gradient problem?
The prevalence of the vanishing gradient problem: We've seen that the gradient can either vanish or explode in the early layers of a deep network. In fact, when using sigmoid neurons the gradient will usually vanish. To see why, consider again the expression \(|wσ′(z)|\). To avoid the vanishing gradient problem we need \(|wσ′(z)|≥1\). You might think this could happen easily if \(w\) is very large. However, it's more difficult than it looks. The reason is that the \(σ′(z)\) term also depends on \(w: σ′(z)=σ′(wa+b)\), where \(a\) is the input activation. So when we make \(w\) large, we need to be careful that we're not simultaneously making \(σ′(wa+b)\) small. That turns out to be a considerable constraint. The reason is that when we make \(w\) large we tend to make \(wa+b\) very large. Looking at the graph of \(σ′\) you can see that this puts us off in the "wings" of the \(σ′\) function, where it takes very small values. The only way to avoid this is if the input activation falls within a fairly narrow range of values (this qualitative explanation is made quantitative in the first problem below). Sometimes that will chance to happen. More often, though, it does not happen. And so in the generic case we have vanishing gradients.
Problems
- Consider the product \(|wσ′(wa+b)|\). Suppose \(|wσ′(wa+b)|≥1\). (1) Argue that this can only ever occur if \(|w|≥4\). (2) Supposing that \(|w|≥4\), consider the set of input activations aa for which \(|wσ′(wa+b)|≥1\). Show that the set of aa satisfying that constraint can range over an interval no greater in width than
\[ \frac{2}{|w|}ln\left(\frac{|w|(1+\sqrt{1−4/|w|}}{2}−1\right).\label{123}\tag{123} \]
(3) Show numerically that the above expression bounding the width of the range is greatest at \(|w|≈6.9\), where it takes a value \(≈0.45\). And so even given that everything lines up just perfectly, we still have a fairly narrow range of input activations which can avoid the vanishing gradient problem. - Identity neuron: Consider a neuron with a single input, \(x\), a corresponding weight, \(w_1\), a bias \(b\), and a weight \(w_2\) on the output. Show that by choosing the weights and bias appropriately, we can ensure \(w_2σ(w_1x+b)≈x\) for \(x∈[0,1]\). Such a neuron can thus be used as a kind of identity neuron, that is, a neuron whose output is the same (up to rescaling by a weight factor) as its input. Hint: It helps to rewrite \(x=1/2+Δ\), to assume \(w_1\) is small, and to use a Taylor series expansion in \(w_1Δ\).