13.分组卷积


欢迎访问个人网络日志🌹🌹知行空间🌹🌹


1.背景介绍

分组卷积最早在2012年的AlexNet中提出,当时受限于单个GPU的内存和运算能力,在通道方向上将特征图和卷积核分成 g g g组分别计算,然后再将每组卷积得到的结果 c o n c a t e n a t e concatenate concatenate后得到最终的结果。

2.标准卷积

卷积输入特征图shape: H 1 × W 1 × C 1 H_1\times W_1\times C_1 H1×W1×C1

标准卷积核的大小: h 1 × w 1 × C 1 h_1\times w_1\times C_1 h1×w1×C1

总共有 C 2 C_2 C2个上述标准卷积核,标准卷积操作会将 C 2 C_2 C2个卷积核应用到输入的特征图上,

最后得到的标准卷积输出的特征图shape为 H 2 × W 2 × C 2 H_2\times W_2 \times C_2 H2×W2×C2

在这里插入图片描述

普通卷积层参数量的计算:

C 1 × h 1 × w 1 × C 2 C_{1}\times h_1 \times w_1 \times C_{2} C1×h1×w1×C2

普通卷积层的FLOPs公式(仅包含乘法,不包括激活/偏置/加法):

C 1 × C 2 × W 2 × H 2 × w 1 × h 1 C_1\times C_2 \times W_2 \times H_2 \times w_1 \times h_1 C1×C2×W2×H2×w1×h1

2.分组卷积

分组卷积是把输入特征图在通道 C 1 C_1 C1这个维度上分成 g g g组,同时把卷积核也分成 g g g组,这样每个组的卷积核大小就变成了 h 1 × w 1 × C 1 g h_1\times w_1 \times \frac{C_1}{g} h1×w1×gC1,输出的特征图在通道维度上被分成了 g g g组。

这样分组卷积相当于把输入特征图,和卷积核在通道上分成了 g g g个卷积,然后将卷积得到的结果相合并。

对于每个组的卷积:

输入特征图的shape H 1 × W 1 × C 1 g H_1\times W_1 \times \frac{C_1}{g} H1×W1×gC1

每个组的卷积核大小 h 1 × w 1 × C 1 g h_1\times w_1 \times \frac{C_1}{g} h1×w1×gC1

输出的特征图的shape:
$
H_2\times W_2\times \frac{C_2}{g}
$

g g g组卷积后得到的特征图在通道维度上合并得:$
H_2\times W_2\times C_2
$

在这里插入图片描述

分组卷积层参数量的计算:

1 g × C 1 × h 1 × w 1 × C 2 \frac{1}{g} \times C_{1}\times h_1 \times w_1 \times C_{2} g1×C1×h1×w1×C2

分组卷积层的FLOPs公式(仅包含乘法,不包括激活/偏置/加法):

1 g × C 1 × C 2 × W 2 × H 2 × w 1 × h 1 \frac{1}{g} \times C_1\times C_2 \times W_2 \times H_2 \times w_1 \times h_1 g1×C1×C2×W2×H2×w1×h1


欢迎访问个人网络日志🌹🌹知行空间🌹🌹


致谢:图片来自于百度paddle框架文档[1]

参考资料

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值