![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=1739585517-QaNHj2VQ8ZbZzPTXPtf7c81A2WelaBCE-0-1bacde928998dd515164010a40646288)
(续)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/29_01.jpg?sign=1739585517-GiIOXFQ7gFDmG2kiCyOT35RGUvhDGdqR-0-22bd0f6cf08380702f2de2c8f35247f0)
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=1739585517-qsKlW6KWZNLmWnYByIbDVQmoyifgswTs-0-8d6d8614009dbda8fb506209b3744b06)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/30_01.jpg?sign=1739585517-UDIJu5Se595bGJvPgEz8ObduBPilwYeO-0-5d2783c4951ba0db5cf28dd72676f3ec)
式中,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=1739585517-LJ1XMJYR57Aajo7DkVGagityR7Clc8r0-0-44c05a058e385e6a99ad45155d5a0cfa)
图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=1739585517-SQPH3Px9mIx8dzDUorZLBmlQ9V0JJCnw-0-f7e98e118d65fb5b805ae87caf0503fb)
卷积核尺寸类似时,步长(stride)的尺寸如表1-4所示。
表1-4 常见卷积步长
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/31_02.jpg?sign=1739585517-G73WgLPA2fVGxGnFON4fqVAusOvluejQ-0-ed1a9247f99bfb2d681b9f3bb23514e3)
需要注意的是,在ResNet中有可能出现步长比卷积核尺寸大的情况,在硬件实现时要考虑对这种情况的支持。
2.池化
池化(Pooling)运算是仅次于卷积的常用算子之一。池化可分为最大池化、平均池化、全局平均池化、上采样池化等,其中最大池化最常见。如图1-20是一个基于最大池化的掩码池化。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/31_03.jpg?sign=1739585517-qRSkgipQN6ZuR7lGitMfg80QShSuFxXo-0-9d35ad3ebbedf7cd358d226169969b62)
图1-20 最大池化的掩码池化
被掩码的元素不参与池化操作,具体哪个元素会被掩码,需要通过参数来控制,这样就会使掩码池化变得有点奇怪。掩码池化仅有个别网络在使用,对于这种会大大增加硬件复杂度,又不太通用的算法,我们在架构设计时可以将其舍弃。如果硬件开销不大,可以考虑从硬件方面给予支持。最大池化运算使用的滤波器尺寸如表1-5所示。
表1-5 最大池化运算使用的滤波器尺寸
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/32_01.jpg?sign=1739585517-ulW6dYj0EGd0aNBYQu5AKOzIzh1wJDuL-0-06ebcf158fe69552704d2a7316148433)
需要说明的是,表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=1739585517-jX7vFGaR75APfzDJa3D07z7p5HcT0S0a-0-a927862fef9ab571ed39c1cf8da7839b)
图1-21 通道池化示例
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/33_02.jpg?sign=1739585517-e0EKxyCTCFj71Cg7IVTUx8QVAonphpLO-0-0de8b6075a7e09b45afbf2068ef52e36)
图1-22 两种全连接算子运算
4.激活函数
目前大部分神经网络引入了激活函数(Activation),从图像处理相关神经网络中提取出来的激活函数一般包括ReLU系列、Sigmoid、Tanh等。下面整理了ReLU系列的激活函数。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/34_01.jpg?sign=1739585517-MPFkmkh68OVNeJb3wqfkXew4P1Se4RK1-0-40b2b272e59c28d60c88c90e6bd1dcf5)
图1-23是几种ReLU函数曲线。
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/34_02.jpg?sign=1739585517-TKSuHPqb9VxZPGbhifb24omQkz9w9Cgs-0-5f7f570a33eaeb9d6785c1cee4674177)
图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=1739585517-SDUFVAsRVyiPi2k4ahNF1D4cFUHDT5kw-0-f57c68bc76fc2ac4573bc3ce187f3b7c)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/35_01.jpg?sign=1739585517-l5aATqW4FkBrIpV4SvnlkzJ2SXjcGIMU-0-9cb577b40e459388f989cfb9f06de425)
图1-24 Sigmoid函数曲线
表1-7 Tanh函数定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/35_02.jpg?sign=1739585517-a6lFXgsTETAZRnMU6xxmp6LYHjPCMwGC-0-2418e4b52257e156a00f0d4501508a5c)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/35_03.jpg?sign=1739585517-fymGxD8IuXeVIaWQVDtTTs6D6RoZMjT6-0-42c83e955fc4f077280c0aa49da5bccd)
图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=1739585517-YoES4ZQ9pyFG0nSxZWnk3faoSaPBHwzA-0-7fbe29b99145065539bdba4fc5b3b92a)
乍一看BatchNorm很复杂,其实可以化简成表1-9的形式。
表1-9 BatchNorm的化简
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/36_02.jpg?sign=1739585517-85txt5pGF0ffWunGERDQ6LoUsT7yVrvT-0-70084fb8e5b2df69d56508a778d03cf9)
LRN的定义如表1-10所示。
表1-10 LRN的定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/36_03.jpg?sign=1739585517-MCA5HSWoNHjx02G3CSvMN6NJMKeLyhdD-0-1cd34dd11df1f1b602285aa050989215)
LRN包括通道内操作和跨通道操作,如表1-11所示,示意图如图1-26、图1-27所示。
表1-11 LRN的操作
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/36_04.jpg?sign=1739585517-34rVFvRsWPxuYGaOfvl57RtsoYsIUijz-0-3486e0cc43fdc217adf46b28f27de532)
(续)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/37_01.jpg?sign=1739585517-rDF65odWwF0opaicd75VssjOi4PHzv9Y-0-5e426e3e95ebf1fedad50a8f61c89a38)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/37_02.jpg?sign=1739585517-sitMvJzaL7FP5aGMUwBgTq6ge17Eo90x-0-e06d42551e9f87496f7cdcfe8c9b56dd)
图1-26 通道内操作示意图
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/37_03.jpg?sign=1739585517-UJI1CSLeVNGgAFrZ3ZVIK232hLxxNOJt-0-6ece132df6d2838fd1cd147d18ccb455)
图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=1739585517-LqhJfR7C173T75hzM11p1hCmxi3tRQeY-0-b9724098b49271f574c6e17fb766576a)
表1-13 L2_Normalization的操作
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/38_01.jpg?sign=1739585517-JdTm3TDjdmNSXqEeV7hwxttAhkxGiIXN-0-810b516a3fe7947f3b6d6fa0aa4b3963)
6.Softmax
对于目前大多数神经网络,在最后会包含一个Softmax层,定义如表1-14所示。
表1-14 Softmax的定义
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/38_02.jpg?sign=1739585517-a9OZcaRR3qdWpYC852idA66CWVRv3Xgy-0-a6082e961a14579a8cc48d287981f28d)
7.其他
目前的神经网络种类很多,其中使用的算子也很多,如表1-15所示。从硬件角度来看,可能不需要实现所有的算子,但最好都了解一下,以防出现严重的功能性缺陷。
表1-15 神经网络中的部分算子
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/38_03.jpg?sign=1739585517-Tu7PSef6wtwyuYgN0ADcRA2JRfe1uSEL-0-016a89b9d58e55b80952df0906e973ff)
(续)
![](https://epubservercos.yuewen.com/37A34B/27732794007146006/epubprivate/OEBPS/Images/39_01.jpg?sign=1739585517-XtnQ0WIkElUSxyEoTN3C4Gw1jzZvJzar-0-ebd2c463d1cb652c4ba72cafcc8be7ed)
算法不止,架构更新不止。目前神经网络算法还在快速演进中,硬件架构师要紧跟算法潮流,弄清目标领域的需求。根据众多影响因素下进行取舍,正是架构设计中最重要的任务所在。