【Pruning系列:一】Learning Efficient Convolutional Networks through Network Slimming

本文介绍了通过在网络中引入BN层的channel-wise scaling factor并应用L1正则化,来实现卷积网络的高效学习。通过使γ参数趋于0,可以有效地剪枝不重要的通道。在目标函数中,结合模型损失与γ的稀疏约束,逐步调整网络结构。实验结果显示,经过网络瘦身,γ值大部分接近0,表明大量通道被有效精简。
摘要由CSDN通过智能技术生成

NS

机构

Tsinghua,Intel

【Pruning系列:二】Learning Efficient Convolutional Networks through Network Slimming|YOLOv3实践 |Pytorch 总结

motivation

训练中的剪枝

基于BN(Batch Normalization)层的广泛使用,在BN层加入channel-wise scaling factor 并对之加L1 regularizer使之稀疏,然后裁剪scaling factor值小的部分对应权重

在这里插入图片描述

  • 用 BN 层的 y 来表示卷积核的重要程度,y 小对应卷积核重要性低

    γ 非常小时,送入下一层的值就非常小,可以直接剪掉

  • 虽然可以通过删减 γ 值接近零的channel,但是一般情况下, γ 值靠近0的channel还是属于少数
    于是作者采用 L1 or smooth-L1 惩罚 γ ,来让 γ 值倾向于0

method

目标函数:

  • 2
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值