1 Introduction
Abstract

The most common method for DNN pruning is hard thresholding of network weights, followed by retraining to recover any lost accuracy. Recently developed smart pruning algorithms use the DNN response over the training set for a variety of cost functions to determine redundant network weights, leading to less accuracy degradation and possibly less retraining time. For experiments on the total pruning time (pruning time + retraining time) we show that hard thresholding followed by retraining remains the most efficient way of reducing the number of network parameters. However smart pruning algorithms still have advantages when retraining is not possible. In this context we propose a novel smart pruning algorithm based on difference of convex functions optimisation and show that it is often orders of magnitude faster than competing approaches while achieving the lowest classification accuracy degradation. Furthermore we investigate theoretically the effect of hard thresholding on DNN accuracy. We show that accuracy degradation increases with remaining network depth from the pruned layer. We also discover a link between the latent dimensionality of the training data manifold and network robustness to hard thresholding.

 

Revisiting hard thresholding for DNN pruning.


 


Konstantinos Pitas                        Mike Davies                        Pierre Vandergheynst

konstanstinos.pitas@epfl.ch                        Mike.Davies@ed.ac.uk                        pierre.vandergheynst@epfl.ch

1 Introduction

Deep neural networks have achieved state-of-the art results in a number of machine learning tasks [LeCun et al., 2015]. Training such networks is computationally intensive and often requires dedicated and expensive hardware. Furthermore, the resulting networks often require a considerable amount of memory to be stored. Using a Pascal Titan X GPU the popular AlexNet and VGG-16 models require 13 hours and 7 days, respectively, to train, while requiring 200MB and 600MB, respectively, to store. The large memory requirements limit the use of DNNs in embedded systems and portable devices such as smartphones, which are now ubiquitous.

A number of approaches have been proposed to reduce the DNN size during training time, often with little or no degradation to classification performance. Approaches include introducing bayesian, sparsity-inducing priors [Louizos et al., 2017], [Blundell et al., 2015], [Molchanov et al., 2017], [Dai et al., 2018] and binarization [Hou et al., 2016] [Courbariaux et al., 2016].Other methods include the hashing trick used in [Chen et al., 2015], tensorisation [Novikov et al., 2015] and efficient matrix factorisations [Yang et al., 2015].

There has also been work in reducing the number of network parameters after training. Some first work in this direction has been [Kim et al., 2015] [Han et al., 2015a] [Han et al., 2015b] where hard thresholding is applied to network weights. However weight magnitude is a mediocre predictor for determining whether a weight is essential to network accuracy and following hard thresholding network accuracy often falls significantly. Therefore hard thresholding often has to be followed by a retraining procedure. Ideally one would want to remove parameters by minimizing the reduction in network accuracy over the training set, but this is computationally prohibitive. Smart pruning algorithms aim to overcome this problem by minimizing a surrogate cost function while pruning. In [Aghasi et al., 2016][Aghasi et al., 2018] the authors propose a convexified layerwise pruning algorithm termed Net-Trim, that minimizes the norm between pruned and unpruned layer output representations. The authors in [Dong et al., 2017] propose LOBS, an algorithm for layerwise pruning by approximating the loss function using a second order Taylor expansion. In [Baykal et al., 2018] the authors use a coreset approach to prune individual layers and enforce an norm penalty between pruned and unpruned layer output representations with high probability.

Given that DNNs represent a highly non-linear function it is also of independent interest to explore theoretically what is the effect of pruning on DNN accuracy. Hard thresholding a neural network layer introduces a perturbation to the latent signal representations generated by that layer. As the pertubated signal passes through layers of non-linear projections, the perturbation could become arbitrarily large. DNN robustness to hidden layer perturbations has been investigated for random noise in [Raghu et al., 2016] and for the case of pruning in [Aghasi et al., 2016] and [Dong et al., 2017]. In the last two works the authors conduct a theoretical analysis using the Lipschitz properties of DNNs showing the stability of the latent representations, over the training set, after pruning by assuming that where are the weights at a given layer. The Lipschitz properties of DNNs have also been used to analyze their generalization error (GE), [Sokolic et al., 2017] [Bartlett et al., 2017] [Neyshabur et al., 2017]. Very close to our work [Bartlett et al., 2017] show that the difficulty of a training set can be captured by normalizing the classification margins of training samples with a ”spectral complexity” term that includes the spectral norms of the layer weights. Randomly labelled data as opposed to real labelled data produce networks with much smaller normalized margins, reflecting the difficulty of the randomly labelled dataset.

