深度学习与MindSpore实践
上QQ阅读APP看书,第一时间看更新

3.2 反向传播

第1章中介绍了梯度下降算法训练回归模型,神经网络模型也一样需要使用梯度下降算法来更新参数。然而一个神经网络通常会有上百万的参数,那么如何高效地计算这百万级别的参数是需要重点考虑的问题。神经网络中使用反向传播(Backward Propagation)算法,使得计算梯度更加有效率。

在介绍反向传播之前,先来介绍一下链式法则。假设有两个函数y=gx)和z=hy),那么zx的求导过程如下:

假设有三个函数x=gs)、y=hs)和z=kxy),zs的求导过程如下:

神经网络的梯度计算,就是依赖链式法则一层层反向传播的。

如图3.4所示的前向神经网络,输入层有n个属性x1x2,…,xn,中间隐藏层有p个神经元,第j个神经元为hjj∈(0,p-1)。

图3.4 神经网络结构

输出层为q维。对隐藏层的每一个神经元hj,先经过一个线性变换,公式如下:

式中,wj0——偏置值;

wj1wj2,…,wjn——作用在属性x1x2,…,xn上的权重。

输入给神经元后,经过激活函数的作用得到。第二层同理有神经元输入:

输出。以上为前向神经网络的前向传播(Forward Propagation)过程。

对单个数据样本(xy),假设损失函数为均方差,则对第k个输出项的损失为:

通过链式法则,损失函数对权重vkj的梯度为:

式中,第一项

第三项

而对于第二项,假设激活函数为sigmoid函数,则梯度有一个很好的性质,即:

三项相乘可以得到:

由于真实标签yk由数据给定,而输出值hj均由前向传播算法计算得到,则可以轻易地计算得到每个中间层权重vkj,并且该计算过程可以并行进行。

类似地,可以得到损失值在隐藏单元hj上的累积梯度为

同理,可以通过链式法则,得到损失函数对第一层权重wji的梯度为(假设隐藏层激活函数为ReLU函数):

式中,

在上一步计算得到后,也可以高效并行地计算出。

在上述过程中,假设了损失函数是均方差,激活函数为sigmoid和ReLU,其实这样的计算法则对任意可微的损失函数和激活函数都是有效的。从计算过程来看,在前向传播得到隐藏层与输出层数值后,先从损失函数算起,接着从顶层逐渐计算梯度,将梯度逐层往输入层传播。这与前向传播的顺序是相反的,也是该算法为什么被称为反向传播的原因。反向传播(见图3.5)得到所有参数的梯度之后,可以利用梯度下降算法对参数进行更新迭代,从而达到训练神经网络的目的。神经网络训练过程如算法3.1所示。

图3.5 反向传播示意图

算法3.1 神经网络训练过程

输入:数据集,步长为α,小批量训练样本的大小为b,迭代次数为T

输出:训练完成的神经网络

(1) 初始化网络参数w0

(2) for t∈{1,2,…,T}

(3) 从m个样本中均匀随机无放回选取b个样本mb

(4) 前向传播逐层计算隐藏层的参数值,得到样本输出

(5) 根据损失函数计算误差,得到输出层梯度

(6) 反向逐层计算隐藏层梯度

(7) 计算连接参数梯度并更新参数