The Challenge of Vanishing/Exploding Gradients in Deep Neural Networks

Yash Last Updated : 21 Jun, 2024
9 min read

Introduction

This article explains the problem of exploding and vanishing gradients while training a deep neural network and the techniques that can be used to cleverly get past this impediment.In this article, you will get a clarify about vanishing and exploding gradients and their problem and solutions with that you will get to know about exploding agents and vanishing agents what they are and how their problem is solved in this article. So let’s begin.

This article was published as a part of the Data Science Blogathon

A Glimpse of the Backpropagation Algorithm

We know that the backpropagation algorithm is the heart of neural network training. Let’s have a glimpse over this algorithm that has proved to be a harbinger in the evolution as well as the revolution of Deep Learning.

exploding and vanishing gradients back propagation
Source: Google Images 

After propagating the input features forward to the output layer through the various hidden layers consisting of different/same activation functions, we come up with a predicted probability of a sample belonging to the positive class ( generally, for classification tasks).

  • Now, the backpropagation algorithm propagates backward from the output layer to the input layer calculating the error gradients on the way.
  • After computing the gradients of the cost function with respect to each parameter (weights and biases) in the neural network, the algorithm uses these gradients to take a gradient descent step towards the minimum, updating the value of each parameter in the network.
exploding and vanishing gradients gif
GIF Source: gyfcat.com

Understanding the Problems

What is Vanishing Gradient?

As the backpropagation algorithm advances downwards(or backward) from the output layer towards the input layer, the gradients often get smaller and smaller and approach zero which eventually leaves the weights of the initial or lower layers nearly unchanged. As a result, the gradient descent never converges to the optimum. This is known as the vanishing gradients problem.

What is Exploding Gradient?

On the contrary, in some cases, the gradients keep on getting larger and larger as the backpropagation algorithm progresses. This, in turn, causes very large weight updates and causes the gradient descent to diverge. This is known as the exploding gradients problems.

Why Do the Gradients Even Vanish/Explode?

Certain activation functions, like the logistic function (sigmoid), have a very huge difference between the variance of their inputs and the outputs. In simpler words, they shrink and transform a larger input space into a smaller output space that lies between the range of [0,1].

exploding and vanishing gradients why
Image source: Google Images

Observing the above graph of the Sigmoid function, we can see that for larger inputs (negative or positive), it saturates at 0 or 1 with a derivative very close to zero. Thus, when the backpropagation algorithm chips in, it virtually has no gradients to propagate backward in the network, and whatever little residual gradients exist keeps on diluting as the algorithm progresses down through the top layers. So, this leaves nothing for the lower layers.

Similarly, in some cases suppose the initial weights assigned to the network generate some large loss. Now the gradients can accumulate during an update and result in very large gradients which eventually results in large updates to the network weights and leads to an unstable network. The parameters can sometimes become so large that they overflow and result in NaN values.

exploding and vanishing gradients google images
Image Source: Google Images

How to Know if Our Model is Suffering From the Exploding/Vanishing Gradient Problem?

Following are some signs that can indicate that our gradients are vanishing and exploding gradients :

Exploding Vanishing
   
There is an exponential growth in the model parameters. The parameters of the higher layers change significantly whereas the parameters of lower layers would not change much (or not at all).
   
The model weights may become NaN during training. The model weights may become 0 during training.
   
The model experiences  avalanche learning. The model learns very slowly and perhaps the training stagnates at a very early stage just after a few iterations.
   

Certainly, neither do we want our signal to explode or saturate nor do we want it to die out. The signal needs to flow properly both in the forward direction when making predictions as well as in the backward direction while calculating gradients.

Solutions

Now that we understand the vanishing/exploding gradients problems, we can learn some techniques to fix them.

Proper Weight Initialization 

In their paper, researchers Xavier Glorot, Antoine Bordes, and Yoshua Bengio proposed a way to remarkably alleviate this problem.