For the practical pruning section including with and without retraining we focus on pruning only the fully connected non-linear layers of a network. Two of the proposed algorithms are only applicable in fully connected layers. Furthermore fully connected layers often include of network parameters, and are the most redundant of layers in most architectures. Finally the softmax linear layer can be readily compressed using an SVD decomposition [Yang et al., 2015].

\@fb@secFB

1.1 Contributions

In this work we make the following contributions

  • We perform pruning and retraining experiments for various datasets and architectures and discover that hard thresholding is the most efficient pruning pipeline often taking just a fraction of the total pruning and retraining time of other approaches. For the case without retraining smart pruning approaches lead to significant accuracy gains for the same level of sparsity, especially for DNNs with only fully connected layers.

  • We propose a novel smart pruning algorithm called FeTa that can be cast as a difference of convex functions problem, and has an efficient solution. For a fully connected layer with input dimension , output dimension and training samples, the time complexity of our iterative algorithm scales like , where is the precision of the solution, is related to the Lipschitz and strong convexity constants, and is the outer iteration number. Competing approaches scale like and and we conduct experiments showing that our algorithm leads to higher accuracy in the pruned DNN and is often orders of magnitude faster.

  • We build upon the work of [Sokolic et al., 2017] to bound the GE of a DNN for the case of hard thresholding of hidden layer weights. In contrast to the analysis of [Aghasi et al., 2016] and [Dong et al., 2017] our analysis correctly predicts that for the same level of sparsity, accuracy degrades with the remaining depth of the thresholded layer.

  • Our theoretical analysis also predicts that a DNN trained on a data manifold with lower intrinsic dimensionality will be more robust to hard thresholding. We provide empirical evidense that validates this prediction.

.

\@fb@secFB

1.2 Notation and Definitions

We use the following notation in the sequel: matrices, column vectors, scalars and sets are denoted by boldface upper-case letters (), boldface lower-case letters (), italic letters () and calligraphic upper-case letters (), respectively. The covering number of with -metric balls of radius is denoted by . A -regular -dimensional manifold, where is a constant that captures ”intrinsic” properties, is one that has a covering number . We define as the number of non-zeros of a vector or matrix, as the rectifier non-linearity, as the elementwise indicator function, and as the Hadamard product.

2 Retraining Experiments

We consider a classification problem, where we observe a vector that has a corresponding class label . The set is called the input space, is called the label space and denotes the number of classes. The samples space is denoted by and an element of is denoted by . We assume that samples from are drawn according to a probability distribution defined on . A training set of samples drawn from is denoted by .

We consider DNN classifiers defined as

(1)

where is the th element of dimensional output of a DNN . We assume that is composed of layers

(2)

where represents the th layer with parameters , . The output of the th layer is denoted , i.e. . The input layer corresponds to and the output of the last layer is denoted by .

For each training signal we assume also that we have access to the inputs and the outputs of a fully connected layer to be pruned. For an unpruned weight matrix the layer is defined by the equation . We denote the matrix concatenating all the latent input vectors and the matrix concatenating all the latent output vectors . We are looking for a new pruned weight matrix such that the new layer will be defined as .


Dense Mnist Conv Mnist Dense F-Mnist Conv F-Mnist Dense Cifar Conv Cifar
Time Acc Time Acc Time Acc Time Acc Time Acc Time Acc
Net-Trim 3m 94.5% 16m 97% 15m 87% 24m 92% 13m 45% 39m 78%
LOBS 3m 96% 3m 97% 3m 87% 3h 92% 1h 45% 3h 78%
Corenet 1m 93% 3m 97% 42m 87% 18m 92% 1h 45% 53m 76%
Threshold 96% 97% 87% 92% 45% 78%
Table 1: Test accuracy and total pruning time (hours (H), minutes (M), seconds (S)) for Mnist, Fashion Mnist and Cifar-10.

We compare three state of the art smart pruning techniques with hard thresholding for pruning feedforward neural networks. A smart pruning algorithm is defined as which takes as inputs the original layer weights, the latent representation inputs and outputs to the layer and outputs the new pruned weight matrix. We compare the following algorithms: (i) Hard thresholding defined as , where is a positive constant. (ii) Net-Trim [Aghasi et al., 2018] (iii) LOBS [Dong et al., 2017] (iv) Corenet [Baykal et al., 2018]. We refer to the original papers for details. We used implementations by the authors for Net-Trim and LOBS and created our own implementation of Corenet.

