![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=1739011989-ALcIJGC2brujLFHmUMHK6QTx8JyFFHLw-0-65265d64614358cbd4bc0e3b211bfdf7)
(续)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/29_01.jpg?sign=1739011989-s3BYplrMDXh6DANWI0dcI0NqEoUb5eyY-0-8c12c53492982db00b2bd7e36c8e699a)
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=1739011989-Iqn4I2S0OqBm8LK4ILCIuCYRWu1tghJy-0-2c4e0c0245522f57dad1ab07441240ea)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/30_01.jpg?sign=1739011989-rkBojxoSxIbPLnBYTYuqB51Rc5NJjBNS-0-f8b3cf970b66e5d54a15b302735a33b2)
式中,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=1739011989-K1LFxyD2xvylxIXb7zdWARHEJjtHCknW-0-f94880cb031813f71d6d3120445002e1)
图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=1739011989-tibcr6CpMeqUgG3p9P0sWstRcsYbOgnw-0-e957a9f249915a867c419cea9defb3f1)
卷积核尺寸类似时,步长(stride)的尺寸如表1-4所示。
表1-4 常见卷积步长
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/31_02.jpg?sign=1739011989-eDTnku6Rb9LRj0v5GyDoOIHAptBjBMMY-0-78a2861573bbad45d0ade44f85fdc4ae)
需要注意的是,在ResNet中有可能出现步长比卷积核尺寸大的情况,在硬件实现时要考虑对这种情况的支持。
2.池化
池化(Pooling)运算是仅次于卷积的常用算子之一。池化可分为最大池化、平均池化、全局平均池化、上采样池化等,其中最大池化最常见。如图1-20是一个基于最大池化的掩码池化。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/31_03.jpg?sign=1739011989-i3fsAIUcdFQo19nmCUzeHLo6vtrX8ULI-0-c89f7973ca167344582d02c8398f0e09)
图1-20 最大池化的掩码池化
被掩码的元素不参与池化操作,具体哪个元素会被掩码,需要通过参数来控制,这样就会使掩码池化变得有点奇怪。掩码池化仅有个别网络在使用,对于这种会大大增加硬件复杂度,又不太通用的算法,我们在架构设计时可以将其舍弃。如果硬件开销不大,可以考虑从硬件方面给予支持。最大池化运算使用的滤波器尺寸如表1-5所示。
表1-5 最大池化运算使用的滤波器尺寸
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/32_01.jpg?sign=1739011989-cokYHadoupms7Q79997Lm3nu8GAuhN7v-0-542192b887bc419c80fa9d7a9def5db7)
需要说明的是,表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=1739011989-G0TefvZHlpx3uc7PGOpmyDYdvXluLQHE-0-8e3eec2649452235e9da1dcc9e5a71c3)
图1-21 通道池化示例
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/33_02.jpg?sign=1739011989-PZAUCeFrMutRUEHCFbyN2P3CJqnnEC8a-0-33d4fbcaed1d9fd8682dc0e820d45a94)
图1-22 两种全连接算子运算
4.激活函数
目前大部分神经网络引入了激活函数(Activation),从图像处理相关神经网络中提取出来的激活函数一般包括ReLU系列、Sigmoid、Tanh等。下面整理了ReLU系列的激活函数。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/34_01.jpg?sign=1739011989-DoD7ib1rtokDoAFLrmwTJ3buXZnc3dlE-0-981e9e742a697b3c31aa87a90c9710be)
图1-23是几种ReLU函数曲线。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/34_02.jpg?sign=1739011989-Osd6VHS928Xbe4MC2MbaufZxd395yz9n-0-0ca2a4dfb3c08f41499ba20b077a5ff8)
图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=1739011989-7GCdtNWWn9OmGnG7cuxZRZG8Z7PNcedQ-0-60512eaf2017fbfd97051ca0c0268548)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/35_01.jpg?sign=1739011989-uaH36yKlNiBhFiTYkxQN67Utao5zR6w6-0-fd9b3c127657e8079a546decbaa1f656)
图1-24 Sigmoid函数曲线
表1-7 Tanh函数定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/35_02.jpg?sign=1739011989-XDEnMsuSAE3TpcGpsV1pyBFQdM9V7gzD-0-6f377201099a732dd699916fadd4a953)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/35_03.jpg?sign=1739011989-oxVYRtvjALHGvP64GSXCI0m8SkezhSdp-0-1f606d1118a7503c878e364f03520b68)
图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=1739011989-jFdsff3nDfyyrCscus5VsO9Att2sVOk7-0-e1d401d677ca0cb699b98916671ee966)
乍一看BatchNorm很复杂,其实可以化简成表1-9的形式。
表1-9 BatchNorm的化简
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/36_02.jpg?sign=1739011989-06kRJcdwYblVczx0IsEI2RWsKBntM8DZ-0-8a110a16957b12a82132a75259046ca9)
LRN的定义如表1-10所示。
表1-10 LRN的定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/36_03.jpg?sign=1739011989-GlEk8S7GhzG868SBnVDntx1M4uRQVwAQ-0-9c36da8e83edcb7ea53077e6acc80432)
LRN包括通道内操作和跨通道操作,如表1-11所示,示意图如图1-26、图1-27所示。
表1-11 LRN的操作
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/36_04.jpg?sign=1739011989-BkBYtXH4sj6aK69mLbvLZBE39b51p7AS-0-6e40cd0aa25dcec7f689d6cc25266dd4)
(续)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/37_01.jpg?sign=1739011989-tbBLW17E3P9Pp8ax9fb6zjj5ufefRlDn-0-5a6d4a0eb088b7bb9a1689cf69316444)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/37_02.jpg?sign=1739011989-I1vMLciwfdU14nKYkcvnhTP6W8YKtNSW-0-365805fff2c8eaa71848520cae3a8cd5)
图1-26 通道内操作示意图
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/37_03.jpg?sign=1739011989-3uXKPt7bc8VWqQUyTY73XMAk20lLrneU-0-f3131ef9e08d688b08086f690636972d)
图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=1739011989-VRalY2Djl1dHhrbXctybQQhSKGWvlngd-0-024e423245bd8751554cc552856a59b0)
表1-13 L2_Normalization的操作
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/38_01.jpg?sign=1739011989-iLLtGYi1mkpNsUNIla2VMutmQN3W800O-0-fe9835fc1538175c6783377071ac9ba3)
6.Softmax
对于目前大多数神经网络,在最后会包含一个Softmax层,定义如表1-14所示。
表1-14 Softmax的定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/38_02.jpg?sign=1739011989-T0TwW4AWzLVTe8iZwG7T3adYKOcxsFb4-0-376ef42b4f783ab952eeb5bb673ab955)
7.其他
目前的神经网络种类很多,其中使用的算子也很多,如表1-15所示。从硬件角度来看,可能不需要实现所有的算子,但最好都了解一下,以防出现严重的功能性缺陷。
表1-15 神经网络中的部分算子
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/38_03.jpg?sign=1739011989-OIgK7SzSxSx8Fnj3qIq4p1OXutOFBqlA-0-07767482c3327ca9afda2083ebd84156)
(续)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/39_01.jpg?sign=1739011989-Cc0wK44mw6rLIgiX6KPpyGbmY3jueSI5-0-cf4da5bc7e33ad5a14af1fc080db80e0)
算法不止,架构更新不止。目前神经网络算法还在快速演进中,硬件架构师要紧跟算法潮流,弄清目标领域的需求。根据众多影响因素下进行取舍,正是架构设计中最重要的任务所在。