For the proper flow of the signal, the authors argue that:

  • The variance of outputs of each layer should be equal to the variance of its inputs.
  • The gradients should have equal variance before and after flowing through a layer in the reverse direction.

Although both conditions cannot hold for any layer in the network unless the number of inputs to the layer (fanin) equals the number of neurons in the layer (fanout), they proposed a well-proven compromise that works incredibly well in practice. They randomly initialize the connection weights for each layer in the network using the following equation, popularly known as Xavier initialization (after the author’s first name) or Glorot initialization (after his last name).

where  fanavg = ( fanin + fanout ) / 2
  • Normal distribution with mean 0 and variance σ2 = 1/ fanavg
  • Or a uniform distribution between -r  and +r , with r = sqrt( 3 / fanavg )         

Following are some more very popular weight initialization strategies for different activation functions, they only differ by the scale of variance and by the usage of either fanavg or fanin 

for uniform distribution, calculate r as: r = sqrt( 3*σ2 )

exploding and vanishing gradients 3
Source : A book from Aurelien Geron

Using the above initialization strategies can significantly speed up the training and increase the odds of gradient descent converging at a lower generalization error. 

Wait, but how do we put these strategies into code ??

Relax! we will not need to hardcode anything, Keras does it for us.

  • Keras uses Xavier’s initialization strategy with uniform distribution.
  • If we wish to use a different strategy than the default one, this can be done using the kernel_initializer parameter while creating the layer. For example :
keras.layer.Dense(25, activation = "relu", kernel_initializer="he_normal")

or

keras.layer.Dense(25, activation = "relu", kernel_initializer="he_uniform")

 If we wish to use  use the initialization based on fanavg rather than fanin , we can use the VarianceScaling initializer like this :

he_avg_init = keras.initializers.VarianceScaling(scale=2., mode='fan_avg', distribution='uniform')
keras.layers.Dense(20, activation="sigmoid", kernel_initializer=he_avg_init)

Using Non-saturating Activation Functions

In an earlier section, while studying the nature of sigmoid activation function, we observed that its nature of saturating for larger inputs (negative or positive) came out to be a major reason behind the vanishing and exploding gradients thus making it non-recommendable to use in the hidden layers of the network.

So to tackle the issue regarding the saturation of activation functions like sigmoid and tanh, we must use some other non-saturating functions like ReLu and its alternatives.

ReLU ( Rectified Linear Unit )

Source: Google Images
Relu(z) = max(0,z)
  • Outputs 0 for any negative input.
  • Range: [0, infinity]

Unfortunately, the ReLu function is also not a perfect pick for the intermediate layers of the network “in some cases”. It suffers from a problem known as dying ReLus wherein some neurons just die out, meaning they keep on throwing 0 as outputs with the advancement in training.

Read about the dying relus problem in detail here.

Some popular alternative functions of the ReLU that mitigates the problem of vanishing gradients when used as activation for the intermediate layers of the network  are LReLU, PReLU, ELU, SELU :

LReLU (Leaky ReLU)

LReLU
Source: Google images
LeakyReLUα(z) = max(αz, z)
  • The amount of “leak” is controlled by the hyperparameter α, it is the slope of the function for z < 0.
  • The smaller slope for the leak ensures that the neurons powered by leaky Relu never die; although they might venture into a state of coma for a long training phase they always have a chance to eventually wake up.
  • The model can also train α, learning its value during training. This variant, where α is considered a parameter rather than a hyperparameter, is called parametric leaky ReLU (PReLU).

ELU (Exponential Linear Unit)

ELU
Source: Google Images

For z < 0, it takes on negative values which allow the unit to have an average output closer to 0 thus alleviating the vanishing gradient problem

  • For z < 0, the gradients are non zero. This avoids the dead neurons problem.
  • For α = 1, the function is smooth everywhere, this speeds up the gradient descent since it does not bounce right and left around z=0.
  • A scaled version of this function ( SELU: Scaled ELU ) is also used very often in Deep Learning.

Batch Normalization