Experiments were performed on three commonly used datasets, Mnist[LeCun et al., 1998], FashionMnist[Xiao et al., 2017] and Cifar-10[Krizhevsky and Hinton, 2009]. For each dataset we used half the test set as a validation set. We also tested two different architectures for each dataset one fully connected and one convolutional. We plot the architectures tested in Figure 1. We refer to implementation details in APPENDIX C.

For fully connected architectures both pruning and retraining was done on a MacBook Pro with CPU Intel Core i7 @ 2.8GHz and RAM 16GB 1600 MHz DDR3 using only the CPU. For convolutional architectures pruning was done on 48 Intel Xeon CPUs E5-2650 v4 @ 2.20GHz and retraining was done using a single GeForce GTX 1080 GPU.

(a)
(b)
(c)
(d)
(e)
(f)
Figure 1: The DNN architectures that are tested: a) Dense Mnist b) Convolutional Mnist c) Dense Fashion Mnist d) Convolutional Fashion Mnist e) Dense Cifar-10 f) Convolutional Cifar-10. We denote convolutional layers with ouput channels and filter support by . We denote fully connected layers with output dimensions by . We denote max-pooling layers with pooling regions by and softmax layers with output dimensions by .

We first prune the fully connected nonlinear layers of the DNN architectures using the different algorithms to sparsity. We then retrain the entire pruned architecture until the lost accuracy is recovered. We present the results comparing total pruning time (pruning time + retraining time) for all architectures in Table 1. We see that, using retraining, hard thresholding is able to achieve the same accuracy in the pruned model compared to other approaches at only a fraction of the time. We see furthermore that Net-Trim is faster than Corenet and LOBS in the larger scale Cifar experiments. Finally LOBS is by far the most time consuming algorithm.

We conclude that for the architectures tested and the computational resources that where allocated for pruning and retraining, smart pruning techniques represent a significant increase in time complexity without providing any benefit in the final network accuracy. We must note that while we consider the experimental setting which we consider the most realistic it might still be possible to see a benefit from smart pruning techniques if more efficient computational resources are allocated to smart pruning or less efficient resources are available for retraining. For example retraining a convolutional network using only CPUs might be less efficient than smart pruning.

We determine that pruning accuracy vs pruning speed is a crucial parameter to be optimised by any smart pruning algorithm. In this context we derive in the next section a novel pruning algorithm that is often orders of magnitude faster than competing approaches while still achieving the highest network accuracy after pruning.

3 The FeTa algorithm

In this section we derive our novel smart pruning algorithm.

\@fb@secFB

3.1 Net-Trim

We first make a detailed description of the Net-Trim algorithm [Aghasi et al., 2018] which will be the basis for our approach. For , Net-Trim aims to minimize

(3)

where is a tolerance parameter. Intuitively Net-Trim tries to find a sparse matrix such that the output of the pruned nonlinear layer stays close in terms of the Frobenius norm to the original unpruned output . The optimisation above is non-convex and the authors of Net-Trim introduce a convexified formulation. For , and , Net-Trim minimizes

(4)

which is solved using the ADMM approach. There are two main issues with this formulation. One is time complexity as the ADMM implementation has to solve a Cholesky factorisation which scales like where is the layer input dimension. The second is space complexity as the algorithm has to create inequality constraints, and thus allocate memory, proportional to the number of data samples . In [Aghasi et al., 2018] the space complexity under some assumptions has been shown to be of the order . We note however that for large layers can be higher than even for high sparsity levels.

\@fb@secFB

3.2 DC decomposition

We aim to overcome the shortcomings of Net-Trim by reformulating the optimisation problem as a difference of convex functions problem. For we reformulate the optimisation problem that we want to solve as

(5)

where is the sparsity parameter. The term ensures that the nonlinear projection remains the same for training signals. The term is any convex regulariser which imposes the desired structure on the weight matrix .

The objective in Equation 5 is non-convex. We show that the optimisation of this objective can be cast as a difference of convex functions (DC) problem. We assume just one training sample , for simplicity, with latent representations and

(6)

