Network Slimming

Network Slimming是一种有效的卷积神经网络压缩方法,通过在训练过程中引入通道尺度因子γ并应用稀疏正则化,识别并剪枝不重要的通道。结合Batch Normalization,它能在不牺牲精度的情况下减小模型大小、内存占用和计算量。经过剪枝后的模型在微调后能达到或超过原始模型的性能。该方法适用于VGGNet在CIFAR-10数据集上的实验。
摘要由CSDN通过智能技术生成

Network Slimming-Learning Efficient Convolutional Networks through Network Slimming](https://arxiv.org/abs/1708.06519)

Network Slimming为一种神经网络训练方法,其可以减小模型大小,运行内存,计算量,同时没有精度损失,并最小化训练过程.得到的模型不需要专门的库或者软件来运行.

通道剪枝:

img

在卷积层的每个通道引入一个尺度因子 γ ,具体操作为将尺度因子 γ 与该通道的输出相乘. 在训练的过程中,对这些尺度因子采用sparsity regularization操作,以自动鉴定不重要的通道.这样需要训练的参数包括网络本身的参数,以及每个通道的尺度因子 γ ,训练目标函数为:

这里写图片描述

式中,(x,y)表示训练输入,目标输出.W为网络的各层的权重参数. l(f(

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值