Convolution的变形 -- Convolution、 InnerProduct、MatMul、BLAS(GEMM、GEMV) 之间的关联

1 前言

MNN 模型模型转换结束后进行 optimizeNet 的第一项即为 turnInnerProduct2Convolution,抛开各种复杂的优化考虑,单是 减少了概念的维护 一点就已经很让人心动了。

日常开发中,可能大家都感到过 Inner Procuct、MatMul、BLAS、Concolution 等操作存在那么些相似性,但究竟哪里相似,哪里不同,或许没有去仔细地分析过。一方面因为多维操作的分析想象的确比较烧脑;另一方面没有需求啊,推理用就好了,模型用就好了,跑崩了官方提issue啊,找前向框架开发解决啊!

这不失为一个不错的思路。然而,抛开某些紧急场景下我们要自己实现一些模型操作转换设计,又或要实现一些模型操作优化…… 作为一个AI时代的新青年,难道我们不该有一些“AI工程师的好奇心”,培养下 “AI工程师的自我修养” 嘛!

追根究底 是件很有趣的事情, 同事 也是 作为积累 在很多紧急的关键时刻能 “救命”的东西。那么马上开始,让我们一起来看看 类似卷积的相关操作的相互变换 吧。

2 变换

2.1 回忆 - 卷积的关键参数

我们先来回忆一下卷积的相关参数

2.1.1 输入Input

输入

input_width : 输入宽,如图设为10
input_height : 输入高,如图设为10
input_channel : 输入维度,如图设为3
input_padX : 水平方向加边 - 本文不讨论,设为0
input_padY : 垂直方向加边 - 本文不讨论,设为0
input_padMode : 加边模式 - 本文不讨论
input_group_number : 输入分组 - 暂不讨论,设为 1

2.1.2 卷积核Kernal

卷积核

kernal_width : 卷积核宽,如图设为3
kernel_height : 卷积核高,如图设为3
kernal_dilateX : 水平方向膨胀 - 本文不讨论,设为1
kernal_dilateY : 垂直方向膨胀 - 本文不讨论,设为1
kernal_number = input_channel * output_channel,举例的计算结果为 3 x 4 = 12
(output_channel 的说明在 2.1.5 部分)

2.1.3 偏置Bias

偏置

bias_number = output_channel,举例的计算结果为 4
(output_channel 的说明在 2.1.5 部分)

2.1.4 其他配置(步长Stride)

strideX : 水平方向步长 - 本文不讨论,设为1
strideY : 垂直方向步长 - 本文不讨论,设为1

2.1.5 输出Output

输出

output_width : (input_width + input_padX * 2 - ((kernal_width - 1) * kernal_dilateX + 1)) / strideX + 1,按上面的设置计算,为8
output_height : (input_height + input_padY * 2 - ((kernal_height - 1) * kernal_dilateY + 1)) / strideY + 1,按上面的设置计算,为8
output_channel : 输出维度,如图设为4

2.1.6 卷积操作示例

按照上面示例的配置,我们得到一份卷积运算,如下图:

简单的卷积运算

2.2 Convolution的基础参数变换

我们首先关注如下5个配置的修改变化:

input_channel 输入维度(或输入特征)数量
output_channel 输出维度(或输出特征)数量
kernal_width 卷积核宽度
kernel_height 卷积核高度
group 卷积分组数量

卷积基本变换 - 网页中点击图片,选择[查看原图],可查看高清原图

应该蛮直观的吧,相信大家稍微不熟悉的应该只有分组卷积 (Group Convolution) 了吧。

金哥和你一起学模型压缩——结构篇(1)

这篇文章中对分组卷积略有介绍,可以适当参考。
(姚神记得给我支付宝打广告费~!)

2.3 Convolution <--> Inner Product

让我们渐入佳境,看看Convolution与Inner Product的转换关系。

Convolution <--> Inner Product - 网页中点击图片,选择[查看原图],可查看高清原图

Convolution 使用 1x1 的 kernal,再将输入 size 变为 1x1 就退化成了 Inner Product!

有些神奇对不对,更精彩的还在后面~

2.4 Convolution(Inner Product)<-> MatMul

Inner Product 退化为一个矩阵乘法很简单,Inner Product 可以看做是一个特殊的 Convolution,所以 MatMul 也可以看做一个特殊的 Convolution

Convolution(Inner Product)<-> MatMul - 网页中点击图片,选择[查看原图],可查看高清原图

但细心的小伙伴会发现我们举例的 MatMul 的左矩阵是 1 x 3 的,如果换证一个 2 x 3 的正常矩阵?……貌似转换就不太顺利了……
别急,我们先来了解下 BLAS

2.5 BLAS

BLASBasic Linear Algebra Subprograms,即 基础线性代数子程序。
我们比较常见的 GEMM,即广义矩阵乘法就是 BLAS 的一种高级形式。

BLAS - 网页中点击图片,选择[查看原图],可查看高清原图

会到我们 2.4节 遗留的问题,有没有发现 11号GEMM 变换 MatMul 很容易呢?
马上来看看 BLAS & Convolution 的相互变换

2.6 Convolution(InnerProduct) <-> BLAS

我们以退化为 Inner ProductConvolution 为例来描述 Convolution & BLAS 之间的关联

Convolution(InnerProduct) <-> BLAS - 网页中点击图片,选择[查看原图],可查看高清原图

GEMM 可以看做一个 每组kernal都完全相同 的分组卷积!
所以转模型的时候想将一个 GEMM 转化为 Convolution 的话,是要 牺牲一些存储空间 咯。

我们可以理解为分组卷积的变种

2.7 总结

把我们 2.1 ~ 2.6 描述的变换整合到一张图

卷积变换 - 网页中点击图片,选择[查看原图],可查看高清原图

还是有点复杂的,不过针对 有不同操作转换硬需求 的小伙伴,相信这张图能为整理思路节约不少的时间!

3 后记

然而,Convolution 的变换场景还不止如此,比如当 Group ConvolutionInput Channel、Output Channel、Group Number 相等时,就变成了一个 DepthWise Convolution !

DepthWise Convolution

有机会我们再慢慢研究讨论咯~

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,509评论 6 504
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,806评论 3 394
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,875评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,441评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,488评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,365评论 1 302
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,190评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,062评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,500评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,706评论 3 335
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,834评论 1 347
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,559评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,167评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,779评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,912评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,958评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,779评论 2 354

推荐阅读更多精彩内容