Notice that after the split the first term () is convex while the second () is concave. We note that by definition of the ReLu and set

(7)
(8)

Then by summing over all the samples we get

(9)

which is difference of convex functions. The rectifier nonlinearity is non-smooth, but we can alleviate that by assuming a smooth approximation. A common choice for this task is , with a positive constant.

\@fb@secFB

3.3 Optimisation

It is well known that DC programs have efficient optimisation algorithms. We propose to use the DCA algorithm [Tao and An, 1997]. DCA is an iterative algorithm that consists in solving, at each iteration, the convex optimisation problem obtained by linearizing (the non-convex part of ) around the current solution. Although DCA is only guaranteed to reach local minima the authors of [Tao and An, 1997] state that DCA often converges to the global minimum, and has been used succefully to optimise a fully connected DNN layer [Fawzi et al., 2015]. At iteration of DCA, the linearized optimisation problem is given by

(10)

where is the solution estimate at iteration . The detailed procedure is then given in algorithms 1 and 2. We assume that the regulariser is convex but possibly non-smooth in which case the optimisation can be performed using proximal methods.

1:  Choose initial point:
2:  for k = 1,…,K do
3:     Compute .
4:     Use Algorithm 2 to solve the convex optimisation problem:
(11)
5:  end for
6:  If return .
Algorithm 1 FeTa (Fast and Efficient Trimming Algorithm)
1:  Initialization:
2:  for s = 1,…,S do
3:     
4:     
5:     for t = 1,2,…,T do
6:        Choose randomly chosen minibatch.
7:        
8:        
9:        
10:     end for
11:     
12:  end for
13:  Return
Algorithm 2 Acc-Prox-SVRG
(a) Dense Mnist
(b) Dense Fashion Mnist
(c) Dense Cifar
(d) Conv Mnist
(e) Conv Fashion Mnist
(f) Conv Cifar
Figure 2: The test accuracy (%) of pruned architectures vs the running time of the pruning algorithms in hours.

In order to solve the linearized problem we propose to use Accelerated Proximal SVRG (Acc-Prox-SVRG), which was presented in [Nitanda, 2014]. We detail this method in Algorithm 2. At each iteration a minibatch and is drawn. The gradient for the smooth part is calculated and the algorithm takes a step in that direction with step size . Then the proximal operator for the non-smooth regulariser is applied to the result. The hyperparameters for Acc-Prox-SVRG are the acceleration parameter and the gradient step . We have found that in our experiments, using and gives the best results.

Our proposed algorithm has time complexity , where is the precision of the solution, is related to the Lipschitz and strong convexity constants, and is the outer iteration number compared to for Net-trim. Furthermore the optimisation can be done in a stochastic manner using data minibatches greatly reducing the algorithm space complexity. We name our algorithm FeTa, Fast and Efficient Trimming Algorithm.

4 Experiments without retraining.

We now present detailed comparison experiments for pruning DNNs without retraining. The experimental setup (datasets and architectures) is identical to the one of Section 2. We prune both fully connected layers of all architectures to sparsity and plot the tradeoff between accuracy degradation and time complexity in Figure 2. The first thing that we notice is that Corenet and Net-Trim have inconsistent performance with regards to hard thresholding. While in some experiments their results are better or equal to hard thresholding in others they are significantly worse. On the other hand the FeTa and LOBS enjoy significant improvements in accuracy after pruning, while FeTa is up to faster than LOBS in some experiments. We also notice that the gains over hard thresholding are significant in fully connected architectures. On the contrary fully connected layers in convolutional DNN architectures are extremely redundant, with the majority of the non-linear separation of the data manifold being performed by the convolutional layers. Overall FeTa significantly outperforms Hard Thresholding and other smart pruning approaches both in final test accuracy and pruning speed, when retraining is not possible.

5 Generalization Error

We have already seen that hard thresholding followed by retraining is a very efficient pruning method. It is also interesting to explore theoretically what will be the response of the network after hard thresholding different layers at different sparsity levels. In the following section we use tools from the robustness framework [Xu and Mannor, 2012] to bound the generalization error of the new architecture induced by hard thresholding.

We need the following two definitions of the classification margin and the score that we take from [Sokolic et al., 2017]. These will be useful later for measuring the generalization error.

Definition 5.1.

(Score). For a classifier a training sample has a score

(12)

where is the Kronecker delta vector with , and is the output class for from classifier which can also be .

