Stabilizing Elastic Weight Consolidation method in practical ML tasks and using weight importances for neural network pruning

09/21/2021
by   Alexey Kutalev, et al.
0

This paper is devoted to the features of the practical application of Elastic Weight Consolidation method. Here we will more rigorously compare the known methodologies for calculating the importance of weights when applied to networks with fully connected and convolutional layers. We will also point out the problems that arise when applying the Elastic Weight Consolidation method in multilayer neural networks with convolutional layers and self-attention layers, and propose method to overcome these problems. In addition, we will notice an interesting fact about the use of various types of weight importance in the neural network pruning task.

READ FULL TEXT

Please sign up or login with your details

Forgot password? Click here to reset