深度学习网络压缩模型方法总结(model compression)

两派

1. 新的卷机计算方法

这种是直接提出新的卷机计算方式,从而减小参数,达到压缩模型的效果,例如SqueezedNet,mobileNet网络

  1. SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size学习

    修改网络结构,相似于mobileNet编码

  2. MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationslua

    修改卷机计算方式,depth-wise 卷积.net

2. 已训练好的模型上作裁剪

这种就是在训练好的模型上作一些修改,而后在fine-tuning到原来的准确率,主要有一些方法blog

  • 剪枝:神经网络是由一层一层的节点经过边链接,每一个边上会有权重,所谓剪枝,就是当咱们发现某些边上的权重很小,能够认为这样的边不重要,进而能够去掉这些边。在训练的过程当中,在训练完大模型以后,看看哪些边的权值比较小,把这些边去掉,而后继续训练模型;ci

  • 权值共享:就是让一些边共用一个权值,达到缩减参数个数的目的。假设相邻两层之间是全链接,每层有1000个节点,那么这两层之间就有1000*1000=100万个权重参数。能够将这一百万个权值作聚类,利用每一类的均值代替这一类中的每一个权值大小,这样同属于一类的不少边共享相同的权值,假设把一百万个权值聚成一千类,则能够把参数个数从一百万降到一千个。get

  • 量化:通常而言,神经网络模型的参数都是用的32bit长度的浮点型数表示,实际上不须要保留那么高的精度,能够经过量化,好比用0~255表示原来32个bit所表示的精度,经过牺牲精度来下降每个权值所须要占用的空间。it

  • 神经网络二值化:比量化更为极致的作法就是神经网络二值化,也即将全部的权值不用浮点数表示了,用二进制的数表示,要么是+1,要么是-1,用二进制的方式表示,原来一个32bit权值如今只须要一个bit就能够表示,能够大大减少模型尺寸。io

  1. XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networks

    网络权重二值化、输入二值化,从头开始训练一个二值化网络,不是在已有的网络上二值化

  2. Learning both Weights and Connections for Efficient Neural Networks

    学习重要的链接,根据链接的权重进行裁剪,fine-tuning让网络保持稀疏的链接

  3. Exploiting linear structure within convolutional networks for efficient evaluation.

    对已经训练好的网络应用奇异值分解

  4. Eie: Efficient inference engine on compressed deep neural network.

    加速器

  5. Deep compression: Compressing DNNs with pruning, trained quantization and huffman coding.

    裁剪(阈值)、量化(8bit,存储方式)、哈夫曼编码
    http://blog.csdn.net/may0324/article/details/52935869

  6. Deep Model Compression: Distilling Knowledge from Noisy Teachers

    Teacher-student Framework,一个网络指导另一个网络的训练

  7. PerforatedCNNs: Acceleration through Elimination of Redundant Convolutions

    在一些稀疏位置跳过CNN求值,加速效果不明显,且不能压缩模型

  8. Binarized Neural Networks: Training Neural Networks with Weights and Activations Constrained to +1 or −1

    训练二值化网络
    https://tensortalk.com/?cat=model-compression-parameter-pruning

  9. PRUNING FILTERS FOR EFFICIENT CONVNETS

    计算filter的L1范数,直接过滤掉较小L1范数对应的feature map, 而后再次训练,有两种,一是每裁剪一层训练一下,一是直接裁剪整个网络,而后fine-tuning. 相比权值链接的裁剪,这种比较暴力,好处就是不会引入稀疏矩阵的计算,从而也不须要稀疏矩阵库,坏处就是可能没法恢复到最开始的准确率。

相关文章
相关标签/搜索