![AI加速器架构设计与实现](https://wfqqreader-1252317822.image.myqcloud.com/cover/311/48213311/b_48213311.jpg)
1.4 网络的算子
了解了网络的基本块之后,我们还需要对网络中的具体运算进行深入的分析和理解,将重要的、常用的运算进行分类总结,提取相应的算子。表1-2是从图像处理相关领域的神经网络中提取的算子,我们选择一些重要的算子进行说明。
表1-2 神经网络中的算子
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/28_01.jpg?sign=1739407204-qTI9Z2CkmRcatzzua9H9ZO4FezLihf39-0-30fa962d6f97d26bd6278ced12c57431)
(续)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/29_01.jpg?sign=1739407204-ywGJF7Bz3Gl7959Rcdc5PyocdDAHPZVG-0-b328076f88ba778408b8f4d0d1e31491)
1.卷积
卷积(Convolution)是卷积神经网络中最重要的运算方式,这可能也是CNN被称作卷积神经网络的原因。无论从运算量还是参数量来看,卷积运算所占的比例都很高,进行硬件架构设计前的首要任务就是把卷积理解透彻。
卷积按照运算特点可分为Norm Conv、group Conv、3D Conv、de-Conv、dilate Conv等。Conv是基础,后面几种都是在Conv的基础上进行的变化。由于卷积运算中一般含有对偏置(bias)的加法操作,这个加法操作在硬件实现上具有独立性,因此本书中描述的卷积不包括对偏置的加法操作。对于输入为W×H×C的特征图,对应的权重为S×R×C×K,输出特征图为W'×H'×K,其卷积运算过程如下。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/29_02.jpg?sign=1739407204-NhH6Fqi8lVb2p7UcBHNG2n4gNzt2Dbgn-0-919423d9035406488f0df532248e9d83)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/30_01.jpg?sign=1739407204-rCwWIL8ixhLnzrzmagGWN299LMWFl7y2-0-9a9f962d7f8148f6cf2ad02586f88430)
式中,W表示输入宽度(Width),H表示输入高度(Height),C表示输入通道数(Channel),S表示权重宽度(Weight Width),R表示权重高度(Weight Height),K表示输出通道数(Kernel)。DX表示水平方向的空洞卷积步长(Dilate X),DY表示垂直方向的空洞卷积步长(Dilate Y),LP表示向左填补(Left Pad),RP表示向右填补(Right Pad),SX表示水平方向步长(Stride X),SY表示垂直方向步长(Stride Y),TP表示向上填补(Top Pad),BP表示向下填充(Bottom Pad)。本书后续公式参数含义同此处一致。
图1-19是一个S×R=3×3的例子。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/30_02.jpg?sign=1739407204-MvyTIg3uad0kk6fM5ZNcj2SwDs0jZiaf-0-3ff2262d05bb09c4572959b4f0bf7659)
图1-19 S×R=3×3的卷积运算
对于不同的输出通道,使用的权重也是不同的,图1-19只画出了第一个输出通道使用的权重。对于普通卷积运算来说,权重R、S的尺寸也不相同,常见的卷积核尺寸如表1-3所示。
表1-3 常见卷积核尺寸
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/31_01.jpg?sign=1739407204-RU5mnqk6HzzG5i4bxRUdJVM91tbUxqcT-0-73b55eb4aed6bcb6d0c3705e43c8d71d)
卷积核尺寸类似时,步长(stride)的尺寸如表1-4所示。
表1-4 常见卷积步长
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/31_02.jpg?sign=1739407204-LdIyOZGwGCfBcKdIIebaofePfTVssQBO-0-906b4a3cbd9ff1a1cd93fd7dc5546b9f)
需要注意的是,在ResNet中有可能出现步长比卷积核尺寸大的情况,在硬件实现时要考虑对这种情况的支持。
2.池化
池化(Pooling)运算是仅次于卷积的常用算子之一。池化可分为最大池化、平均池化、全局平均池化、上采样池化等,其中最大池化最常见。如图1-20是一个基于最大池化的掩码池化。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/31_03.jpg?sign=1739407204-5Bpvip06oddckPxEQY6mKTFLp40Ljwbs-0-fefd8ba878a388966c53aef0c1a993d4)
图1-20 最大池化的掩码池化
被掩码的元素不参与池化操作,具体哪个元素会被掩码,需要通过参数来控制,这样就会使掩码池化变得有点奇怪。掩码池化仅有个别网络在使用,对于这种会大大增加硬件复杂度,又不太通用的算法,我们在架构设计时可以将其舍弃。如果硬件开销不大,可以考虑从硬件方面给予支持。最大池化运算使用的滤波器尺寸如表1-5所示。
表1-5 最大池化运算使用的滤波器尺寸
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/32_01.jpg?sign=1739407204-CLsZMFMYEbYBwOFoJhYiHOwT344mCpd8-0-72783caea404b8d1e474b5d988d618c7)
需要说明的是,表1-5仅针对最大池化运算,进行其他池化操作时,滤波器尺寸可能会比较大,甚至超过3×3,在硬件实现时需要注意。此外多数情况下,池化操作的步长是2或者3,也有出现其他情况的可能。
如图1-21是一个在注意力网络(Attention Net)中使用通道池化的例子。
从硬件角度来看,相对于卷积操作,池化的运算量不大,硬件实现也相对简单。需要注意的是,池化操作的种类繁多,并且在神经网络中池化层跟卷积层交替排列,如果池化架构有问题,就会影响整个加速器的性能。
3.全连接
从硬件实现的角度来看,全连接运算是卷积核尺寸和输入特征图尺寸相同的卷积运算的特例。如图1-22所示是两种常见的全连接算子运算过程。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/33_01.jpg?sign=1739407204-ufNuY6lFrVilsApNth0HmEM7Mm16fhX2-0-8d170a2d8fc34d54f9824ee42d91c9d2)
图1-21 通道池化示例
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/33_02.jpg?sign=1739407204-Hagcxq4ylLYgJxKwW4adr47LUaxQRzUE-0-9aa8f169263d6ee7e3fa275b618ad65b)
图1-22 两种全连接算子运算
4.激活函数
目前大部分神经网络引入了激活函数(Activation),从图像处理相关神经网络中提取出来的激活函数一般包括ReLU系列、Sigmoid、Tanh等。下面整理了ReLU系列的激活函数。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/34_01.jpg?sign=1739407204-HNY39O1J0bQJAlZC0WnHlpv6py5fCglC-0-24674d10336f91430a60a77554c44706)
图1-23是几种ReLU函数曲线。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/34_02.jpg?sign=1739407204-GJvIPtNjGO8klGBPUW0vjXRBbzjxzbqC-0-20fa26e067d294a34a0bc30e1103c50f)
图1-23 几种ReLU函数曲线
Sigmoid函数的定义如表1-6所示,函数曲线如图1-24所示。
Tanh函数的定义如表1-7所示,函数曲线如图1-25所示。
表1-6 Sigmoid函数定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/34_03.jpg?sign=1739407204-1jgiKekJERuJdDMqNbOFlJZZ6z34X6yf-0-ac71fac6dd2278f6910a20fd379122e4)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/35_01.jpg?sign=1739407204-eRfICEr2HOXeU1bPFSGQy1lx9E1ILSFY-0-89dadaabc444d1e5f3dc3c1a3cead1dd)
图1-24 Sigmoid函数曲线
表1-7 Tanh函数定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/35_02.jpg?sign=1739407204-1SRmQDjruElhZMWkNlCFqHKPTFhO2KJp-0-ec5d092c579ef98122215b24fcb84ffa)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/35_03.jpg?sign=1739407204-VmVl5a9LKJ1ncYdM5otscQA1sM5XPPFz-0-981153645c4d673265b76bd5300fbb07)
图1-25 Tanh函数曲线
从硬件角度来看,除了ReLU函数外,其他激活函数如果直接用硬件实现,代价会比较高,可以考虑用LUT(Look Up Table,查找表)实现这些非线性函数。
5.归一化
为了解决某些问题,算法研究者引入了一些归一化操作,例如BatchNorm、LRN、L2 Norm等。BatchNorm的定义如表1-8所示。
表1-8 BatchNorm的定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/36_01.jpg?sign=1739407204-1zNgbafoNJOgh9u40gmHuO7ZksFekboU-0-713baa229128ff2224ee93276fbcb5e2)
乍一看BatchNorm很复杂,其实可以化简成表1-9的形式。
表1-9 BatchNorm的化简
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/36_02.jpg?sign=1739407204-62MYxHMFDmdjcvxMhMfTopcIAJpd053p-0-b5197b91c3fb86bd26181c15121734a9)
LRN的定义如表1-10所示。
表1-10 LRN的定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/36_03.jpg?sign=1739407204-akeGPFYGNj5ISvo612onu5BZaORxuTLw-0-42178e4079d3d190a4ffb0adfeba2d3e)
LRN包括通道内操作和跨通道操作,如表1-11所示,示意图如图1-26、图1-27所示。
表1-11 LRN的操作
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/36_04.jpg?sign=1739407204-PTeuIKej0R6UYuhoBO5RXQ8QES5nRXrd-0-75eea314014c00c5d526a611926da608)
(续)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/37_01.jpg?sign=1739407204-pIFGc3ryBo0lXGGkfgog1oZVfCBkx9za-0-1de4599c6f1748d847affd540af63575)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/37_02.jpg?sign=1739407204-CjvvjFlifCtQiOqHpIMWfnPCaDL2MmJd-0-4f35bca87f31d8e20df92cd7c56f78a3)
图1-26 通道内操作示意图
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/37_03.jpg?sign=1739407204-aqNi306TrkrtWfVLXwEV4S98mNAQDSiN-0-becfdca4d55d022a9e11cc63155b4bff)
图1-27 跨通道操作示意图
L2_Normalization(L2_Norm)也包括通道内操作和跨通道操作两类,如表1-12、表1-13所示。
表1-12 L2_Normalization的定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/37_04.jpg?sign=1739407204-LsBylBXp2Z4fp2PYDC6mQW13BExVR2oi-0-d2e36221e775d2c503287c3707085771)
表1-13 L2_Normalization的操作
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/38_01.jpg?sign=1739407204-zmACiPgorszNFeacAOW2rjxSKX8CkLIW-0-58c4eb6b4d6de233a16ff4f9af0a5478)
6.Softmax
对于目前大多数神经网络,在最后会包含一个Softmax层,定义如表1-14所示。
表1-14 Softmax的定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/38_02.jpg?sign=1739407204-Jc1ZrtHuglChD3P4K0OzdJBm1YKLuHcl-0-e977896694ab2b32e9acd7395798422d)
7.其他
目前的神经网络种类很多,其中使用的算子也很多,如表1-15所示。从硬件角度来看,可能不需要实现所有的算子,但最好都了解一下,以防出现严重的功能性缺陷。
表1-15 神经网络中的部分算子
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/38_03.jpg?sign=1739407204-4xPxWArWQIwvEukIqcZjoEWq6XCgSzg8-0-cb0af4a6ccdcc74f64cf065a3332854d)
(续)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/39_01.jpg?sign=1739407204-JVww3rsKrPchzUyaV8o1wWDWz0JFSMz6-0-51fa5c9c1be44dbc1653788c1a3bbe6b)
算法不止,架构更新不止。目前神经网络算法还在快速演进中,硬件架构师要紧跟算法潮流,弄清目标领域的需求。根据众多影响因素下进行取舍,正是架构设计中最重要的任务所在。