Definition 5.2.

(Training Sample Margin). For a classifier a training sample has a classification margin measured by the norm if

(13)

The classification margin of a training sample is the radius of the largest metric ball (induced by the norm) in centered at that is contained in the decision region associated with the classification label . Note that it is possible for a classifier to misclassify a training point .

We are now ready to state our main result.

Theorem 5.1.

Assume that is a (subset of) -regular k-dimensional manifold, where . Assume also that the DNN classifier achieves a lower bound to the classification score and take to be the loss. Furthermore assume that we prune classifier on layer using hard thresholding to obtain a new classifier . Then for any , with probability at least , when ,

(14)

where and can be considered constants related to the data manifold and the training sample size, is the margin and is the maximum error of the layer over the training set.

The detailed proof can be found in Appendix A. The bound depends on two constants related to intrinsic properties of the data manifold, the regularity constant and the intrinsic data dimensionality . In particular the bound depends exponentially on the intrinsic data dimensionality . Thus more complex datasets are expected to lead DNNs that are less robust to pruning. The bound also depends on the spectral norm of the hidden layers . Small spectral norms lead to a larger base in and thus to tigher bounds.

Our result is quite pessimistic as the hard thresholding error is multiplied by the factor . Thus in our analysis the GE grows exponentially with respect to the remaining layer depth of the pertubated layer. This is in line with previous work [Raghu et al., 2016] [Han et al., 2015b] that demonstrates that layers closer to the input are much less robust compared to layers close to the output.

We can use a perturbation bound introduced in [Neyshabur et al., 2017] to extend the above bound to include pruning of multiple layers

Theorem 5.2.

Assume that is a (subset of) -regular k-dimensional manifold, where . Assume also that the DNN classifier achieves a lower bound to the classification score and take to be the loss. Furthermore assume that we prune classifier on all layers using hard thresholding, to obtain a new classifier . Then for any , with probability at least , when ,

(15)

where and can be considered constants related to the data manifold and the training sample size, is the margin and is the per layer perturbation matrix induced by hard thresholding.

(a)
(b)
(c)
Figure 3: a)Mnist convolutional b)Cifar convolutional c)Cifar convolutional for different PCA values. Layer ”0” corresponds to the input layer.

The detailed proof can be found in Appendix B. We note also the generality of our result; even though we have assumed a specific form of pruning, the GE bound holds for any type of bounded perturbation to a hidden layer.

We make here an important note regarding generalization bounds for neural networks in general. All state of the art recent results such as [Neyshabur et al., 2017] [Bartlett et al., 2017] [Arora et al., 2018] [Golowich et al., 2018] provide only vacuous estimates of the generalization error. At the same time while the values of the generalization error estimated by these techinques are loose by orders of magnitude they still provide meanigfull insights into various properties of deep neural networks. In this context we will validate our theoretical framework by showing remaining layer depth correlates with an exponential decrease in DNN test accuracy. Also we will show that increased intrinsic data dimentionality correlates with decreased DNN robustness to pruning.

Finally in the statement of our theorem we assume that this implies that all training samples remain correctly classified after pruning. What happens if this quantity goes to zero? In this case we can imagine the following three step procedure. We first start from a clean network and prune it and use our formula until the above quantity goes to zero. We can then recompute all relevant quantities for our bound including accuracy over the training set, margins, scores and spectral norms. We can then reapply our formula as the the quantity is by definition positive for a small enough pruning. Importantly the rate of margin decrease remains the same for all sparsity levels.

6 Experiments on Generalization Error

\@fb@secFB

6.1 Pruning different layers

In this experiment we use the convolutional MNIST and the convolutional CIFAR architectures from section 2. We prune individual layers from the two networks using hard thresholding for different sparsity levels from to sparsity and compute the network accuracy. We plot the results in Figure 3. We see that as predicted by our bound the network accuracy drops exponentially with remaining layer depth (the number of layers from the pruned layer to the output). This exponential behaviour is the result of the term , and not the result of the dimensionality exponent factor . This suggests that if one has to chose it is beneficial to prune the layers close to the output first as this is have a minimal impact on the network accuracy.

\@fb@secFB

6.2 Manifold dimensionality and pruning

