paper:Learning Efficient Convolutional Networks through Network Slimming
这是一篇做剪枝的paper,2017的ICCV文章。
主要亮点在于直接使用BN层的gamma参数进行剪枝评估。在卷积核的级别做剪枝,非常方便好用。
问题:
这篇paper中,Lamda的选择标准和阈值标准没有给出,应该就是作者实验试出来的。
因此,lamda能不能通过网络学习得到呢,这样其实更实用。
Idea
先看Batch Norm(BN):
上图是剪枝的示意图,可以看到,是在卷积核层面进行的剪枝。
一般情况下的网络剪枝是通过额外增加一些对原网络层的计算,得到卷积层的重要性信息。
Network Slimming:想法比较巧妙,通过BN层