算法工程师必备系列更新啦!继上次推出了算法工程师必备的数学基础后,小编继续整理了必要的机器学习知识,全部以干货的内容呈现,哪里不会学哪里,老板再也不用担心你的基础问题!
反向传播算法
前向传播 forward propagation
过程:当前馈神经网络接收输入 并产生输出 时,信息前向流动。
输入提供初始信息,然后信息传播到每一层的隐单元,最终产生输出 。
-
反向传播算法back propagation
允许来自代价函数的信息通过网络反向流动以便计算梯度。
反向传播并不是用于学习整个神经网络的算法,而是仅用于计算梯度的算法。
神经网络的学习算法是随机梯度下降这类基于梯度的算法。
反向传播不仅仅适用于神经网络,原则上它适用于计算任何函数的导数。
图中的每个节点代表一个变量(可以是标量、向量、矩阵或者张量)。
操作:operation
为一个或者多个变量的简单函数。
• 多个操作组合在一起可以描述一个更复杂的函数。
• 一个操作仅返回单个输出变量(可以是标量、向量、矩阵或者张量)。
如果变量 是变量 通过一个操作计算得到,则在图中绘制一条从 到的有向边。如: 的计算图:

一、链式法则
在多维情况下,设:, 为的映射且满足 , 为到 的映射且满足 。则有:
使用向量记法,可以等价地写作:
其中: 为 g 的阶雅可比矩阵, 为 z 对的梯度, 为 z 对 的梯度:
反向传播算法由很多这样的雅可比矩阵与梯度的乘积操作组成。
1.1 张量链式法则
记
为 对张量 的梯度。 现在有多个索引(如:二维张量有两个索引),可以使用单个变量来表示 的索引元组(如 i= 表示:一个二维张量的索引,每个维度三个元素)。
这就与向量中的索引方式完全一致: 。
如:
则有:
设 ,用单个变量 j 来表示的索引元组。则张量的链式法则为:
如:
则有:
1.2 重复子表达式
给定计算图以及计算图中的某个标量 z ,根据链式法则可以很容易地写出 z 对于产生 z 的任意节点的梯度的数学表达式。
但是在计算该表达式的时候,许多子表达式可能在计算整个梯度表达式的过程中重复很多次。

如图中:
在复杂的计算图中,可能存在指数量级的重复子表达式,这使得原始的链式法则几乎不可实现。
一个解决方案是:计算 一次并将它存储在 x 中,然后采用来计算梯度。
这也是反向传播算法采用的方案:在前向传播时,将节点的中间计算结果全部存储在当前节点上。其代价是更高的内存开销。
2. 有时候必须重复计算子表达式。这是以较高的运行时间为代价,来换取较少的内存开销。
二、反向传播
2.1 前向传播
假设有个输入节点: 。它们对应的是模型的参数和输入。
假设每个非输入节点 ,操作 与其关联,并且通过对该函数求值得到:。
通过仔细排序(有向无环图的拓扑排序算法),使得可以依次计算。

• 计算图
• 初始化向量
2.2 反向传播
计算时需要构造另一张计算图
:它的节点与中完全相同,但是计算顺序完全相反。
计算图 如下图所示:

对于图中的任意一非输出节点 (非),根据链式法则:
其中 表示图 中的边。
若图 中存在边,则在图 中存在边 ,则为的子节点。
• 计算图
• 初始化参数向量
• 运行计算 的前向算法,获取每个节点的值。
• 给出一个 grad_table表,它存储的是已经计算出来的偏导数。
• 初始化 。
• 沿着计算图 计算偏导数。遍历 j 从 到 1 :
反向传播算法计算所有的偏导数,计算量与 中的边的数量成正比。
其中每条边的计算包括计算偏导数,以及执行一次向量点积。
上述反向传播算法为了减少公共子表达式的计算量 ,并没有考虑存储的开销。这避免了重复子表达式的指数级的增长。
某些算法可以通过对计算图进行简化从而避免更多的子表达式。
有些算法会重新计算这些子表达式而不是存储它们,从而节省内存。
2.3 反向传播示例
对于 ,将公式拆分成 和 ,则有:
根据链式法则,有。
假设 ,则计算图如下。其中:绿色为前向传播的值,红色为反向传播的结果。