Using He initialization along with any variant of the ReLU activation function can significantly reduce the chances of vanishing/exploding problems at the beginning. However, it does not guarantee that the problem won’t reappear during training.

In 2015, Sergey Ioffe and Christian Szegedy proposed a paper in which they introduced a technique known as Batch Normalization to address the problem of vanishing/exploding gradient problem.

The Following key points explain the intuition behind BN and how it works:

  • It consists of adding an operation in the model just before or after the activation function of each hidden layer.
  • This operation simply zero-centers and normalizes each input, then scales and shifts the result using two new parameter vectors per layer: one for scaling, the other for shifting.
  • In other words, the operation lets the model learn the optimal scale and mean of each of the layer’s inputs.
  • To zero-center and normalize the inputs, the algorithm needs to estimate each input’s mean and standard deviation.
  • It does so by evaluating the mean and standard deviation of the input over the current mini-batch (hence the name “Batch Normalization”).
model = keras.models.Sequential([keras.layers.Flatten(input_shape=[28, 28]),keras.layers.BatchNormalization(),keras.layers.Dense(300, activation="relu"),keras.layers.BatchNormalization(),keras.layers.Dense(100, activation="relu"),keras.layers.BatchNormalization(),keras.layers.Dense(10, activation="softmax")])

we just added batch normalization after each layer ( dataset : FMNIST)model.summary()

model summary

Gradient Clipping

Another popular technique to mitigate the exploding gradient problem is to clip the gradients during backpropagation so that they never exceed some threshold. This is called Gradient Clipping.

  • This optimizer will clip every component of the gradient vector to a value between –1.0 and 1.0.
  • This means we will clip all the partial derivatives of the loss with respect to each trainable parameter between –1.0 and 1.0.
optimizer = keras.optimizers.SGD(clipvalue = 1.0)
  • The threshold is a hyperparameter we can tune.
  • The orientation of the gradient vector may change due to this: for eg, let the original gradient vector be [0.9, 100.0] pointing mostly in the direction of the second axis, but once we clip it by some value, we get [0.9, 1.0] which now points somewhere around the diagonal between the two axes.
  • To ensure that the orientation remains intact even after clipping, we should clip by norm rather than by value.
optimizer = keras.optimizers.SGD(clipnorm = 1.0)
  • If the threshold we pick is less than the ℓ2 norm, we will clip the whole gradient. For example, if clipnorm=1, we will clip the vector [0.9, 100.0] to [0.00899, 0.999995], thus preserving its orientation.

Conclusion

I hope the article would have helped you understand the problem of exploding/vanishing gradients thoroughly and you will be able to identify if by any chance your model suffers from it and you can easily tackle it and train your model efficiently.

Thanks for your time. HAPPY LEARNING

The media shown in this article are not owned by Analytics Vidhya and are used at the Author’s discretion.

Frequently Asked Questions

Q1. What are exploding gradients and vanishing gradients?

A. Exploding gradients occur when model gradients grow uncontrollably during training, causing instability. Vanishing gradients happen when gradients shrink excessively, hindering effective learning and updates.

Q2. What is meant by the vanishing gradient problem?

A. The vanishing gradient problem refers to the phenomenon where gradients diminish as they are propagated backward through layers, making it difficult for models to learn and update weights, especially in deep networks.

Q3. What is vanishing exploding gradient problem in RNN?

A. In RNNs, the vanishing gradient problem impedes learning long-term dependencies due to diminishing gradients, while the exploding gradient problem causes instability and divergent updates due to excessively large gradients.

Q4. What is exploding gradient plot?

A. An exploding gradient plot visualizes the sudden increase in gradient values during training, indicating instability and potential divergence, often requiring gradient clipping or other techniques to mitigate the issue.

Responses From Readers

Clear

Congratulations, You Did It!
Well Done on Completing Your Learning Journey. Stay curious and keep exploring!

We use cookies essential for this site to function well. Please click to help us improve its usefulness with additional cookies. Learn about our use of cookies in our Privacy Policy & Cookies Policy.

Show details