Our theoretical analysis suggests also that data with higher intrinsic dimensionality will lead to networks that are less robust to pruning. We test this hypothesis on the convolutional CIFAR architecture. We upper bound the intrinsic dimensionality of the data manifold by applying PCA for different number of latent dimensions, and then train different classification networks. Then we pruned using hard thresholding only the first convolutional layer and computed the accuracy over the test set. We average the results over 10 different networks for each upper bound. Denoting the dimensionality upper bound by we see that for the drop in accuracy is significantly larger compared to . The degradation is not exactly exponential as our dimensionality factor suggests, however this is expected as our theoretical bound involves also other quantities such as the margin and the spectral norm of different layers which change for each trained network and will also affect the resulting accuracy.

7 Conclusion

We have seen that hard thresholding followed by retraining remains the most efficient way of pruning fully connected DNN layers. For the case without retraining we have introduced a novel algorithm called FeTa that is often orders of magnitude faster than competing approaches while maintaining network accuracy. We have also shown theoretically and empirically that it is more profitable to prune layers close to the output as the perturbation that is introduced results in a much smaller drop in accuracy compared to pruning earlier layers. We discover also that networks trained on data that lies on a manifold of high intrinsic dimensionality exhibit significantly reduced robustness to pruning.

Appendix

\@fb@secFB

A. Proof of theorem 5.1.

We will proceed as follows. We first introduce some prior results which hold for the general class of robust classifiers. We will then give specific prior generalization error results for the case of classifiers operating on datapoints from -regular manifolds. Afterwards we will provide prior results for the specific case of DNN classifiers. Finally we will prove our novel generalization error bound and provide a link with prior bounds.

We first formalize robustness for generic classifiers . In the following we assume a loss function that is positive and bounded .

Definition 7.1.

An algorithm is robust if can be partitioned into K disjoint sets, denoted by , such that , ,

(16)

Now let and denote the expected error and the training error, i.e,

(17)

we can then state the following theorem from [Xu and Mannor, 2012]:

Theorem 7.1.

If consists of i.i.d. samples, and is -robust, then for any , with probability at least ,

(18)

The above generic bound can be specified for the case of -regular manifolds as in [Sokolic et al., 2017]. We recall the definition of the sample margin as well as the following theorem:

Theorem 7.2.

If there exists such that

(19)

then the classifier is -robust.

By direct substitution of the above result and the definiton of a -regular manifold into Theorem 0.1 we get:

Corollary 7.2.1.

Assume that is a (subset of) regular dimensional manifold, where . Assume also that classifier achieves a classification margin and take to be the loss. Then for any , with probability at least ,

(20)

Note that in the above we have used the fact that and therefore . The above holds for a wide range of algorithms that includes as an example SVMs. We are now ready to specify the above bound for the case of DNNs, adapted from [Sokolic et al., 2017],

Theorem 7.3.

Assume that a DNN classifier , as defined in equation 8, and let be the training sample with the smallest score . Then the classification margin is bounded as

(21)

We now prove our main result. We want to relate the generalization error of an original unpruned classifier to that of a new pruned classifier at layer . Our analysis will be based on the maximum norm error of the layer over the training set, after pruning. In the notation we have omitted the layer weights and introduced instead a numbered superscript and denoting unpruned and pruned layers respectively. We will denote by the training sample with the smallest score. For this training sample we will denote the second best guess of the classifier . Throughout the proof, we will use the notation .

First we assume the score of the point for the original classifier . Then, for the second classifier , we take a point that lies on the decision boundary between and such that . We assume for simplicity that, after pruning, the classification decisions do not change such that . We then make the following calculations

(22)

. From the above we can therefore write

(23)

By following the derivation of the margin from the original paper [Sokolic et al., 2017] and taking into account the definition of the margin we know that

(24)

Therefore we can finally write

(25)

The theorem follows from direct application of Corollary 0.2.1.

\@fb@secFB

B. Proof of theorem 5.2.

We start as in theorem 3.2 but instead of dealing with the maximum error over the training set we assume that pruning induces a perturbation matrix to each layer . We will then use the following stability result from [Neyshabur et al., 2017].

Theorem 7.4.

[Neyshabur et al., 2017](Perturbation Bound). For any , let be a layer neural network with ReLU activations. Then for any , and , and any perturbation such that , the change in the output of the network can be bounded as follows:

(26)

We can directly apply the above result to our case by assuming that layer pruning results in perturbation matrices per layer .

