L AYER - ADAPTIVE SPARSITY FOR THE M AGNITUDE - BASED P RUNING
A BSTRACT
Recent discoveries on neural network pruning reveal that, with a carefully chosen layerwise sparsity a simple magnitude-based pruning achieves state-of-the-art tradeoff between sparsity and performance.
However, without a clear consensus on “how to choose the layerwise sparsities are mostly selected algorithm- by algorithm, often resorting to handcrafted heuristics or an extensive hyperparameter search
To ﬁll this gap, we propose a novel importance score for global pruning coined layer -adaptive magnitude based pruning LAMP score; the score is a rescaled version of weight magnitude that incorporates the model level (cid:96) distortion incurred by pruning , and does not require any hyperparameter tuning or heavy computation
Under various image classiﬁcation setups , LAMP consistently outper forms popular existing schemes for layerwise sparsity selection.
Furthermore we observe that LAMP continues to outperform baselines even in weight rewinding setups while the connectivity-oriented layerwise sparsity (the strongest baseline overall) performs worse than a simple global magnitude-based pruning in this case
Code: https://github.com/jaeho-lee/layer-adaptive-sparsity
I NTRODUCTION
Neural network pruning is an art of removing “unimportant weights” from a model, with an intention to meet practical constraints (Han et al., 2015), mitigate overﬁtting (Hanson & Pratt, 1988), enhance interpretability (Mozer & Smolensky, 1988), or deepen our understanding on neural network training (Frankle & Carbin, 2019).
Yet, the importance of weight is still a vaguely deﬁned notion, and thus a wide range of pruning algorithms based on various importance scores has been proposed.
One popular approach is to estimate the loss increment from removing the target weight to use as an importance score, e.g., Hessian-based approximations (LeCun et al., 1989; Hassibi & Stork, 1993;
Dong et al., 2017), coreset-based estimates (Baykal et al., 2019; Mussay et al., 2020), convex optimization (Aghasi et al., 2017), and operator distortion (Park et al., 2020).
Other approaches include on-the-ﬂyregularization (Louizos et al., 2018; Xiao et al., 2019), Bayesian methods (Molchanov et al., 2017;
Louizos et al., 2017; Dai et al., 2018), and reinforcement learning (Lin et al., 2017).
Recent discoveries (Gale et al., 2019; Evci et al., 2020) demonstrate that, given an appropriate choice of layerwise sparsity , simply pruning on the basis of weight magnitude yields a surprisingly powerful unstructured pruning scheme.
For instance, Gale et al.
(2019) evaluates the performance of magnitudebased pruning (MP; Han et al. (2015); Zhu & Gupta (2018))
with an extensive hyperparameter tuning, and shows that MP achieves comparable or better performance than state-of-the-art pruning algorithms that use more complicated importance scores.
To arrive at such a performance level, the authors introduce the following handcrafted heuristic: Leave the ﬁrst convolutional layer fully dense, and prune up to only 80% of weights from the last fully-connected layer; the heuristic is motivated by the sparsity pattern from other state-of-the-art algorithms (Molchanov et al., 2017) and additional experimental/architectural observations.
Unfortunately, there is an apparent lack of consensus on “how to choose the layerwise sparsity” for the magnitude-based pruning.
Instead, the layerwise sparsity is selected mostly on an algorithm-byalgorithm basis.
One common method is the global MP criteria (see, e g., Morcos et al.
(2019)), where the layerwise sparsity is automatically determined by using a single global threshold on weight magnitude.
Lin et al.
(2020) propose a magnitude-based pruning algorithm using a feedback signal, using a heuristic rule of keeping the last fully connected layer dense.
A recent work by Evci et al.
(2020) proposes a magnitude-based dynamic sparse training method, adopting layerwise sparsity inspired from the network science approach toward neural network pruning (Mocanu et al., 2018).
Contributions.
In search of a “go-to” layerwise sparsity for MP, we take a model-level distortion minimization perspective towards MP.
We build on the observation of Dong et al.
; Park et al.
(2020) that each neural network layer can be viewed as an operator, and MP is a choice that incurs minimum (cid:96) 2 distortion to the operator output (given a worst-case input signal).
We bring the perspective further to examine the “model-level” distortion incurred by pruning a layer; preceding layers scale the input signal to the target layer, and succeeding layers scale the output distortion.
Based on the distortion minimization framework, we propose a novel importance score for global pruning, coined LAMP (Layer-Adaptive Magnitude-based Pruning).
The LAMP score is a rescaled weight magnitude, approximating the model-level distortion from pruning.
Importantly, the LAMP score is designed to approximate the distortion on the model being pruned , i.e., all connections with a smaller LAMP score than the target weight is already pruned.
Global pruning 2 with the LAMP score is equivalent to the MP with an automatically determined layerwise sparsity.
At the same time, pruning with LAMP keeps the beneﬁts of MP intact; the LAMP score is efﬁciently computable, hyperparameter-free, and does not rely on any model-speciﬁc knowledge.
We validate the effectiveness of LAMP under a diverse experimental setup, encompassing various convolutional neural network architectures (VGG-16, ResNet-18/34, DenseNet-121, EfﬁcientNet-B0) and various image datasets (CIFAR-10/100, SVHN, Restricted ImageNet).
In all considered setups,
LAMP consistently outperforms the baseline layerwise sparsity selection schemes.
We also perform additional ablation studies with one-shot pruning and weight-rewinding setup to conﬁrm that LAMP performs reliably well under a wider range of scenarios.
Organization.
In Section 2, we brieﬂy describe existing methods to choose the layerwise sparsity for magnitude-based pruning.
In Section 3, we formally introduce LAMP and describe how the (cid:96)distortion minimization perspective motivates the LAMP score.
In Section 4, we empirically validate the effectiveness and versatility of LAMP.
In Section 5, we take a closer look at the layerwise sparsity discovered by LAMP and compare with baseline methods and previously proposed handcrafted heuristics.
In Section 6, we summarize our ﬁndings and discuss future directions.
Appendices include the experimental details (Appendix A), complexity analysis (Appendix B), derivation of the LAMP score (Appendix C), additional experiments on Transformer (Appendix D), and detailed experimental results with standard deviations (Appendix E).
R ELATED WORK
This section gives a (necessarily non-exhaustive) survey of various layerwise sparsity selection schemes used for magnitude-based pruning algorithms.
Magnitude-based pruning of neural networks dates back to the early works of Janowsky (1989); LeCun et al.
and has been actively studied again under the context of model compression since the work of Han et al.
In Han et al.
, the authors propose an iterative pruning scheme where the layerwise pruning threshold is determined by the standard-deviation-based heuristic.
Zhu & Gupta (2018) propose a uniform pruning algorithm with a carefully tuned gradual pruning schedule combined with weight re-growths.
Gale et al.
(2019) reﬁne the algorithm by adding a heuristic constraint of keeping the ﬁrst convolutional layer fully dense and keeping at least 20% of the weights surviving in the last fully-connected layer.
MP has also been widely used in the context of “pruning at initialization.”
Frankle & Carbin (2019) combine MP with weight rewinding to discover efﬁciently trainable subnetworks: for small nets, the authors employ uniform layerwise sparsity, but use different rates for convolutional layers and fully-connected layers (with an added heuristic on the last fully-connected layer); for larger nets, authors use global MP.
Morcos et al.
(2019) consider transferring the “winning ticket” initializations, using the global MP.
Evci et al.
(2020) proposes a training scheme for sparsely initialized neural networks, where the layerwise sparsity is given by the Erd˝os-R´enyi kernel method; the method generalizes the scheme initially proposed by Mocanu et al.
(2018) to convolutional neural networks.
We note that there is a line of results on the trainable layerwise sparsity ; we refer the interested readers to the recent work of Kusupati et al.
(2020) for a concise survey.
However, we do not make direct comparisons to these methods, as our primary purpose is to deliver an easy-to-use layerwise sparsity selection scheme without requiring the modiﬁcation of training objective, or an extensive hyperparameter tuning.
We also note that we focus on the unstructured sparsity .
While such unstructured pruning techniques have been considered less practical (compared to structured pruning), several recent breakthroughs provide promising methods to bridge this gap; see Gale et al.
; Elsen et al.
L AYER - ADAPTIVE MAGNITUDE - BASED PRUNING (LAMP)
We now formally introduce the Layer-Adaptive Magnitude-based Pruning (LAMP) score.
Consider a depth- d feedforward neural network with weight tensors W (1) , . .
, W ( d ) associated with each fully-connected/convolutional layer.
For fully-connected layers, corresponding weight tensors are twodimensional matrices, and for 2d convolutional layers, corresponding tensors are four-dimensional.
To give a uniﬁed deﬁnition of the LAMP score for both fully-connected and convolutional layers, we assume that each weight tensor is unrolled (or ﬂattened) to a one-dimensional vector.
For each of these unrolled vectors, we assume (without loss of generality) that the weights are sorted in an ascending order according to the given index map, i.e., | W [ u ] | ≤ | W [ v ] | holds whenever u < v , where W [ u ] denote the entry of W mapped by the index u .
The LAMP score for the u -th index of the weight tensor W is then deﬁned as
[Equation]
Informally, the LAMP score (Eq. 1) measures the relative importance of the target connection among all surviving connections belonging to the same layer, where the connections with a smaller weight magnitude (in the same layer) have already been pruned.
As a consequence, two connections with identical weight magnitudes have different LAMP scores, depending on the index map being used.
Once the LAMP score is computed, we globally prune the connections with smallest LAMP scores until the desired global sparsity constraint is met; the procedure is equivalent to performing MP with an automatically selected layerwise sparsity.
To see this, it sufﬁces to check that
[Equation]
holds for any weight tensor W and a pair of indices u, v .
From the deﬁnition of the LAMP score (Eq. 1), it is easy to see that the logical relation (2) holds.
Indeed, for the connection with a larger weight magnitude, the denominator of Eq.
1 is smaller, while the numerator is larger.
We note that the global pruning with respect to the LAMP score is not identical to the global pruning with respect to the magnitude score | W [ u ] | (or ( W [ u ]) 2 , equivalently).
Indeed, in each layer, there exists exactly one connection with the LAMP score of 1 , which is the maximum LAMP score possible.
In other words, LAMP keeps at least one surviving connection in each layer.
The same does not hold for the global pruning with respect to the weight magnitude score.
We also note that the LAMP score is easy-to-use.
Similar to the vanilla MP, the LAMP score does not have any hyperparameter to be tuned, and is easily implementable via elementary tensor operations.
Furthermore, the LAMP score can be computed with only a minimal computational overhead; the sorting of squared weight magnitudes required to compute the denominator in Eq.
1 is already a part of typical vanilla MP algorithms.
For more discussions, see Appendix B.
3. D ESIGN MOTIVATION : M INIMIZING OUTPUT (cid:96) 2 DISTORTION
The LAMP score (Eq. 1) is motivated by the following observation: The layerwise MP is the solution of the layerwise minimization of Frobenius distortion incurred by pruning, which can be viewed as a relaxation of the output (cid:96) 2 distortion minimization with respect to the worst-case input.
This observation leads us to the speculation “Reducing the pruning-incurred (cid:96) 2 distortion of the model output with respect to the worst-case output may be beneﬁcial to the performance of the retrained model (and perhaps that is why MP works well in practice).”
This speculation is not entirely new; the optimal brain damage (OBD; (LeCun et al., 1989)) is also designed around a similar philosophy of loss minimization, without a complete understanding on how the beneﬁt of loss minimization seems to pertain after retraining .
Nevertheless, we use this speculation as a guideline to design LAMP as a natural extension of layerwise MP to a global pruning scheme with an automatically determined layerwise sparsity.
To make arguments a bit more formal, consider a depth- d fully-connected 4 neural net, whose output given the input x is
[Equation]
where σ denotes the ReLU activation and W i denotes the weight matrix for the i -th layer, and
W (1: d ) =
( W (1) , . . .
, W ( d ) ) denotes the set of weight matrices.
Viewing MP as a relaxed layerwise (cid:96) 2 distortion minimization.
We ﬁrst focus on a single fullyconnected layer (instead of a full model), and consider the problem of minimizing the pruning-incurred (cid:96) 2 distortion in the layer output, given the worst-case input signal.
We then observe that the problem can be relaxed to the minimization of Frobenius distortion in the weight tensor, whose solution coincides with the layerwise MP.
Formally, let ξ ∈ R n be an input vector to a fully-connected layer with the weight tensor W ∈ R m × n .
We want to prune the tensor to (cid:102) W :=
M (cid:12) W , where M is an m × n binary matrix (i.e., having only 0 s and 1 s as its entries) satisfying some predeﬁned sparsity constraint (cid:107) M (cid:107) 0 ≤ κ imposed by the operational constraints (e.g., model size requirements).
We wish to ﬁnd the pruning mask M that incurs the minimum (cid:96) 2 distortion in the output given the worst-case (cid:96) 2 -bounded input, i.e.,
[Equation]
The minimax distortion (4) upper-bounds the minimum expected (cid:96) 2 distortion for any distribution of ξ supported on the unit ball, and thus can be viewed as a data-oblivious version of the pruning algorithms designed for loss minimization (using squared loss).
By the deﬁnition of the spectral norm, 5 Eq. 4 is equivalent to
[Equation]
(5) where (cid:107) ·
(cid:107) denotes the spectral norm.
Using the fact that (cid:107) A (cid:107) ≤ (cid:107) A (cid:107) F holds for any matrix A(where (cid:107) · (cid:107) F denotes the Frobenius norm), the optimization (5) can be relaxed to the Frobenius
The inequality is a consequence of the Cauchy-Schwarz inequality: (cid:107) Ax (cid:107)
=
(cid:80) i ( (cid:80)
A
≤
[Table]
[Equation]
where W ij , M ij denote ( i, j ) -th entries of W, M , respectively.
From the right-hand side of Eq.
6, we see that the layerwise MP, i.e., setting M ij = 1 for ( i, j ) pairs with top- κ largest W ij , is the optimal choice to minimize the Frobenius distortion incurred by pruning.
This observation motivates us to view the layerwise MP as the (approximate) solution of the output (cid:96) 2 distortion minimization procedure, and speculate the connection between the small output (cid:96) 2 distortion and the favorable performance of the pruned-retrained subnetwork (given the unreasonable effectiveness of seeminglyna¨ıve MP as demonstrated by Gale et al.
LAMP: greedy, relaxed minimization of model output distortion.
Building on this speculation, we now ask the following question: “How can we select the layerwise sparsity of MP to have small model-level output distortion?”
To formalize, we consider the minimization
[Equation]
where κ denotes the model-level sparsity constraint imposed by the operational requirements and (cid:102)
[Equation]
i ) denotes the pruned version of the i -th layer weight matrix.
Due to the nonlinearities from the activation functions, it is difﬁcult to solve Eq.
7 exactly.
Instead, we consider the following greedy procedure: At each step, we (a) approximate the distortion incurred by pruning a single connection , (b) remove the connection with the smallest score, and then (c) go back to step (a) and re-compute the scores based on the pruned model .
Once we assume that only one connection is pruned at a single iteration of the step (a), we can use the following upper bound of the model output distortion to relax the optimization (7): With (cid:102) W i denoting a pruned version of W i , we have
[Equation]
(see Appendix C for a derivation).
Despite the sub-optimalities from the relaxation , considering the right-hand side of Eq.
8 provides two favorable properties.
First, the right-hand side is free of any activation function, and is equivalent to the layerwise MP.
Second, the score can be computed in advance , i.e., does not require re-computing after pruning each connection.
In particular, the product term (cid:81) dj
=1 (cid:107) W ( j ) (cid:107) F does not affect the pruning decision, and the denominator can be pre-computed with the cumulative sum (cid:80) v ≥ u ( W ( i ) [ v ]) 2 for each index u for W ( i ) .
This computational trick leads us to the LAMP score (1).
E XPERIMENTS & A NALYSES
To empirically validate the effectiveness of the proposed method, we compare LAMP with following layerwise sparsity selection schemes for magnitude-based pruning:
• Global.
A global threshold on the weight magnitudes is imposed on every layer to meet the global sparsity constraint, and the layerwise sparsity is automatically determined according to this threshold; see, e.g., Morcos et al.
• Uniform.
Every layer is pruned to have identical layerwise sparsity levels, which is in turn equal to the global sparsity constraint; see, e.g., Zhu & Gupta (2018).
• Uniform+.
Same as Uniform, but we impose two additional constraints: (1) we keep the ﬁrst convolutional layer unpruned, and (2) retain at least 20% of connections alive in the last fullyconnected layer; this heuristic rule is proposed by Gale et al.
[Figure]
• Erd˝os-R´enyi kernel.
An extension of Erd˝os-R´enyi method (originally given by Mocanu et al.
accounting for convolutional layers, as proposed by Evci et al.
The numbers of nonzero parameters of sparse convolutional layers are scaled proportional to 1 − n l − 1 + n l + w l + h l n l − 1 · n l · w l · h l , where n l denotes the number of neurons at layer l , and w l , h l denotes the width and height of the l th layer convolutional kernel.
As a default setup, we perform ﬁve independent trials for each baseline method, where in each trial we use iterative pruning-and-retraining (Han et al., 2015): we prune 20% of the surviving weights at each iteration.
For the Restricted-ImageNet experiment, we provide the result from four trials.
For a clear presentation, we only report the average on the ﬁgures appearing in the main text.
Standard deviations for ﬁve-seed results in Section 4.1 will be given in Appendix E.
Also, in Appendix D, we report additional experimental results for language modeling tasks (Penn Treebank and WT-2) on
Transformers (Vaswani et al., 2017).
Summary of observations.
From the experimental results (Figs. 2 to 4), we observe that LAMP consistently outperforms all other baselines, in terms of the sparsity-accuracy tradeoff.
The performance gap between LAMP and baseline methods seems be more pronounced with modern network architectures, e.g., EfﬁcientNet-B0.
We also observe that LAMP performs well under weight rewinding setup, while the strongest baseline (Erd˝os-R´enyi kernel) seems to be sensitive to such rewinding.
4. M AIN RESULTS
Our main experimental results are on image classiﬁcation models.
We explore a diverse set of model architectures and datasets, with a base setup of VGG-16 (Simonyan & Zisserman, 2015) trained on
CIFAR-10 (Krizhevsky & Hinton, 2009) dataset.
In particular, our experiments cover the following models and datasets.
[Figure]
Models.
We consider four network architectures for image classiﬁcation experiments: (1) VGG- (Simonyan & Zisserman, 2015) adapted for CIFAR-10 to have batch normalization layers and one fully-connected layer (as used in Liu et al.
; Frankle & Carbin (2019)); (2) ResNet-20/34 (He et al., 2016); (3) DenseNet-121 (Huang et al., 2017); (4) EfﬁcientNet-B0 (Tan & Le, 2019).
For all four models, we prune the weight tensors for fully-connected and convolutional units.
Biases and batch normalization layers are kept unpruned.
Datasets.
We consider the following datasets; CIFAR-10/100 (Krizhevsky & Hinton, 2009), SVHN (Netzer et al., 2011), and Restricted ImageNet (Tsipras et al., 2019).
All datasets except for Restricted
ImageNet are used for training VGG-16; Restricted ImageNet is used for training ResNet-34.
Other details.
Detailed experimental setup is given in Appendix A.
In Fig.
2, we provide sparsity-accuracy tradeoff curves for four different model architectures trained on CIFAR-10 dataset.
The ﬁrst observation is that LAMP achieves the best tradeoff; in all four models, LAMP consistently outperforms baseline methods.
We also observe that Erd ˝ os-R ´ enyi kernel method also outperforms other baselines in VGG-16, ResNet-20, and EfﬁcientNet-B0, but fails to do so on DenseNet-121.
Furthermore, the gap between LAMP and the Erd ˝ os-R ´ enyi kernel method seems to widen as the model architecture gets more complicated; the gap between two methods is notable especially in EfﬁcientNet-B0, where mobile inverted bottleneck convolutional layers replace traditional convolutional modules.
In particular, LAMP achieves 88 .
1% test accuracy when only 1 .
44% of all weights survive, while Erd˝os-R´enyi kernel achieves 77 .
Lastly, we observe that the heuristic of Gale et al.
(2019) seems to provide an improvement over the Uniform MP.
In Fig.
3, we present the tradeoff curves for three additional datasets: SVHN, CIFAR-100, and
Restricted ImageNet.
Similar to Fig.
2, we observe that LAMP outperforms all other baselines and
Erd ˝ os-R ´ enyi kernel remains to be the most competitive baseline.
4. A BLATIONS : O NE - SHOT PRUNING , W EIGHT REWINDING , AND SNIP
Modern magnitude-based pruning algorithms are often used in combination with customized pruning schedules (e.g., Zhu & Gupta (2018)) or weight rewinding (e.g.
, Frankle & Carbin (2019); Renda et al.
As a sanity check that LAMP perform reliably well along with such techniques, we conduct following additional experiments.
• One-shot pruning.
As an extreme case of pruning schedule, we test the scheme where we only run a single training-pruning-retraining cycle, instead of iterating multiple cycles.
We test the one-shot pruning on VGG-16 trained with CIFAR-10 dataset.
• Weight rewinding.
After pruning, we rewind the remaining weights to their values in the early epoch, as in the “lottery ticket” experiments by Frankle & Carbin (2019).
We perform iterative magnitude-based pruning (IMP) on VGG-16, using the warm-up step and the training schedule described in Frankle et al.
• SNIP.
As an additional experiment, we test whether LAMP can provide a general-purpose layerwise sparsity for pruning schemes other than MP.
We test under the “pruning at initialization” setup with SNIP scores (Lee et al., 2019).
Baselines are similarly modiﬁed to use the SNIP score.
We use
Conv-6 model on CIFAR-10 dataset (see Frankle & Carbin (2019) for more details of the model) with a batch size of 128 .
[Figure]
Again, other experimental details are given at the Appendix A.
Results for all three experiments are given in Fig.
On one-shot pruning, we conﬁrm that LAMP comfortably leads other baselines, as in the iterative pruning case.
We note that the gap between one-shot pruning and iterative pruning is quite small for LAMP; when 1 . 15% of all prunable weights survive, iterative LAMP brings only 1 . 09% accuracy gain over one-shot LAMP.
By contrast, the gain from iteration for Uniform MP is 41 .
62% at the same sparsity level.
On the weight-rewinding experiment, we observe that LAMP remains beneﬁcial over the baseline methods.
We also remark that the Global baseline tend to perform well in this case, even outperforming the Erd ˝ os-R ´ enyi kernel method in the low-sparsity regime.
This phenomenon seems to be connected to the observation of Zhou et al.
that the initial weights and ﬁnal weights of a model are highly correlated; global MP may help preserving connections with larger initial magnitudes, which play important roles in terms of signal propagation at initialization (Lee et al., 2020).
On the SNIP experiment, we observe that LAMP achieves a similar performance to the Global SNIP.
Recalling that the SNIP score is designed for global pruning (Lee et al., 2019), such high performance of LAMP is unexpected.
We suspect that this is because LAMP is also designed for “output distortion minimization,” which shares a similar spirit with the “gradient distortion minimization.”
L AYERWISE S PARSITY : G LOBAL MP, E RD ˝ OS -R ´ ENYI KERNEL , AND LAMP
With the effectiveness of LAMP conﬁrmed, we take a closer look at the layerwise sparsity discovered by LAMP.
We focus on answering two questions: Q1.
Does layerwise sparsity distilled from LAMP behave similarly to the heuristics constructed from experiences, e.g. the one given by Gale et al.
Q2.
Is there any other deﬁning characteristic of LAMP sparsity patterns, which may help us guide the design of (sparse) network architectures?
In Fig.
5, we plot the layerwise survival rates and the number of nonzero weights discovered by iteratively pruning VGG-16 (trained on CIFAR-10), by Global MP, Erd˝os-R´enyi kernel, and LAMP.
Layerwise survival rates are given for the global survival rates of { 51 . 2% , 26 .
52% } (from lighter to darker).
Number of nonzero weights are plotted for the pruned models with total { 3 . 52% , 1 . 80% , 0 .
24% } fraction of all weights surviving.
We observe that LAMP sparsities share a similar tendency to sparsity levels given by the Erd˝os-R´enyi kernel method.
In particular, both methods tend to keep the ﬁrst and layer layers of the neural network relatively dense; this property is reminiscent of the handcrafted heuristic given in Gale et al.
(2019): keep the ﬁrst convolutional layer unpruned, and prune at most 80% from the last fully-connected layer.
While Global MP also keeps a large fraction of the last fully-connected layer unpruned, the ﬁrst convolutional layer gets pruned quickly.
LAMP sparsities differ from Erd˝os-R´enyi kernel sparsities in two senses.
• Although LAMP demonstrates its tendency to keep the ﬁrst and the last layers relatively unpruned, this tendency is softer.
When 3 .
52% of weights survive, LAMP keeps ∼ 79% and ∼ 62% of weights unpruned from the ﬁrst and last layers (respectively), while Erd˝os-R´enyi kernel does not prune
any weight from those two layers.
[Figure]
• LAMP tends to keep the number of nonzero weights relatively uniform throughout the layers at extreme sparsity levels (indeed, the ﬁrst observation can be understood as a consequence of the second observation).
In contrast, Erd ˝ os-R ´ enyi kernel method keeps the relative ratio constant, regardless of the global sparsity level.
Following the second observation, we conjecture that having a similar number of nonzero connections in each layer may be an essential condition to guarantee maximal memory capacity (see, e.g., Yun et al.
given a global sparsity constraint on the neural network.
Theoretical investigations of the approximability by such sparse neural networks may be an interesting future research direction, potentially leading to a more principled and robust pruning algorithms.
As an additional remark, we note that the layerwise sparsity discovered by LAMP behaves similarly to that of AMC (He et al., 2018), which uses a reinforcement learning agent to search over the space of all available layerwise sparsity.
We provide further details in Appendix F.
C ONCLUSION
In this paper, we investigate an under-explored problem on the layerwise sparsity for magnitude-based pruning scheme.
The proposed method, coined LAMP (Layer-Adaptive Magnitude-based Pruning), is motivated from the (cid:96) 2 distortion minimization perspective on magnitude-based pruning, and provides a consistent performance gain on a wide range of models and datasets.
Furthermore, LAMP performs reliably well when combined with one-shot pruning schedule or weight rewinding, which makes it an attractive candidate as a “go-to” layerwise sparsity for the magnitude-based pruning.
Taking a deeper look at LAMP-discovered layerwise sparsities, we observe that LAMP automatically recovers the handcrafted rules for the layerwise sparsity.
Furthermore, we observe that LAMP tend to keep the number of nonzero weights relatively uniform throughout the layers, as we consider more extreme sparsity levels.
We conjecture that such uniformity in the number of unpruned weights may be an essential condition for a maximal expressive power of sparsiﬁed neural networks.
A CKNOWLEDGMENTS
This work was supported in part by Institute of Information & Communications Technology Planning & Evaluation (IITP) grant funded by the Korea government (MSIT) (No.2019-0-00075, Artiﬁcial
Intelligence Graduate School Program (KAIST))
and in part by Samsung Advanced Institute of
Technology (SAIT).
A E XPERIMENTAL S ETUPS
For any implementational details not given in this section, we refer to the code at: https://github.com/jaeho-lee/layer-adaptive-sparsity
Optimizer.
With an exception of the weight rewinding experiment, we use AdamW (Loshchilov &
Hutter, 2019) with learning rate 0 .
0003 ; we use vanilla Adam with learning rate 0 .
0003 for the weight rewinding experiment, following the setup of Frankle & Carbin (2019).
For other hyperparameters, we follow the PyTorch default setup: (cid:126)β = (0 . 9 , 0 . 999) , wd = 0 .
01 , ε = 10 − 8 .
Pre-processing.
CIFAR-10/100 dataset is augmented with random crops with a padding of 4 and random horizontal ﬂips.
We normalize both training and test datasets with constants (0 . 4914 , 0 .
SVHN training dataset is augmented with random crops with a padding of 2 .
We normalize both training and test datasets with constants (0 . 4377 , 0 . 4438 , 0 . 4728) , (0 . 198 , 0 . 201 , 0 . 197) .
Restricted ImageNet training dataset is augmented with Random resized crop to size 224 and random horizontal ﬂips.
Restricted ImageNet test dataset is resized to 256 and then center-cropped to size 224 .
We normalize both training and test datasets with constants (0 . 485 , 0 .
Models.
Some of the models used for CIFAR-10 datasets are originally developed for ImageNet:
VGG-16, DenseNet-121, EfﬁcientNet-B0.
The models are adapted for CIFAR-10/100 datasets by modifying only the average pooling and the (ﬁrst) fully-connected layer (and not convolutional layers) to ﬁt the 32 × 32 resolution of the input image.
[Table]
B C OMPUTATIONAL AND I MPLEMENTATIONAL A SPECTS OF LAMP
In this section, we discuss the computational complexity of LAMP and describe how LAMP can be implemented.
We consider a depth- d neural network, with n i number of connections in i -th layer.
We also let n = (cid:80) di
=1 n i .
The global MP is comprised of two steps:
• Step 1.
Pool and sort the weight magnitudes of all connections, taking O ( n log n ) operations.
• Step 2.
Assign 0 s to connections with smallest weights until the desired sparsity level is achieved, taking O ( n ) computations.
The total computational cost is of order O ( n log n ) .
Similarly, we see that performing MP with any pre-determined sparsity levels incurs O ( (cid:80) di =1 n i log n i ) computations.
LAMP, on the other hand, can be done in four steps.
• Step 1.
Weight magnitudes are squared and sorted for each layer, which can be done with a computation cost of O ( (cid:80) di =1 n i log n i ) .
• Step 2.
LAMP score denominators (cid:80) v ≥ u W 2 i [ v ] for each layer is computed by summing-andstoring the squared weight magnitudes according to a descending order; takes O ( n ) computations.
• Step 3.
The LAMP score is computed by dividing squared weight magnitudes by the denominators, using O ( n ) steps.
• Step 4.
We sort and prune as in global MP, taking O ( n log n ) steps.
We observe that step 4 is the dominant term, which is shared by the global MP.
The ﬁrst three steps can be easily implemented in PyTorch as follows.
[Table]
C D ERIVATION OF INEQUALITY (8)
In this section
[Equation]
we prove the following inequality.
[Equation]
This inequality is a simpliﬁed and modiﬁed version of what is popularly known as “peeling” procedure in the generalization literature (e
[Equation]
g., (Neyshabur et al., 2015))
[Equation]
and we present the proof only for the sake of completeness.
We begin by peeling the outermost layer as
[Equation]
where we have used Cauchy-Schwarz inequality for the ﬁrst inequality
[Equation]
and the 1 -Lipschitzness of
ReLU activation with respect to (cid:96) 2 norm.
We keep on peeling until we get
[Equation]
The second multiplicative term on the right-hand side requires a slightly different treatment.
Via
Cauchy-Schwarz, we get
[Equation]
The activation functions from this point require zero-in-zero-out (ZIZO; σ ( 0 ) = 0 ) property to be peeled, in addition to the Lipschitzness.
Indeed, we can proceed as
[Equation]
Iteratively applying the Cauchy-Schwarz inequality and the step Eq.
(19), we arrive at Eq.
D E XPERIMENTAL RESULTS ON LANGUAGE MODELING
In this section, we apply the proposed LAMP and baseline layerwise sparsities on language modeling tasks.
We note, however, that it is still unclear whether magnitude-based pruning approaches achieve state-of-the-art results for pruning natural language processing tasks (see, e.g., Sanh et al.
(2020), for more discussions).
In particular, NLP model architectures (e.g., embedding layers) and learning pipelines (e.g., more prevalent use of pre-trained models) add more complications to model pruning, obscuring how pruning algorithms should be fairly compared.
We also note that we do not test
Uniform+ baseline for this task, as the heuristic is proposed explicitly for image classiﬁcation models, not language models (Gale et al., 2019).
Model.
We consider the simpliﬁed version of Transformer-XL (Dai et al., 2018) architecture, which has six self-attention layers (original has 16) and removed some regularizing heuristics, such as exponential moving average and cosine annealing.
Similar to the setting of Sanh et al.
(2020), we focus on pruning self-attention layers of the model (approximately 7 .
57 M parameters).
Optimizer.
We use AdamW (Loshchilov & Hutter, 2019) with learning rate 0.0003, following the
PyTorch default setup for other hyperparameters: (cid:126)β = (0 . 9 , 0 . 999) , wd = 0 .
01 , ε = 10 − 8 .
We use batch size 20 and maximum sequence length 70.
We train the initial unpruned model for 200, iterations and re-train the model after pruning for 50,000 iterations.
Datasets.
We use Penn Treebank (Marcus et al., 1993) and WikiText-2 (Merity et al., 2016) datasets.
We provide the experimental results in Tables 2 and 3.
We observe that LAMP consistently achieves the near-best performance among all considered methods.
We note, however, that the gain is marginal.
We suspect that this marginal-ity is due to the fact that widths of the Transformer layers are relatively uniform compared to image classiﬁcation models.
We do not observe any notable pattern among the baseline methods.
[Table] [Table]
[Figure]
[Table]
[Figure]
E D ETAILED EXPERIMENTAL RESULTS ( WITH STANDARD DEVIATIONS )
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Figure]
F P EAKS AND CRESTS OF LAMP SPARSITY
In this section, we compare the layerwise sparsity discovered by LAMP (and other non-uniform baselines) to the sparsity discovered by AMC (He et al., 2018), which uses a reinforcement learning agent to explicitly search for the optimal layerwise sparsity.
In particular, we focus on whether the “peaks and crests” phenomenon observed for AMC-discovered layerwise sparsity also takes place in the layerwise sparsity induced by LAMP:
He et al.
(2018) observe that the layerwise sparsity of ResNet-50 (trained on ImageNet) and pruned by AMC exhibits what they call peaks and crests, i.e., the sparsity oscillates between high and low periodically; see Figure 3 of He et al.
The authors speculate that such phenomenon takes place, because AMC automatically learns that 3 ×convolution has more redundancy than 1 × 1 convolution and can be pruned more .
To see if LAMP also automatically discovers such “peaks and crests,” we prune the ImageNetpretrained ResNet-50 model with one-shot LAMP, global MP, and Erd¨os-R´enyi kernel.
The layerwise survival ratios are reported in Fig.
From the ﬁgure, we observe that LAMP also discovers different sparsity ratio for 1 × 1 convolution and 3 × 3 convolution layers; 3 × 3 convolutional ﬁlters are pruned more.
Such pattern is more noticeable in the later layers, where more weights are pruned.
In other words, LAMP discovers a layerwise sparsity similar to that of AMC, even without training a reinforcement learning agent.
We note, however, a slight discrepancy in the setting: AMC reports the sparsity from an iterative pruning setup, and the results in Fig.
6 are from a one-shot pruning setup.
In the model pruned with global MP, such pattern does not stand out.
In the model pruned with
Erd ¨ os-R ´ enyi kernel method, peaks and crests are extremely evident, even more than those discovered by AMC.
G LAMP WITH AN EXTENSIVE TRAINING SCHEDULE
In this section, we depart from the standardized training setup in the main text which uses Adam with weight decay Loshchilov & Hutter (2019), without any learning rate decay.
Instead, we validate the effectiveness of LAMP using the optimizer and hyperparameters extensively tuned for the maximum performance of the model.
In particular, we use conﬁgurations from Liu et al.
for training
VGG-16 on the CIFAR-10 dataset.
The experimental result is reported in Fig.
Similar to the experiments appearing in the main text, we observe that LAMP continues to outperform or match the performances of the baselines.