前向传播,计算从输入到输出(绿色);反向传播,计算从尾部开始到输入(红色)。
在整个计算图中,每个单元的操作类型,以及输入是已知的。通过这两个条件可以计算出两个结果:
◾ 这个单元的输出值。
◾ 这个单元的输出值关于输入值的局部梯度比如 和 。
每个单元计算这两个结果是独立完成的,它不需要计算图中其他单元的任何细节。
但是在反向传播过程中,单元将获取整个网络的最终输出值(这里是 f )在单元的输出值上的梯度,即回传的梯度。
链式法则指出:单元应该将回传的梯度乘以它对其输入的局部梯度,从而得到整个网的络输出对于该单元每个输入值的梯度。如:。
在多数情况下,反向传播中的梯度可以被直观地解释。如:加法单元、乘法单元、最大值单元。
假设:,前向传播的计算从输入到输出(绿色),反向传播的计算从尾部开始到输入(红色)。
加法单元 ,则 。如果 ,则有:
这表明:加法单元将回传的梯度相等的分发给它的输入。
乘法单元 ,则 。如果 ,则有:
这表明:乘法单元交换了输入数据,然后乘以回传的梯度作为每个输入的梯度。
如果,则有:
这表明:取最大值单元将回传的梯度分发给最大的输入。
通常如果函数 的表达式非常复杂,则当对 进行微分运算,运算结束后会得到一个巨大而复杂的表达式。
实际上并不需要一个明确的函数来计算梯度,只需要如何使用反向传播算法计算梯度即可。
可以把复杂的表达式拆解成很多个简单的表达式(这些表达式的局部梯度是简单的、已知的),然后利用链式法则来求取梯度。
在计算反向传播时,前向传播过程中得到的一些中间变量非常有用。实际操作中,最好对这些中间变量缓存。
2.4 深度前馈神经网络应用
给定一个样本,其定义代价函数为 ,其中 为神经网络的预测值。
考虑到正则化项,定义损失函数为:。其中 为正则化项,而
包含了所有的参数(包括每一层的权重 和每一层的偏置 。
这里给出的是单个样本的损失函数,而不是训练集的损失函数。
3. 前向传播用于计算深度前馈神经网络的损失函数。算法为:
○ 输入:
◾网络层数 l
◾每一层的权重矩阵
◾每一层的偏置向量
◾每一层的激活函数
也可以对所有的层使用同一个激活函数
◾ 输入 和对应的标记 y 。
◾隐层到输出的函数 。
○ 输出:损失函数
○ 算法步骤:
反向传播用于计算深度前馈网络的损失函数对于参数的梯度。
梯度下降算法需要更新模型参数,因此只关注损失函数对于模型参数的梯度,不关心损失函数对于输入的梯度 。
• 根据链式法则有:
因为 ,因此:
上式仅仅考虑从 传播到中的梯度。考虑到损失函数中的正则化项 包含了权重和偏置,因此需要增加正则化项的梯度。则有:
5. 反向传播算法:
◾ 输出:梯度
◾ 算法步骤:
计算输出层的导数: 。
这里等式成立的原因是:正则化项 与模型输出 无关。
迭代:,迭代步骤如下:
每一轮迭代开始之前,维持不变式:。
○ 计算 :
三、算法实现
3.1 符号-数值 / 符号-符号方法
代数表达式和计算图都是对符号symbol
进行操作,这些基于代数的表达式或者基于图的表达式称作符号表达式。
当训练神经网络时,必须给这些符号赋值。如:对于符号 赋予一个实际的数值,如 。
符号到数值的方法:给定计算图,以及图的一组输入的数值,然后返回在这些输入值处的梯度。
这种方法用于Torch
和Caffe
之类的库中。
符号到符号的方法:给定计算图,算法会添加额外的一些节点到计算图中,这些额外的节点提供了所需的导数的符号描述。
这种方法用于Theano
和TensorFlow
之类的库中。
下图左侧为 的计算图,右侧添加了若干节点从而给出了计算 的计算图。
4. 符号到符号的方法的优点:导数可以使用与原始表达式相同的编程语言来描述。
导数只是另外一张计算图,因此可以再次运行反向传播算法对导数再进行求导,从而获取更高阶的导数。
5. 推荐使用符号到符号的方法来求导数。一旦构造出了添加导数后的计算图,那么随后如果给出了输入的数值,可以对图中的任意子节点求值。
目前通用的计算图求解引擎的做法是:任意时刻,一旦一个节点的父节点都求值完毕,那么该节点将能够立即求值。
6. 事实上符号到数值的方法与符号到符号的方法执行了同样的计算过程,区别在于:
符号到符号的方法将各种求导运算暴露出来,添加到计算图中成为了节点。
3.2 算法框架
假设计算图 中的每个节点对应一个变量。这里将变量描述为一个张量 ,它可以具有任意维度并且可能是标量、向量、矩阵。
根据前面介绍的张量的链式法则,,则张量的链式法则为:
其中 为张量 展平为一维向量后的索引, 为张量 展平为一维向量之后的第 个元素。
3.2.1 三个子过程
:返回用于计算 的操作operation
。它就是tensorflow
中的Operation
对象。该函数通常返回一个操作对象 :
该对象有个 方法,该方法给出了父节点到 的函数: 。
其中 为 的父节点集合:
该操作对象有个bprop
方法。给定 的某个子节点 ,该方法用于已知 的梯度 ,求解 对于 的梯度的贡献: 。
如果考虑 的所有子节点集合 ,则 它们的梯度贡献之和就是总的梯度:
:返回图 中节点 的子节点列表,也就是节点 的子节点集合: 。
:返回图中节点 的父节点列表,也就是 的父节点集合: 。
op.bprop
方法总是假定其输入节点各不相同。
如果定义了一个乘法操作,而且每条输入节点都是x,则op.bprop
方法也会认为它们是不同的:op.bprop
会认为其输入分别为y和z,然后求出表达式之后再代入 。
大多数反向传播算法的实现都提供了operation
对象以及它的bprop
方法。
如果希望添加自己的反向传播过程,则只需要派生出op.bprop
方法即可。
3.2.2 反向传播过程
build_grad
过程采用符号-符号方法 ,用于求解单个结点
。
build_grad
在求解过程中会用到裁剪的计算图 ,会剔除所有与 梯度无关的节点,保留与 梯度有关的节点。
如果 已经就在 中,则直接返回
。
这样可以节省大量的重复计算
在 中插入节点 来更新计算图 。插入过程不仅增加了节点 ,还增加了 的父节点到 的边。
裁剪为 ,使得 仅包含 的祖先之中,那些同时也是 的后代的节点。
因为这里只关心 中节点的梯度
初始化
,它是一个表,各表项存储的是 z 对于对应节点的偏导数。
3.2.3 算法复杂度
算法复杂度分析过程中,我们假设每个操作的执行都有大概相同的时间开销。
实际上每个操作可能包含多个算术运算,如:将矩阵乘法视为单个操作的话,就包含了很多乘法和加法。因此每个操作的运行时间实际上相差非常大。
在具有 个节点的计算图中计算梯度,不会执行超过的操作,也不会执行超过 个存储。
因为最坏的情况下前向传播将遍历执行图中的全部 个节点,每两个节点之间定义了一个梯度。
大多数神经网络的代价函数的计算图是链式结构,因此不会执行超过 的操作。
从 降低到 是因为:并不是所有的两个节点之间都有数据通路。
如果直接用梯度计算公式来求解则会产生大量的重复子表达式,导致指数级的运行时间。
反向传播过程是一种表填充算法,利用存储中间结果(存储子节点的梯度) 来对表进行填充。计算图中的每个节点对应了表中的一个位置,该位置存储的就是该节点的梯度。
通过顺序填充这些表的条目,反向传播算法避免了重复计算公共表达式。这种策略也称作动态规划。
3.4、应用
考虑只有单个隐层的最简单的深度前馈网络,使用小批量(minibatch
)随机梯度下降法训练模型。反向传播算法用于计算单个minibatch
上的代价函数的梯度。
3. 目标是通过小批量随机梯度下降法求解代价函数的最小值,因此需要计算 。
从图中看出有两种不同的路径从 :
一条路径是通过正则化项。
这条路径对于梯度的贡献相对简单,它对于 的梯度贡献为。
对于 ,这条分支其梯度的贡献为 ,其中 ,将 替换为
• 首先计算 。
• 然后根据relu
操作的反向传播规则:根据 中小于零的部分,对应地将
对应位置清零,记清零后的结果为 。
• 分支的梯度贡献为:。
4. 该算法的计算成本主要来源于矩阵乘法:
前向传播阶段(为了计算对各节点求值):乘-加运算的数量为,其中 w 为权重的数量。
5. 算法的主要存储成本是:需要存储隐层非线性函数的输入。因此存储成本是 ,其中 m 为 minibatch
中样例的数量,是隐单元的数量。
6. 这里描述的反向传播算法要比现实中实际使用的实现更简单。
这里定义的operation
限制为返回单个张量的函数,大多数软件实现支持返回多个张量的operation
。
这里未指定如何控制反向传播的内存消耗。反向传播经常涉及将许多张量加在一起。
• 朴素算法将分别计算这些张量,然后第二步中将所有张量求和,内存需求过高。
• 可以通过维持一个buffer
,并且在计算时将每个值加到buffer中来避免该瓶颈。
反向传播的具体实现还需要处理各种数据类型,如32位浮点数、整数等。
一些operation
具有未定义的梯度,需要跟踪这些情况并向用户报告。
四、自动微分
自动微分研究如何数值计算导数,反向传播算法只是自动微分的一种方法,它是反向模式累加算法的特殊情况。
不同的自动微分算法以不同的顺序计算链式法则的子表达式,找到计算梯度的最优操作序列是NP
完全问题。
事实上,反向传播算法不能化简梯度计算公式,它只会通过计算图来传播梯度。如:
而反向传播算法并不知道可以这样简化。
有些软件如Theano
和TenserFlow
能够尝试执行某些类型的代数替换来部分地简化梯度计算公式。
若前向图 具有单个输出节点,并且每个偏导数 都能够以恒定的计算量来计算时,反向传播保证梯度的计算数量和前向计算的计算数量都在同一个量级。计算量都是 ,l 为边的数量。
反向传播算法可以扩展到计算Jacobi
矩阵。矩阵来源可能是图中的 k 个不同标量节点;也可能来源于某个节点,但是该节点是个张量,张量内部具有 k 个内部标量。
当图的输出 数目大于输入数目时,有时倾向于使用另一种形式的自动微分,称作前向模式累加。
前向模式累加避免了存储整个图的值和梯度,是计算效率和内存消耗的折中。
前向模式和后向模式(反向传播算法就是后向模式的一种)的关系类似于:左乘、右乘一系列矩阵。如:
。
• 如果从右向左进行,则只需要计算矩阵-向量乘积。这对应着反向模式。
• 如果从左向右进行,则需要计算矩阵-矩阵乘积(只有最后一次计算是矩阵-向量乘积)。这对应着前向模式,这时总的计算量和存储空间更大。
在机器学习以外的领域,传统的编程语言比如C/Python/C++
直接实现了求导数的库。
在深度学习中无法使用这些库,而是必须使用专用库创建明确的数据结构来表示求导。
深度学习需要深度学习库的开发人员为每个操作定义bprop
方法,并且限制库的用户只能使用该方法来求导。
一些软件框架支持使用高阶导数(如Theano/TensorFlow
),这些库需要使用一种数据结构来描述要被微分的原始函数。
它们使用相同的数据结构来描述原始函数的导数(导数也是一个函数对象),这意味着符号微分机制可以产生高阶导数。
在深度学习领域很少关注标量函数的单个二阶导数,而是关注海森矩阵的性质。
假设函数 ,则海森矩阵大小为 。在某些深度学习任务中 n 为模型参数数量,很可能达到十亿,因此完整的海森矩阵无法在内存中表示。
一个解决方案是使用Krylov
方法。该方法使用矩阵-向量乘法来迭代求解矩阵的逆、矩阵特征值、矩阵特征向量等问题。
为了使用Krylov
方法,需要计算海森矩阵和任意向量 。
因为 无关,因此在[]
里面或者外面没有区别。
上式中的两个梯度计算都可以由软件库自动完成。
注意:如果
本身如果已经是计算图产生的一个向量,则需要告诉软件库不要对产生 的节点进行微分。