We assume the score of the point for the original classifier . Then, for the second classifier , we take a point that lies on the decision boundary between and such that . We assume as before that the classification decisions do not change such that . We write

(27)

We can then write

(28)

Then as before

(29)

The theorem follows from direct application of Corollary 0.2.1.

\@fb@secFB

C. Experiment Details

All architectures were trained using SGD with momentum , learning rate of , exponential decay and minibatch size of . The architectures where trained until the validation accuracy saturated usually for epochs.

We now give some details regarding the hyperparameters of the pruning algorithms. For all algorithms the entire training set was used to perform pruning.

For Corenet algorithm we implemented neuron pruning (Corenet+) but did not implement amplification (Corenet++). Amplification increased linearly the computation time of the algorithm and we did not notice any improvements in accuracy for high sparsity levels.

For Feta the algorithm was ran for to outer loups. The acceleration parameter was set to , the SVRG learning rate was set to with minibatch size of . In SVRG we need to set a number of full gradient computations denoted by the parameter we set this to .

For Net-Trim we ran the algorithm for 400 iterations and . We set the sparsity level by modifying the parameter as detailed in the authors instructions.

The LOBS argorithm does not contain any tunable parameters.

We did not implement GPU acceleration for any of the algorithms and it is not clear that this will provide speedups for common matrix-matrix and matrix-vector operations which are at the core of the implemented pruning algorithms.

References

  • [Aghasi et al., 2018] Aghasi, A., Abdi, A., and Romberg, J. (2018). Fast convex pruning of deep neural networks. arXiv preprint arXiv:1806.06457.
  • [Aghasi et al., 2016] Aghasi, A., Nguyen, N., and Romberg, J. (2016). Net-trim: A layer-wise convex pruning of deep neural networks. arXiv preprint arXiv:1611.05162.
  • [Arora et al., 2018] Arora, S., Ge, R., Neyshabur, B., and Zhang, Y. (2018). Stronger generalization bounds for deep nets via a compression approach. arXiv preprint arXiv:1802.05296.
  • [Bartlett et al., 2017] Bartlett, P., Foster, D. J., and Telgarsky, M. (2017). Spectrally-normalized margin bounds for neural networks. arXiv preprint arXiv:1706.08498.
  • [Baykal et al., 2018] Baykal, C., Liebenwein, L., Gilitschenski, I., Feldman, D., and Rus, D. (2018). Data-dependent coresets for compressing neural networks with applications to generalization bounds. arXiv preprint arXiv:1804.05345.
  • [Blundell et al., 2015] Blundell, C., Cornebise, J., Kavukcuoglu, K., and Wierstra, D. (2015). Weight uncertainty in neural networks. arXiv preprint arXiv:1505.05424.
  • [Chen et al., 2015] Chen, W., Wilson, J., Tyree, S., Weinberger, K., and Chen, Y. (2015). Compressing neural networks with the hashing trick. In International Conference on Machine Learning, pages 2285–2294.
  • [Courbariaux et al., 2016] Courbariaux, M., Hubara, I., Soudry, D., El-Yaniv, R., and Bengio, Y. (2016). Binarized neural networks: Training deep neural networks with weights and activations constrained to+ 1 or-1. arXiv preprint arXiv:1602.02830.
  • [Dai et al., 2018] Dai, B., Zhu, C., and Wipf, D. (2018). Compressing neural networks using the variational information bottleneck. arXiv preprint arXiv:1802.10399.
  • [Dong et al., 2017] Dong, X., Chen, S., and Pan, S. J. (2017). Learning to prune deep neural networks via layer-wise optimal brain surgeon. arXiv preprint arXiv:1705.07565.
  • [Fawzi et al., 2015] Fawzi, A., Davies, M., and Frossard, P. (2015). Dictionary learning for fast classification based on soft-thresholding. International Journal of Computer Vision, 114(2-3):306–321.
  • [Golowich et al., 2018] Golowich, N., Rakhlin, A., and Shamir, O. (2018). Size-independent sample complexity of neural networks. In Conference On Learning Theory, pages 297–299.
  • [Han et al., 2015a] Han, S., Mao, H., and Dally, W. J. (2015a). Deep compression: Compressing deep neural networks with pruning, trained quantization and huffman coding. arXiv preprint arXiv:1510.00149.
  • [Han et al., 2015b] Han, S., Pool, J., Tran, J., and Dally, W. (2015b). Learning both weights and connections for efficient neural network. In Advances in Neural Information Processing Systems, pages 1135–1143.
  • [Hou et al., 2016] Hou, L., Yao, Q., and Kwok, J. T. (2016). Loss-aware binarization of deep networks. arXiv preprint arXiv:1611.01600.
  • [Kim et al., 2015] Kim, Y.-D., Park, E., Yoo, S., Choi, T., Yang, L., and Shin, D. (2015). Compression of deep convolutional neural networks for fast and low power mobile applications. arXiv preprint arXiv:1511.06530.
  • [Krizhevsky and Hinton, 2009] Krizhevsky, A. and Hinton, G. (2009). Learning multiple layers of features from tiny images. Technical report, Citeseer.
  • [LeCun et al., 2015] LeCun, Y., Bengio, Y., and Hinton, G. (2015). Deep learning. Nature, 521(7553):436–444.
  • [LeCun et al., 1998] LeCun, Y., Bottou, L., Bengio, Y., and Haffner, P. (1998). Gradient-based learning applied to document recognition. Proceedings of the IEEE, 86(11):2278–2324.
  • [Louizos et al., 2017] Louizos, C., Ullrich, K., and Welling, M. (2017). Bayesian compression for deep learning. arXiv preprint arXiv:1705.08665.
  • [Molchanov et al., 2017] Molchanov, D., Ashukha, A., and Vetrov, D. (2017). Variational dropout sparsifies deep neural networks. arXiv preprint arXiv:1701.05369.
  • [Neyshabur et al., 2017] Neyshabur, B., Bhojanapalli, S., McAllester, D., and Srebro, N. (2017). A pac-bayesian approach to spectrally-normalized margin bounds for neural networks. arXiv preprint arXiv:1707.09564.
  • [Nitanda, 2014] Nitanda, A. (2014). Stochastic proximal gradient descent with acceleration techniques. In Advances in Neural Information Processing Systems, pages 1574–1582.
  • [Novikov et al., 2015] Novikov, A., Podoprikhin, D., Osokin, A., and Vetrov, D. P. (2015). Tensorizing neural networks. In Advances in Neural Information Processing Systems, pages 442–450.
  • [Raghu et al., 2016] Raghu, M., Poole, B., Kleinberg, J., Ganguli, S., and Sohl-Dickstein, J. (2016). On the expressive power of deep neural networks. arXiv preprint arXiv:1606.05336.
  • [Sokolic et al., 2017] Sokolic, J., Giryes, R., Sapiro, G., and Rodrigues, M. R. (2017). Robust large margin deep neural networks. IEEE Transactions on Signal Processing.
  • [Tao and An, 1997] Tao, P. D. and An, L. T. H. (1997). Convex analysis approach to dc programming: Theory, algorithms and applications. Acta Mathematica Vietnamica, 22(1):289–355.
  • [Xiao et al., 2017] Xiao, H., Rasul, K., and Vollgraf, R. (2017). Fashion-mnist: a novel image dataset for benchmarking machine learning algorithms. arXiv preprint arXiv:1708.07747.
  • [Xu and Mannor, 2012] Xu, H. and Mannor, S. (2012). Robustness and generalization. Machine learning, 86(3):391–423.
  • [Yang et al., 2015] Yang, Z., Moczulski, M., Denil, M., de Freitas, N., Smola, A., Song, L., and Wang, Z. (2015). Deep fried convnets. In Proceedings of the IEEE International Conference on Computer Vision, pages 1476–1483.
Comments 0
Request Comment
You are adding the first comment!
How to quickly get a good reply:
  • Give credit where it’s due by listing out the positive aspects of a paper before getting into which changes should be made.
  • Be specific in your critique, and provide supporting evidence with appropriate references to substantiate general statements.
  • Your comment should inspire ideas to flow and help the author improves the paper.

The better we are at sharing our knowledge with each other, the faster we move forward.
""
The feedback must be of minimum 40 characters and the title a minimum of 5 characters
   
Add comment
Cancel
Loading ...
366101
This is a comment super asjknd jkasnjk adsnkj
Upvote
Downvote
""
The feedback must be of minumum 40 characters
The feedback must be of minumum 40 characters
Submit
Cancel

You are asking your first question!
How to quickly get a good answer:
  • Keep your question short and to the point
  • Check for grammar or spelling errors.
  • Phrase it like a question
Test
Test description