TensorFlow 2.0深度学习算法实战 第七章 反向传播算法

第七章 反向传播算法

    • 7.1 导数与梯度
    • 7.2 导数常见性质
      • 7.2.1 基本函数的导数
      • 7.2.2 常用导数性质
      • 7.2.3 导数求解实战
      • 7.3 激活函数导数
      • 7.3.1 Sigmoid 函数导数
      • 7.3.2 ReLU 函数导数
      • 7.3.3 LeakyReLU 函数导数
      • 7.3.4 Tanh 函数梯度
    • 7.4 损失函数梯度
      • 7.4.1 均方误差函数梯度
      • 7.4.2 交叉熵函数梯度
        • 7.4.2.1Softmax 梯度
        • 7.4.2.2交叉熵梯度
    • 7.5 全连接层梯度
      • 7.5.1 单个神经元梯度
      • 7.5.2 全连接层梯度
    • 7.6 链式法则
    • 7.7 反向传播算法
    • 7.8 Himmelblau 函数优化实战
    • 7.9 反向传播算法实战
      • 7.9.1 数据集
      • 7.9.2 网络层
      • 7.9.3 网络模型
      • 7.9.4 网络训练
      • 7.9.5 网络性能

The longer you can look back, the farther you can look forward. - 丘吉尔

第 6 章我们已经系统地介绍完基础的神经网络算法:从输入和输出的表示开始,介绍感知机的模型,介绍多输入、多输出的全连接网络层,然后扩展至多层神经网络;介绍了针对不同的问题场景下输出层的设计,最后介绍常用的损失函数,及实现方法。

本章我们将从理论层面学习神经网络中的核心算法之一:反向传播算法(Backpropagation,BP)。实际上,反向传播算法在 1960 年代早期就已经被提出,然而并没有引起业界重视。1970 年,Seppo Linnainmaa 在其硕士论文中提出了自动链式求导方法,并将反向传播算法实现在计算机上。1974 年,Paul Werbos 在其博士论文中首次提出了将反向传播算法应用到神经网络的可能性,但遗憾的是,Paul Werbos 并没有后续的相关研究发表。实际上,Paul Werbos 认为,这种研究思路对解决感知机问题是有意义的,但是由于人工智能寒冬,这个圈子大体已经失去解决那些问题的信念。直到 10 年后,1986 年,Geoffrey Hinton 等人在神经网络上应用反向传播算法 (Rumelhart, Hinton, & Williams,1986),使得反向传播算法在神经网络中焕发出勃勃生机。

有了深度学习框架自动求导、自动更新权值的功能,算法设计者几乎不需要对反向传播算法有深入的了解也可以搭建复杂的模型和网络,通过优化工具方便地训练网络模型。但是反向传播算法和梯度下降算法是神经网络的核心算法,深刻理解其工作原理十分重要。我们先回顾导数、梯度等数学概念,然后推导常用激活函数、损失函数的梯度形式,并开始逐渐推导感知机、多层神经网络的梯度传播方式。

7.1 导数与梯度

在高中阶段,我们就接触到导数(Derivative)的概念,它被定义为自变量在 x 0 x_{0} x0处产生一个微小扰动∆后,函数输出值的增量∆与自变量增量∆的比值在∆趋于 0 时的极限,如果存在,即为在0处的导数:
a = lim ⁡ Δ x → 0 Δ y Δ x = lim ⁡ Δ x → 0 f ( x + x 0 ) − f ( x ) Δ x a=\lim _{\Delta x \rightarrow 0} \frac{\Delta y}{\Delta x}=\lim _{\Delta x \rightarrow 0} \frac{f\left(x+x_{0}\right)-f(x)}{\Delta x} a=Δx0limΔxΔy=Δx0limΔxf(x+x0)f(x)

函数的导数可以记为 f ′ ( x ) f^{\prime}(x) f(x) d y d x \frac{d y}{d x} dxdy。从几何角度来看,一元函数在某处的导数就是函数的切线在此处的斜率,函数值沿着方向的变化率。考虑物理学中例子:自由落体运动的位移函
数的表达式 y = 1 2 g t 2 y=\frac{1}{2} g t^{2} y=21gt2,位移对时间的导数 d y d t = d 1 2 g t 2 d t = g t \frac{d y}{d t}=\frac{d \frac{1}{2} g t^{2}}{d t}=g t dtdy=dtd21gt2=gt,考虑到速度定义为位移的变化率,因此 = gt,位移对时间的导数即为速度。

实际上,(方向)导数是一个非常宽泛的概念,只是因为我们以前接触到的函数大多是一元函数,自变量∆只有 2 个方向:+,−。当函数的自变量大于一个时,函数的导数拓展为函数值沿着任意∆方向的变化率。导数本身是标量,没有方向,但是导数表征了函数值在某个方向∆的变化率。在这些任意∆方向中,沿着坐标轴的几个方向比较特殊,此时的导数也叫做偏导数(Partial Derivative)。对于一元函数,导数记为 d y d x \frac{d y}{d x} dxdy;对于多元函数的偏导数,记为 ∂ y ∂ x 1 , ∂ y ∂ x 2 , … \frac{\partial y}{\partial x_{1}}, \frac{\partial y}{\partial x_{2}}, \ldots x1y,x2y,等。偏导数是导数的特例,也没有方向。

考虑为多元函数的神经网络模型,比如 shape 为[784, 256]的权值矩阵 W,它包含了784×256 个连接权值,我们需要求出 784*256 个偏导数。需要注意的是,在数学公式中,我们一般要讨论的自变量是,但是在神经网络中,一般用来表示输入,比如图片,文本,语音数据等,网络的自变量是网络参数集 = {1, 1, 2, 2, … },我们关心的也是误差对网络参数的偏导数 ∂ L ∂ w 1 , ∂ L ∂ b 1 \frac{\partial \mathcal{L}}{\partial w_{1}}, \frac{\partial \mathcal{L}}{\partial b_{1}} w1L,b1L等。它其实就是函数输出ℒ沿着某个自变量变化方向上的导数,即偏导数。利用梯度下降算法优化网络时,需要求出网络的所有偏导数。我们把函数所有偏导数写成向量形式:
∇ L = ( ∂ L ∂ θ 1 , ∂ L ∂ θ 2 , ∂ L ∂ θ 3 , … ∂ L ∂ θ n ) \nabla \mathcal{L}=\left(\frac{\partial \mathcal{L}}{\partial \theta_{1}}, \frac{\partial \mathcal{L}}{\partial \theta_{2}}, \frac{\partial \mathcal{L}}{\partial \theta_{3}}, \quad \ldots \frac{\partial \mathcal{L}}{\partial \theta_{n}}\right) L=(θ1L,θ2L,θ3L,θnL)
此时梯度下降算法可以按着向量形式进行更新:
θ ′ = θ − η ∗ ∇ L \theta^{\prime}=\theta-\eta * \nabla \mathcal{L} θ=θηL
梯度下降算法一般是寻找函数的最小值,有时希望求解函数的最大值,如增强学习中希望最大化奖励函数,则可按着梯度方向更新
θ ′ = θ + η ∗ ∇ L \theta^{\prime}=\theta+\eta * \nabla \mathcal{L} θ=θ+ηL
即可,这种方式的更新称为梯度上升算法。其中向量 ( ∂ L ∂ θ 1 , ∂ L ∂ θ 2 , ∂ L ∂ θ 3 , … ∂ L ∂ θ n ) \left(\frac{\partial \mathcal{L}}{\partial \theta_{1}}, \frac{\partial \mathcal{L}}{\partial \theta_{2}}, \frac{\partial \mathcal{L}}{\partial \theta_{3}}, \dots \frac{\partial \mathcal{L}}{\partial \theta_{n}}\right) (θ1L,θ2L,θ3L,θnL)叫做函数的梯度(Gradient),它由所有偏导数组成,表征方向,梯度的方向表示函数值上升最快的方向,梯度的反向表示函数值下降最快的方向

通过梯度下降算法并不能保证得到全局最优解,这主要是目标函数的非凸性造成的。考虑图 7.1 非凸函数,深蓝色区域为极小值区域,不同的优化轨迹可能得到不同的最优数值解。
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第1张图片
神经网络的模型表达式非常复杂,模型参数量可达千万、数亿级别,几乎所有的神经网络的优化问题都是依赖于深度学习框架去自动计算每层的梯度,然后采用梯度下降算法循环迭代优化网络的参数直至性能满足需求

在介绍多层神经网络的反向传播算法之前,我们先介绍导数的常见属性,常见激活函数、损失函数的梯度推导,然后再推导多层神经网络的梯度传播规律。

7.2 导数常见性质

本节介绍常见的求导法则和样例,为神经网络函数求导铺垫。

7.2.1 基本函数的导数

❑ 常数函数 c 导数为 0,如 = 2函数导数′ = 0

❑ 线性函数 = ∗ + 导数为,如函数 = 2 ∗ + 1导数′ = 2

❑ 幂函数 x a x^{a} xa导数为 ∗ x a − 1 x^{a-1} xa1,如 = x 2 x^{2} x2函数′ = 2 ∗

❑ 指数函数 a x a^{x} ax 导数为 ∗ ,如 = e x e^{x} ex函数′ = e x e^{x} ex ∗ = e x e^{x} ex

❑ 对数函数 log ⁡ a x \log _{a} x logax导数为 1 x ln ⁡ a \frac{1}{x \ln a} xlna1,如 = 函数′ = 1 x ln ⁡ e \frac{1}{x \ln e} xlne1

7.2.2 常用导数性质

❑ 函数加减 ( + )′ = ′ + ′

❑ 函数相乘 ()′ = ′ ∗ + ∗ ′

❑ 函数相除 ( f g ) ′ = f ′ g − f g ′ g 2 , g ≠ 0 \left(\frac{f}{g}\right)^{\prime}=\frac{f^{\prime} g-f g^{\prime}}{g^{2}}, g \neq 0 (gf)=g2fgfg,g=0

❑ 复合函数的导数 考虑复合函数(()),令 = (),其导数
d f ( g ( x ) ) d x = d f ( u ) d u d g ( x ) d x = f ′ ( u ) ∗ g ′ ( x ) \frac{d f(g(x))}{d x}=\frac{d f(u)}{d u} \frac{d g(x)}{d x}=f^{\prime}(u) * g^{\prime}(x) dxdf(g(x))=dudf(u)dxdg(x)=f(u)g(x)

7.2.3 导数求解实战

考虑目标函数 L = x ⋅ w 2 + b 2 \mathcal{L}=x \cdot w^{2}+b^{2} L=xw2+b2,则
∂ L ∂ w = ∂ x ∗ w 2 ∂ w = x ∗ 2 w ∂ L ∂ b = ∂ b 2 ∂ b = 2 b \begin{array}{c} \frac{\partial \mathcal{L}}{\partial w}=\frac{\partial x * \mathrm{w}^{2}}{\partial w}=x * 2 w \\ \frac{\partial \mathcal{L}}{\partial b}=\frac{\partial b^{2}}{\partial b}=2 b \end{array} wL=wxw2=x2wbL=bb2=2b
考虑目标函数 L = x ⋅ e w + e b \mathcal{L}=x \cdot e^{w}+e^{b} L=xew+eb,则
∂ L ∂ w = ∂ x ∗ e w ∂ w = x ∗ e w ∂ L ∂ b = ∂ e b ∂ b = e b \begin{array}{c} \frac{\partial \mathcal{L}}{\partial w}=\frac{\partial x * \mathrm{e}^{\mathrm{w}}}{\partial w}=x * \mathrm{e}^{\mathrm{w}} \\ \frac{\partial \mathcal{L}}{\partial b}=\frac{\partial e^{b}}{\partial b}=e^{b} \end{array} wL=wxew=xewbL=beb=eb
考虑目标函数 L = [ y − ( x w + b ) ] 2 = [ ( x w + b ) − y ] 2 \mathcal{L}=[y-(x w+b)]^{2}=[(x w+b)-y]^{2} L=[y(xw+b)]2=[(xw+b)y]2,令 = + − y
∂ L ∂ w = 2 g ∗ ∂ g ∂ w = 2 g ∗ x = 2 ( x w + b − y ) ∗ x \frac{\partial \mathcal{L}}{\partial w}=2 \mathrm{g} * \frac{\partial \mathrm{g}}{\partial w}=2 g * x=2(x w+b-y) * x wL=2gwg=2gx=2(xw+by)x
∂ L ∂ b = 2 g ∗ ∂ g ∂ b = 2 g ∗ 1 = 2 ( x w + b − y ) \frac{\partial \mathcal{L}}{\partial b}=2 \mathrm{g} * \frac{\partial \mathrm{g}}{\partial b}=2 g * 1=2(x w+b-y) bL=2gbg=2g1=2(xw+by)
考虑目标函数ℒ = ( + ),令 = + ,则
∂ L ∂ w = a ∗ 1 g ∗ ∂ g ∂ w = a x w + b ∗ x ∂ L ∂ b = a ∗ 1 g ∗ ∂ g ∂ b = a x w + b \begin{array}{c} \frac{\partial \mathcal{L}}{\partial w}=\mathrm{a} * \frac{1}{g} * \frac{\partial \mathrm{g}}{\partial w}=\frac{a}{x w+b} * x \\ \frac{\partial \mathcal{L}}{\partial b}=\mathrm{a} * \frac{1}{g} * \frac{\partial \mathrm{g}}{\partial b}=\frac{a}{x w+b} \end{array} wL=ag1wg=xw+baxbL=ag1bg=xw+ba

7.3 激活函数导数

7.3.1 Sigmoid 函数导数

回顾 Sigmoid 函数表达式:
σ ( x ) = 1 1 + e − x \sigma(x)=\frac{1}{1+e^{-x}} σ(x)=1+ex1
我们来推导 Sigmoid 函数的导数表达式:
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第2张图片

可以看到,Sigmoid 函数的导数表达式最终可以表达为激活函数的输出值的简单运算,利用这一性质,在神经网络的梯度计算中,通过缓存每层的 Sigmoid 函数输出值,即可在需要的时候计算出其导数。Sigmoid 函数的导数曲线如图 7.2 所示。
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第3张图片
为了不使用 TensorFlow 的自动求导功能,本章将使用 Numpy 实现一个通过反向传播算法优化的多层神经网络,因此本章的实现部分我们使用 Numpy 演示,实现 Sigmoid 函数的导数:

# 导入 numpy 库
import numpy as np 
from matplotlib import pyplot as plt
plt.rcParams['font.size'] = 16
plt.rcParams['font.family'] = ['STKaiti']
plt.rcParams['axes.unicode_minus'] = False

def set_plt_ax():
    # get current axis 获得坐标轴对象
    ax = plt.gca()                                           

    ax.spines['right'].set_color('none') 
    # 将右边 上边的两条边颜色设置为空 其实就相当于抹掉这两条边
    ax.spines['top'].set_color('none')         

    ax.xaxis.set_ticks_position('bottom')   
    # 指定下边的边作为 x 轴,指定左边的边为 y 轴
    ax.yaxis.set_ticks_position('left') 

    # 指定 data  设置的bottom(也就是指定的x轴)绑定到y轴的0这个点上
    ax.spines['bottom'].set_position(('data', 0)) 
    ax.spines['left'].set_position(('data', 0))

def sigmoid(x): 
    # 实现 sigmoid 函数
    return 1 / (1 + np.exp(-x))

def sigmoid_derivative(x): 
    # sigmoid 导数的计算
    # sigmoid 函数的表达式由手动推导而得
    return sigmoid(x)*(1-sigmoid(x))
x = np.arange(-6.0, 6.0, 0.1)
sigmoid_y = sigmoid(x)
sigmoid_derivative_y = sigmoid_derivative(x)

set_plt_ax()
plt.plot(x, sigmoid_y, color='C9', label='Sigmoid')
plt.plot(x, sigmoid_derivative_y, color='C4', label='导数')
plt.xlim(-6, 6)
plt.ylim(0, 1)
plt.legend(loc=2)
plt.show()

7.3.2 ReLU 函数导数

回顾 ReLU 函数的表达式:
ReLU ⁡ ( x ) : = max ⁡ ( 0 , x ) \operatorname{ReLU}(x):=\max (0, x) ReLU(x):=max(0,x)
它的导数推导非常简单,直接可得
d d x Re ⁡ L U = { 1 x ≥ 0 0 x < 0 \frac{d}{d x} \operatorname{Re} L U=\left\{\begin{array}{ll} 1 & x \geq 0 \\ 0 & x<0 \end{array}\right. dxdReLU={ 10x0x<0
可以看到,ReLU 函数的导数计算简单,x 大于等于零的时候,导数值恒为 1,在反向传播的时候,它既不会放大梯度,造成梯度爆炸(Gradient exploding);也不会缩小梯度,造成梯度弥散(Gradient vanishing)。ReLU 函数的导数曲线如图 7.3 所示。
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第4张图片
在 ReLU 函数被广泛应用之前,神经网络中激活函数采用 Sigmoid 居多,但是 Sigmoid函数容易出现梯度弥散现象,当网络的层数增加后,较前层的参数由于梯度值非常微小,参数长时间得不到(有效)更新,无法训练深层神经网络,导致神经网络的研究一直停留在浅层;随着 ReLU 函数的提出,很好的解决了梯度弥散的现象,神经网络的层数能够地达到较深层数,如 AlexNet 中采用了 ReLU 激活函数,层数达到了 8 层,后续提出的上百层的卷积神经网络也多是采用 ReLU 激活函数。

通过 Numpy,我们可以方便地实现 ReLU 函数的导数:

def relu(x):
    return np.maximum(0, x)

def relu_derivative(x): # ReLU 函数的导数
    d = np.array(x, copy=True) # 用于保存梯度的张量
    d[x < 0] = 0 # 元素为负的导数为 0
    d[x >= 0] = 1 # 元素为正的导数为 1
    return d
x = np.arange(-6.0, 6.0, 0.1)
relu_y = relu(x)
relu_derivative_y = relu_derivative(x)

set_plt_ax()
plt.plot(x, relu_y, color='C9', label='ReLU')
plt.plot(x, relu_derivative_y, color='C4', label='导数')
plt.xlim(-6, 6)
plt.ylim(0, 6)
plt.legend(loc=2)
plt.show()

7.3.3 LeakyReLU 函数导数

回顾 LeakyReLU 函数的表达式:
 LeakyReLU  = { x x ≥ 0 p ∗ x x ≤ 0 \text { LeakyReLU }=\left\{\begin{array}{cl} x & x \geq 0 \\ p * x & x \leq 0 \end{array}\right.  LeakyReLU ={ xpxx0x0
它的导数可以推导为
d d x  LeakyReLU  = { 1 x ≥ 0 p x < 0 \frac{d}{d x} \text { LeakyReLU }=\left\{\begin{array}{ll} 1 & x \geq 0 \\ p & x<0 \end{array}\right. dxd LeakyReLU ={ 1px0x<0
它和 ReLU 函数的不同之处在于,当 x 小于零时,LeakyReLU 函数的导数值并不为 0,而是,p 一般设置为一个较小的数值,如 0.01 或 0.02,LeakyReLU 函数的导数曲线如图 7.4所示。
在这里插入图片描述
LeakyReLU 函数有效的克服了 ReLU 函数的缺陷,使用也比较广泛。我们可以通过Numpy 实现 LeakyReLU 函数的导数如下:

def leakyrelu(x, p):
    y = np.copy(x)
    y[y < 0] = p * y[y < 0]
    return y

# 其中 p 为 LeakyReLU 的负半段斜率,为超参数
def leakyrelu_derivative(x, p):
    dx = np.ones_like(x) # 创建梯度张量,全部初始化为 1
    dx[x < 0] = p # 元素为负的导数为 p
    return dx
x = np.arange(-6.0, 6.0, 0.1)
p = 0.1
leakyrelu_y = leakyrelu(x, p)
leakyrelu_derivative_y = leakyrelu_derivative(x, p)

set_plt_ax()
plt.plot(x, leakyrelu_y, color='C9', label='LeakyReLU')
plt.plot(x, leakyrelu_derivative_y, color='C4', label='导数')
plt.xlim(-6, 6)
plt.yticks(np.arange(-1, 7))
plt.legend(loc=2)
plt.show()

7.3.4 Tanh 函数梯度

回顾 tanh 函数的表达式:
tanh ⁡ ( x ) = ( e x − e − x ) ( e x + e − x ) \tanh (x)=\frac{\left(e^{x}-e^{-x}\right)}{\left(e^{x}+e^{-x}\right)} tanh(x)=(ex+ex)(exex)

= 2 ∗ sigmoid ⁡ ⁡ ( 2 x ) − 1 =2 * \operatorname{sigmoid} \operatorname{} (2 x)-1 =2sigmoid(2x)1
它的导数推导为
d d x tanh ⁡ ( x ) = ( e x + e − x ) ( e x + e − x ) − ( e x − e − x ) ( e x − e − x ) ( e x + e − x ) 2 = 1 − ( e x − e − x ) 2 ( e x + e − x ) 2 = 1 − tanh ⁡ 2 ( x ) \begin{aligned} \frac{d}{d x} \tanh (x) &=\frac{\left(e^{x}+e^{-x}\right)\left(e^{x}+e^{-x}\right)-\left(e^{x}-e^{-x}\right)\left(e^{x}-e^{-x}\right)}{\left(e^{x}+e^{-x}\right)^{2}} \\ &=1-\frac{\left(e^{x}-e^{-x}\right)^{2}}{\left(e^{x}+e^{-x}\right)^{2}}=1-\tanh ^{2}(x) \end{aligned} dxdtanh(x)=(ex+ex)2(ex+ex)(ex+ex)(exex)(exex)=1(ex+ex)2(exex)2=1tanh2(x)
tanh 函数及其导数曲线如图 7.5 所示。
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第5张图片
在 Numpy 中,可以实现 Tanh 函数的导数如下:

def sigmoid(x): # sigmoid 函数实现
    return 1 / (1 + np.exp(-x))
    
def tanh(x): # tanh 函数实现
    return 2*sigmoid(2*x) - 1
    
def tanh_derivative(x): # tanh 导数实现
    return 1-tanh(x)**2
x = np.arange(-6.0, 6.0, 0.1)
tanh_y = tanh(x)
tanh_derivative_y = tanh_derivative(x)

set_plt_ax()
plt.plot(x, tanh_y, color='C9', label='Tanh')
plt.plot(x, tanh_derivative_y, color='C4', label='导数')
plt.xlim(-6, 6)
plt.ylim(-1.5, 1.5)
plt.legend(loc=2)
plt.show()

7.4 损失函数梯度

前面已经介绍了常见的损失函数,我们这里推导均方误差损失函数交叉熵损失函数的梯度表达式。

7.4.1 均方误差函数梯度

均方差损失函数表达式为:
L = 1 2 ∑ k = 1 K ( y k − o k ) 2 \mathcal{L}=\frac{1}{2} \sum_{k=1}^{K}\left(y_{k}-o_{k}\right)^{2} L=21k=1K(ykok)2

则它的偏导数 ∂ L ∂ o i \frac{\partial \mathcal{L}}{\partial o_{i}} oiL可以展开为

∂ L ∂ o i = 1 2 ∑ k = 1 K ∂ ∂ o i ( y k − o k ) 2 \frac{\partial \mathcal{L}}{\partial o_{i}}=\frac{1}{2} \sum_{k=1}^{K} \frac{\partial}{\partial o_{i}}\left(y_{k}-o_{k}\right)^{2} oiL=21k=1Koi(ykok)2

利用链式法则分解为
∂ L ∂ o i = 1 2 ∑ k = 1 K 2 ∗ ( y k − o k ) ∗ ∂ ( y k − o k ) ∂ o i \frac{\partial \mathcal{L}}{\partial o_{i}}=\frac{1}{2} \sum_{k=1}^{K} 2 *\left(y_{k}-o_{k}\right) * \frac{\partial\left(y_{k}-o_{k}\right)}{\partial o_{i}} oiL=21k=1K2(ykok)oi(ykok)

∂ L ∂ o i = ∑ k = 1 K ( y k − o k ) ⋅ − 1 ⋅ ∂ o k ∂ o i = ∑ k = 1 K ( o k − y k ) ⋅ ∂ o k ∂ o i \begin{aligned} \frac{\partial \mathcal{L}}{\partial o_{i}} &=\sum_{k=1}^{K}\left(y_{k}-o_{k}\right) \cdot-1 \cdot \frac{\partial o_{k}}{\partial o_{i}} \\ &=\sum_{k=1}^{K}\left(o_{k}-y_{k}\right) \cdot \frac{\partial o_{k}}{\partial o_{i}} \end{aligned} oiL=k=1K(ykok)1oiok=k=1K(okyk)oiok
∂ o k ∂ o i \frac{\partial o_{k}}{\partial o_{i}} oiok仅当 = 时才为 1,其他点都为 0,也就是说, ∂ L ∂ θ i \frac{\partial L}{\partial \theta_{i}} θiL只与第 i i i号节点相关,与其他节点无关。因此上式中的求和符号可以去掉。均方差的导数可以推导为:
∂ L ∂ o i = ( o i − y i ) \frac{\partial \mathcal{L}}{\partial o_{i}}=\left(o_{i}-y_{i}\right) oiL=(oiyi)

7.4.2 交叉熵函数梯度

可以参考文章:Softmax函数详解以及求导过程

在计算交叉熵损失函数时,一般将Softmax函数与交叉熵函数统一实现。我们先推导Softmax 函数的梯度,再推导交叉熵函数的梯度。

7.4.2.1Softmax 梯度

回顾 Softmax 函数的表达式:

p i = e z i ∑ k = 1 K e z k p_{i}=\frac{e^{z_{i}}}{\sum_{k=1}^{K} e^{z_{k}}} pi=k=1Kezkezi

它的功能是将 K K K个输出节点的值转换为概率,并保证概率之和为 1,如图 7.6 所示。

TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第6张图片
回顾
f ( x ) = g ( x ) h ( x ) f(x)=\frac{g(x)}{h(x)} f(x)=h(x)g(x)

函数的导数

f ′ ( x ) = g ′ ( x ) h ( x ) − h ′ ( x ) g ( x ) h ( x ) 2 f^{\prime}(x)=\frac{g^{\prime}(x) h(x)-h^{\prime}(x) g(x)}{h(x)^{2}} f(x)=h(x)2g(x)h(x)h(x)g(x)
对于 Softmax 函数, g ( x ) = e z i , h ( x ) = ∑ k = 1 K e z k g(x)=e^{z_{i}}, h(x)=\sum_{k=1}^{K} e^{z_{k}} g(x)=ezi,h(x)=k=1Kezk,下面我们根据 = 时和 ≠ 来分别推导Softmax 函数的梯度。

❑ 当 = 时 Softmax 函数的偏导数可以展开为

TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第7张图片
提取公共项 e Z i e^{Z_{i}} eZi=
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第8张图片
拆分为 2 部分=
在这里插入图片描述
可以看到,它们即是 2 个概率值的相乘,同时 = j := (1 − j )

❑ 当 ≠ 时 展开 Softmax 函数为

∂ e z i ∑ k = 1 K e z k ∂ z j = 0 − e z j e z i ( ∑ k = 1 K e z k ) 2 \frac{\partial \frac{e^{z_{i}}}{\sum_{k=1}^{K} e^{z_{k}}}}{\partial z_{j}}=\frac{0-e^{z_{j}} e^{z_{i}}}{\left(\sum_{k=1}^{K} e^{z_{k}}\right)^{2}} zjk=1Kezkezi=(k=1Kezk)20ezjezi
去掉 0 项,并分解为 2 项
= − e z j ∑ k = 1 K e z k × e z i ∑ k = 1 K e z k = − p j ⋅ p i \begin{array}{c} =\frac{-e^{z_{j}}}{\sum_{k=1}^{K} e^{z_{k}}} \times \frac{e^{z_{i}}}{\sum_{k=1}^{K} e^{z_{k}}} \\ =-p_{j} \cdot p_{i} \end{array} =k=1Kezkezj×k=1Kezkezi=pjpi
可以看到,虽然 Softmax 函数的梯度推导稍复杂,但是最终的结果还是很简洁的:
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第9张图片

7.4.2.2交叉熵梯度

考虑交叉熵损失函数的表达式
L = − ∑ k y k log ⁡ ( p k ) \mathcal{L}=-\sum_{k} y_{k} \log \left(p_{k}\right) L=kyklog(pk)
我们直接来推导最终损失函数对网络输出 logits 变量 z i z_{i} zi的偏导数,展开为
∂ L ∂ z i = − ∑ k y k ∂ log ⁡ ( p k ) ∂ z i \frac{\partial \mathcal{L}}{\partial z_{i}}=-\sum_{k} y_{k} \frac{\partial \log \left(p_{k}\right)}{\partial z_{i}} ziL=kykzilog(pk)

将 ℎ复合函数利用链式法则分解为
= − ∑ k y k ∂ log ⁡ ( p k ) ∂ p k × ∂ p k ∂ z i =-\sum_{k} y_{k} \frac{\partial \log \left(p_{k}\right)}{\partial p_{k}} \times \frac{\partial p_{k}}{\partial z_{i}} =kykpklog(pk)×zipk

= − ∑ k y k 1 p k × ∂ p k ∂ z i =-\sum_{k} y_{k} \frac{1}{p_{k}} \times \frac{\partial p_{k}}{\partial z_{i}} =kykpk1×zipk
其中 ∂ p k ∂ z i \frac{\partial p_{k}}{\partial z_{i}} zipk即为我们已经推导的 Softmax 函数的偏导数。

将求和符号分开为 = 以及 ≠ 的 2 种情况,并代入 ∂ p k ∂ z i \frac{\partial p_{k}}{\partial z_{i}} zipk求解的公式,可得
∂ L ∂ z i = − y i ( 1 − p i ) − ∑ k ≠ i y k 1 p k ( − p k ⋅ p i ) \frac{\partial \mathcal{L}}{\partial z_{i}}=-y_{i}\left(1-p_{i}\right)-\sum_{k \neq i} y_{k} \frac{1}{p_{k}}\left(-p_{k} \cdot p_{i}\right) ziL=yi(1pi)k=iykpk1(pkpi)
进一步化简为
= − y i ( 1 − p i ) + ∑ k ≠ i y k ⋅ p i = − y i + y i p i + ∑ k ≠ i y k ⋅ p i \begin{aligned} &=-y_{i}\left(1-p_{i}\right)+\sum_{k \neq i} y_{k} \cdot p_{i}\\ &=-y_{i}+y_{i} p_{i}+\sum_{k \neq i} y_{k} \cdot p_{i} \end{aligned} =yi(1pi)+k=iykpi=yi+yipi+k=iykpi
提供公共项 p i p_{i} pi
= p i ( y i + ∑ k ≠ i y k ) − y i =p_{i}\left(y_{i}+\sum_{k \neq i} y_{k}\right)-y_{i} =piyi+k=iykyi
完成交叉熵函数的梯度推导。

特别地,对于分类问题中 y 通过 one-hot 编码的方式,则有
∑ k y k = 1 y i + ∑ k ≠ i y k = 1 \begin{aligned} &\sum_{k} y_{k}=1\\ &y_{i}+\sum_{k \neq i} y_{k}=1 \end{aligned} kyk=1yi+k=iyk=1
因此交叉熵的偏导数可以进一步简化为
∂ L ∂ z i = p i − y i \frac{\partial \mathcal{L}}{\partial \mathbf{z}_{i}}=p_{i}-y_{i} ziL=piyi

7.5 全连接层梯度

在介绍完梯度的基础知识后,我们正式地进入到神经网络的反向传播算法的推导中。实际使用中的神经网络的结构多种多样,我们将以全连接层,激活函数采用 Sigmoid 函数,误差函数为 Softmax+MSE 损失函数的神经网络为例,推导其梯度传播方式。

7.5.1 单个神经元梯度

对于采用 Sigmoid 激活函数的神经元模型,它的数学模型可以写为

o 1 = σ ( w 1 x + b 1 ) o^{1}=\sigma\left(w^{1} x+b^{1}\right) o1=σ(w1x+b1)

其中变量的上标表示层数,如 o 1 o^{1} o1表示第一个隐藏层的输出,表示网络的输入,我们以权值的偏导数推导为例。

为了方便演示,我们将神经元模型绘制如图 7.7 所示,图中未画出偏置,输入节点数为J。其中输入第个节点到输出 o 1 o^{1} o1的权值连接记为 w j 1 1 w_{j1}^{1} wj11上标表示权值属于的层数下标表示当前连接的起始节点号和终止节点号如下标1表示上一层的第号节点到当前层的 1 号节点。经过激活函数之前的变量叫做 z 1 1 z_{1}^{1} z11,经过激活函数之后的变量叫 o 1 o^{1} o1,由于只有一个输出节点, o 1 1 o^{1}_{1} o11 = o 1 o^{1} o1。输出与真实标签之间计算误差,误差记为ℒ。
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第10张图片
如果我们采用均方差误函数,考虑到单个神经元只有一个输出 o 1 1 o^{1}_{1} o11,那么损失可以表
达为
L = 1 2 ( o 1 1 − t ) 2 \mathcal{L}=\frac{1}{2}\left(o_{1}^{1}-t\right)^{2} L=21(o11t)2
其中为真实标签值,添加1/2并不影响梯度的方向,计算更简便。我们以权值连接的第 ∈[1,]号节点的权值 w j 1 w_{j1} wj1为例,考虑损失函数ℒ对其的偏导数 ∂ L ∂ w j 1 \frac{\partial \mathcal{L}}{\partial w_{j 1}} wj1L:
∂ L ∂ w j 1 = ( o 1 − t ) ∂ o 1 ∂ w j 1 \frac{\partial \mathcal{L}}{\partial w_{j 1}}=\left(o_{1}-t\right) \frac{\partial o_{1}}{\partial w_{j 1}} wj1L=(o1t)wj1o1
o 1 = σ ( z 1 ) o_{1}=\sigma\left(z_{1}\right) o1=σ(z1)分解,考虑到 Sigmoid 函数的导数 σ ′ = σ ( 1 − σ ) \sigma^{\prime}=\sigma(1-\sigma) σ=σ(1σ)
∂ L ∂ w j 1 = ( o 1 − t ) ∂ σ ( z 1 ) ∂ w j 1 = ( o 1 − t ) σ ( z 1 ) ( 1 − σ ( z 1 ) ) ∂ z 1 1 ∂ w j 1 \begin{aligned} &\frac{\partial \mathcal{L}}{\partial w_{j 1}}=\left(o_{1}-t\right) \frac{\partial \sigma\left(z_{1}\right)}{\partial w_{j 1}}\\ =&\left(o_{1}-t\right) \sigma\left(z_{1}\right)\left(1-\sigma\left(z_{1}\right)\right) \frac{\partial z_{1}^{1}}{\partial w_{j 1}} \end{aligned} =wj1L=(o1t)wj1σ(z1)(o1t)σ(z1)(1σ(z1))wj1z11
σ ( z 1 ) \sigma\left(z_{1}\right) σ(z1)写成 o 1 o_{1} o1,继续推导 ∂ L ∂ w j 1 \frac{\partial \mathcal{L}}{\partial w_{j 1}} wj1L:

∂ L ∂ w j 1 = ( o 1 − t ) o 1 ( 1 − o 1 ) ∂ z 1 1 ∂ w j 1 \frac{\partial \mathcal{L}}{\partial w_{j 1}}=\left(o_{1}-t\right) o_{1}\left(1-o_{1}\right) \frac{\partial z_{1}^{1}}{\partial w_{j 1}} wj1L=(o1t)o1(1o1)wj1z11
考虑 ∂ z 1 1 ∂ w j 1 = x j \frac{\partial z_{1}^{1}}{\partial w_{j 1}}=x_{j} wj1z11=xj,可得:
∂ L ∂ w j 1 = ( o 1 − t ) o 1 ( 1 − o 1 ) x j \frac{\partial \mathcal{L}}{\partial w_{j 1}}=\left(o_{1}-t\right) o_{1}\left(1-o_{1}\right) x_{j} wj1L=(o1t)o1(1o1)xj
从上式可以看到,误差对权值 w j 1 w_{j1} wj1的偏导数只与输出值 o 1 o_{1} o1、真实值以及当前权值连接的输入 x j x_{j} xj有关。

7.5.2 全连接层梯度

我们把单个神经元模型推广到单层全连接层的网络上,如图 7.8 所示。输入层通过一个全连接层得到输出向量1,与真实标签向量计算均方误差。输入节点数为 J J J,输出节点数为 K K K
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第11张图片
多输出的全连接网络层模型与单个神经元模型不同之处在于,它多了很多的输出节点 o 1 1 , o 2 1 , o 3 1 , … , o K 1 o_{1}^{1}, o_{2}^{1}, o_{3}^{1}, \ldots, o_{K}^{1} o11,o21,o31,,oK1,同样的每个输出节点分别对应到真实标签 t 1 , t 2 , … , t K t_{1}, t_{2}, \dots, t_{K} t1,t2,,tK。均方误差可以表达为
L = 1 2 ∑ i = 1 K ( o i 1 − t i ) 2 \mathcal{L}=\frac{1}{2} \sum_{i=1}^{K}\left(o_{i}^{1}-t_{i}\right)^{2} L=21i=1K(oi1ti)2
由于 ∂ L ∂ w j k \frac{\partial \mathcal{L}}{\partial w_{j k}} wjkL只与节点 o k 1 o_{k}^{1} ok1有关联,上式中的求和符号可以去掉,即 = :
∂ L ∂ w j k = ( o k − t k ) ∂ o k ∂ w j k \frac{\partial \mathcal{L}}{\partial w_{j k}}=\left(o_{k}-t_{k}\right) \frac{\partial o_{k}}{\partial w_{j k}} wjkL=(oktk)wjkok
o k = σ ( z k ) o_{k}=\sigma\left(z_{k}\right) ok=σ(zk)代入
∂ L ∂ w j k = ( o k − t k ) ∂ σ ( z k ) ∂ w j k \frac{\partial \mathcal{L}}{\partial w_{j k}}=\left(o_{k}-t_{k}\right) \frac{\partial \sigma\left(z_{k}\right)}{\partial w_{j k}} wjkL=(oktk)wjkσ(zk)
考虑 Sigmoid 函数的导数′ = (1 − ):
∂ L ∂ w j k = ( o k − t k ) σ ( z k ) ( 1 − σ ( z k ) ) ∂ z k 1 ∂ w j k \frac{\partial \mathcal{L}}{\partial w_{j k}}=\left(o_{k}-t_{k}\right) \sigma\left(z_{k}\right)\left(1-\sigma\left(z_{k}\right)\right) \frac{\partial z_{k}^{1}}{\partial w_{j k}} wjkL=(oktk)σ(zk)(1σ(zk))wjkzk1
σ ( Z k ) \sigma\left(Z_{k}\right) σ(Zk)记为 O k O_{k} Ok
∂ L ∂ w j k = ( o k − t k ) o k ( 1 − o k ) ∂ z k 1 ∂ w j k \frac{\partial \mathcal{L}}{\partial w_{j k}}=\left(o_{k}-t_{k}\right) o_{k}\left(1-o_{k}\right) \frac{\partial z_{k}^{1}}{\partial w_{j k}} wjkL=(oktk)ok(1ok)wjkzk1
最终可得
∂ L ∂ w j k = ( o k − t k ) o k ( 1 − o k ) x j \frac{\partial \mathcal{L}}{\partial w_{j k}}=\left(o_{\mathrm{k}}-t_{k}\right) o_{k}\left(1-o_{k}\right) x_{j} wjkL=(oktk)ok(1ok)xj
由此可以看到,某条连接 W j k W_{jk} Wjk上面的连接,只与当前连接的输出节点 O k 1 O_{k}^{1} Ok1,对应的真实值节点的标签 t k 1 t_{k}^{1} tk1,以及对应的输入节点 X j X_{j} Xj有关。
我们令 δ k = ( o k − t k ) o k ( 1 − o k ) \delta_{k}=\left(o_{k}-t_{k}\right) o_{k}\left(1-o_{k}\right) δk=(oktk)ok(1ok),则 ∂ L ∂ w j k \frac{\partial \mathcal{L}}{\partial w_{j k}} wjkL可以表达为
∂ L ∂ w j k = δ k x j \frac{\partial \mathcal{L}}{\partial w_{j k}}=\delta_{k} x_{j} wjkL=δkxj
其中 δ k \delta_{k} δk变量表征连接线的终止节点的梯度传播的某种特性,使用 δ k \delta_{k} δk表示后, ∂ L ∂ w j k \frac{\partial \mathcal{L}}{\partial w_{j k}} wjkL偏导数只与当前连接的起始节点 x j x_{j} xj,终止节点处 δ k \delta_{k} δk有关,理解起来比较直观。后续我们将会在看到 δ k \delta_{k} δk在循环推导梯度中的作用。

现在我们已经推导完单层神经网络(即输出层)的梯度传播方式接下来我们尝试推导倒数第二层的梯度传播公式,完成了导数第二层的传播推导,可以循环往复推导所有隐藏层的梯度传播公式,从而完成所有参数的梯度计算。

在介绍反向传播算法之前,我们先学习一个导数传播的核心法则:链式法则

7.6 链式法则

前面我们介绍了输出层的梯度 ∂ L ∂ w j k \frac{\partial \mathcal{L}}{\partial w_{j k}} wjkL计算方法,我们现在来介绍链式法则,它是能在不显式推导神经网络的数学表达式的情况下,逐层推导梯度的核心公式,非常重要。
其实我们前面在推导梯度的过程中已经或多或少地用到了链式法则。考虑复合函数 y = f ( u ) , u = g ( x ) y=f(u), u=g(x) y=f(u),u=g(x),则 d y d x \frac{d y}{d x} dxdy可由 d y d u \frac{d y}{d u} dudy d u d x \frac{d u}{d x} dxdu推导出:
d y d x = d y d u ⋅ d u d x = f ′ ( g ( x ) ) ⋅ g ′ ( x ) \frac{d y}{d x}=\frac{d y}{d u} \cdot \frac{d u}{d x}=f^{\prime}(g(x)) \cdot g^{\prime}(x) dxdy=dudydxdu=f(g(x))g(x)
考虑多元复合函数, = (, ),其中 = (), = ℎ(),那么 d z d t \frac{d z}{d t} dtdz的导数可以由 d z d x \frac{d z}{d x} dxdz d z d y \frac{d z}{d y} dydz等推导出,具体表达为
d z d t = ∂ z ∂ x d x d t + ∂ z ∂ y d y d t \frac{d z}{d t}=\frac{\partial z}{\partial x} \frac{d x}{d t}+\frac{\partial z}{\partial y} \frac{d y}{d t} dtdz=xzdtdx+yzdtdy
例如, z = ( 2 t + 1 ) 2 + e t 2 z=(2 t+1)^{2}+e^{t^{2}} z=(2t+1)2+et2,令 = 2 + 1, y = t 2 y=t^{2} y=t2,则 z = x 2 + e y z=x^{2}+e^{y} z=x2+ey
d z d t = ∂ z ∂ x d x d t + ∂ z ∂ y d y d t = 2 x ∗ 2 + e y ∗ 2 t d z d t = 2 ( 2 t + 1 ) ∗ 2 + e t 2 ∗ 2 t d z d t = 4 ( 2 t + 1 ) + 2 t e t 2 \begin{aligned} \frac{d z}{d t}=\frac{\partial z}{\partial x} \frac{d x}{d t}+\frac{\partial z}{\partial y} \frac{d y}{d t}=2 x * 2+e^{y} * 2 t \\ \frac{d z}{d t}=2(2 t+1) * 2+e^{t^{2}} * 2 t \\ \frac{d z}{d t}=4(2 t+1)+2 t e^{t^{2}} \end{aligned} dtdz=xzdtdx+yzdtdy=2x2+ey2tdtdz=2(2t+1)2+et22tdtdz=4(2t+1)+2tet2
神经网络的损失函数ℒ来自于各个输出节点 O k K O_{k}^{K} OkK,如下图 7.9 所示,其中输出节点 O k K O_{k}^{K} OkK又与隐藏层的输出节点 O j J O_{j}^{J} OjJ相关联,因此链式法则非常适合于神经网络的梯度推导。让我们来考虑损失函数ℒ如何应用链式法则。
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第12张图片
前向传播时,数据经过 w i j J w_{ij}^{J} wijJ传到倒数第二层的节点 O j J O_{j}^{J} OjJ,再传播到输出层的节点 O k K O_{k}^{K} OkK。在每层只有一个节点时, ∂ L ∂ w j k 1 \frac{\partial \mathcal{L}}{\partial w_{j k}^{1}} wjk1L可以利用链式法则,逐层分解为:
∂ L ∂ w i j J = ∂ L ∂ o j J ∂ o j J ∂ w i j J = ∂ L ∂ o k K ∂ o k K ∂ o j J ∂ o j J ∂ w i j J \frac{\partial \mathcal{L}}{\partial w_{i j}^{J}}=\frac{\partial \mathcal{L}}{\partial o_{j}^{J}} \frac{\partial o_{j}^{J}}{\partial w_{i j}^{J}}=\frac{\partial \mathcal{L}}{\partial o_{k}^{K}} \frac{\partial o_{k}^{K}}{\partial o_{j}^{J}} \frac{\partial o_{j}^{J}}{\partial w_{i j}^{J}} wijJL=ojJLwijJojJ=okKLojJokKwijJojJ
其中 ∂ L ∂ o k K \frac{\partial \mathcal{L}}{\partial o_{k}^{K}} okKL可以由误差函数直接推导出, ∂ o k K ∂ o j J \frac{\partial o_{k}^{K}}{\partial o_{j}^{J}} ojJokK可以由全连接层公式推导出, ∂ o j J ∂ w i j J \frac{\partial o_{j}^{J}}{\partial w_{i j}^{J}} wijJojJ的导数即为输入 x i I x_{i}^{I} xiI。可以看到,通过链式法则,我们不需要显式计算 L = f ( w i j J ) \mathcal{L}=f\left(w_{i j}^{J}\right) L=f(wijJ)的具体数学表达式,直接可以将偏导数分解,层层迭代即可推导出。

我们简单使用 TensorFlow 来体验链式法则的魅力:

import tensorflow as tf
#构建待优化变量
x=tf.constant(1.)
w1=tf.constant(2.)
b1=tf.constant(1.)
w2=tf.constant(2.)
b2=tf.constant(1.)
# 构建梯度记录器
with tf.GradientTape(persistent=True) as tape:
    # 非tf.Variable类型的张量需要人为设置记录梯度信息
    tape.watch([w1,b1,w2,b2])
    #构建两层线性网络
    y1=x*w1+b1
    y2=y1*w2+b2

# 独立求解出各个偏导数
dy2_dy1=tape.gradient(y2,[y1])[0]
dy1_dw1=tape.gradient(y1,[w1])[0]
dy2_dw1=tape.gradient(y2,[w1])[0]

#验证连式法则
print(dy2_dy1*dy1_dw1)
print(dy2_dw1)

我们通过自动求导功能计算出 ∂ y 2 ∂ y 1 \frac{\partial y_{2}}{\partial y_{1}} y1y2 ∂ y 1 ∂ w 1 \frac{\partial y_{1}}{\partial w_{1}} w1y1 ∂ y 2 ∂ w 1 \frac{\partial y_{2}}{\partial w_{1}} w1y2,借助链式法则我们可以推断 ∂ y 2 ∂ y 1 \frac{\partial y_{2}}{\partial y_{1}} y1y2* ∂ y 1 ∂ w 1 \frac{\partial y_{1}}{\partial w_{1}} w1y1 ∂ y 2 ∂ w 1 \frac{\partial y_{2}}{\partial w_{1}} w1y2应该是相等的,他们的计算结果如下:

tf.Tensor(2.0, shape=(), dtype=float32)
tf.Tensor(2.0, shape=(), dtype=float32)

可以看到 ∂ y 2 ∂ y 1 ∗ ∂ y 1 ∂ w 1 = ∂ y 2 ∂ w 1 \frac{\partial y_{2}}{\partial y_{1}} * \frac{\partial y_{1}}{\partial w_{1}}=\frac{\partial y_{2}}{\partial w_{1}} y1y2w1y1=w1y2,偏导数的传播是符合链式法则的。

7.7 反向传播算法

现在我们来推导隐藏层的反向传播算法。简单回顾一下输出层的偏导数公式:
∂ L ∂ w j k = ( o k − t k ) o k ( 1 − o k ) x j = δ k x j \frac{\partial \mathcal{L}}{\partial w_{j k}}=\left(o_{k}-t_{k}\right) o_{k}\left(1-o_{k}\right) x_{j}=\delta_{k} x_{j} wjkL=(oktk)ok(1ok)xj=δkxj
考虑网络倒数第二层的偏导数 ∂ L ∂ w i j \frac{\partial \mathcal{L}}{\partial w_{i j}} wijL,如图 7.10 所示,输出层节点数为 K,输出为 O k = O^{k}= Ok= [ o 1 K , o 2 K , … , o K K ] \left[o_{1}^{K}, o_{2}^{K}, \ldots, o_{K}^{K}\right] [o1K,o2K,,oKK];倒数第二层节点数为 J J J,输出为 O j = O^{j}= Oj= [ o 1 j , o 2 j , … , o j j ] \left[o_{1}^{j}, o_{2}^{j}, \ldots, o_{j}^{j}\right] [o1j,o2j,,ojj];倒数第三层的节点数为 I I I,输出为 O i = O^{i}= Oi= [ o 1 i , o 2 i , … , o i i ] \left[o_{1}^{i}, o_{2}^{i}, \ldots, o_{i}^{i}\right] [o1i,o2i,,oii].
TensorFlow 2.0深度学习算法实战 第七章 反向传播算法_第13张图片
首先将均方差误差函数展开:
∂ L ∂ w i j = ∂ ∂ w i j 1 2 ∑ k ( o k − t k ) 2 \frac{\partial \mathcal{L}}{\partial w_{i j}}=\frac{\partial}{\partial w_{i j}} \frac{1}{2} \sum_{k}\left(o_{k}-t_{k}\right)^{2} wijL=wij21k(oktk)2

由于ℒ通过每个输出节点 o k o_{k} ok w i j w_{ij} wij 相关联,故此处不能去掉求和符号,运用链式法则将均方
差函数拆解:

∂ L ∂ w i j = ∑ k ( o k − t k ) ∂ ∂ w i j o k \frac{\partial \mathcal{L}}{\partial w_{i j}}=\sum_{k}\left(o_{k}-t_{k}\right) \frac{\partial}{\partial w_{i j}} o_{k} wijL=k(oktk)wijok

o k = σ ( z k ) o_{k}=\sigma\left(z_{k}\right) ok=σ(zk)代入

∂ L ∂ w i j = ∑ k ( o k − t k ) ∂ ∂ w i j σ ( z k ) \frac{\partial \mathcal{L}}{\partial w_{i j}}=\sum_{k}\left(o_{k}-t_{k}\right) \frac{\partial}{\partial w_{i j}} \sigma\left(z_{k}\right) wijL=k(oktk)wijσ(zk)

利用 Sigmoid 函数的导数′ = (1 − )进一步分解
∂ L ∂ w i j = ∑ k ( o k − t k ) ∂ ∂ w i j σ ( z k ) \frac{\partial \mathcal{L}}{\partial w_{i j}}=\sum_{k}\left(o_{k}-t_{k}\right) \frac{\partial}{\partial w_{i j}} \sigma\left(z_{k}\right) wijL=k(oktk)wijσ(zk)

σ ( z k ) \sigma\left(z_{k}\right) σ(zk)写回 o k o_{k} ok形式,并利用链式法则,将 ∂ z k ∂ w i j \frac{\partial z_{k}}{\partial w_{i j}} wijzk分解
∂ L ∂ w i j = ∑ k ( o k − t k ) o k ( 1 − o k ) ∂ z k ∂ o j ⋅ ∂ o j ∂ w i j \frac{\partial \mathcal{L}}{\partial w_{i j}}=\sum_{k}\left(o_{k}-t_{k}\right) o_{k}\left(1-o_{k}\right) \frac{\partial z_{k}}{\partial o_{j}} \cdot \frac{\partial o_{j}}{\partial w_{i j}} wijL=k(oktk)ok(1ok)ojzkwijoj

其中 ∂ z k ∂ o j = w j k \frac{\partial z_{k}}{\partial o_{j}}=w_{j k} ojzk=wjk,因此
∂ L ∂ w i j = ∑ k ( o k − t k ) o k ( 1 − o k ) w j k ∂ o j ∂ w i j \frac{\partial \mathcal{L}}{\partial w_{i j}}=\sum_{k}\left(o_{k}-t_{k}\right) o_{k}\left(1-o_{k}\right) w_{j k} \frac{\partial o_{j}}{\partial w_{i j}} wijL=k(oktk)ok(1ok)wjkwijoj

考虑到 ∂ o j ∂ w i j \frac{\partial o_{j}}{\partial w_{i j}} wijoj与 k 无关,可提取公共项
∂ L ∂ w i j = ∂ o j ∂ w i j ∑ k ( o k − t k ) o k ( 1 − o k ) w j k \frac{\partial \mathcal{L}}{\partial w_{i j}}=\frac{\partial o_{j}}{\partial w_{i j}} \sum_{k}\left(o_{k}-t_{k}\right) o_{k}\left(1-o_{k}\right) w_{j k} wijL=wijojk(oktk)ok(1ok)wjk

进一步利用 o j = σ ( z j ) o_{j}=\sigma\left(z_{j}\right) oj=σ(zj),并利用 Sigmoid 导数 σ ′ = σ ( 1 − σ ) \sigma^{\prime}=\sigma(1-\sigma) σ=σ(1σ) ∂ o j ∂ w i j \frac{\partial o_{j}}{\partial w_{i j}} wijoj拆分为
∂ L ∂ w i j = o j ( 1 − o j ) ∂ z j ∂ w i j ∑ k ( o k − t k ) o k ( 1 − o k ) w j k \frac{\partial \mathcal{L}}{\partial w_{i j}}=o_{j}\left(1-o_{j}\right) \frac{\partial z_{j}}{\partial w_{i j}} \sum_{k}\left(o_{k}-t_{k}\right) o_{k}\left(1-o_{k}\right) w_{j k} wijL=oj(1oj)wijzjk(oktk)ok(1ok)wjk

其中 ∂ z j ∂ w i j \frac{\partial z_{j}}{\partial w_{i j}} wijzj的导数可直接推导出为 O i O_{i} Oi,上式可写为
∂ L ∂ w i j = o j ( 1 − o j ) o i ∑ k ( o k − t k ) o k ( 1 − o k ) ⏟ δ k K w j k \frac{\partial \mathcal{L}}{\partial w_{i j}}=o_{j}\left(1-o_{j}\right) o_{i} \sum_{k} \underbrace{\left(o_{k}-t_{k}\right) o_{k}\left(1-o_{k}\right)}_{\delta_{k}^{K}} w_{j k} wijL=oj(1oj)oikδkK (oktk)ok(1ok)wjk
其中 δ k K = ( o k − t k ) o k ( 1 − o k ) \delta_{k}^{K}=\left(o_{k}-t_{k}\right) o_{k}\left(1-o_{k}\right) δkK=(oktk)ok(1ok),则 ∂ L ∂ w i j \frac{\partial \mathcal{L}}{\partial w_{i j}} wijL的表达式可简写为
∂ L ∂ w i j = o j ( 1 − o j ) o i ∑ k δ k K w j k \frac{\partial \mathcal{L}}{\partial w_{i j}}=o_{j}\left(1-o_{j}\right) o_{i} \sum_{\mathbf{k}} \delta_{k}^{K} w_{j k} wijL=oj(1oj)oikδkKwjk
我们仿照输出层 ∂ L ∂ w j k = δ k K x j \frac{\partial \mathcal{L}}{\partial w_{j k}}=\delta_{k}^{K} x_{j} wjkL=δkKxj的书写方式,将 δ j J \delta_{j}^{J} δjJ定义为
δ j J : = o j ( 1 − o j ) ∑ k δ k K w j k \delta_{j}^{J}:=o_{j}\left(1-o_{j}\right) \sum_{k} \delta_{k}^{K} w_{j k} δjJ:=oj(1oj)kδkKwjk

此时 ∂ L ∂ w i j \frac{\partial \mathcal{L}}{\partial w_{i j}} wijL可以写为当前连接的起始节点的输出值 o i o_{i} oi与终止节点 j 的梯度信息 δ j J \delta_{j}^{J} δjJ的简单相乘运算:
∂ L ∂ w i j = δ j J O i I \frac{\partial \mathcal{L}}{\partial w_{i j}}=\delta_{j}^{J} O_{i}^{I} wijL=δjJOiI
通过定义变量,每一层的梯度表达式变得更加清晰简洁,其中可以简单理解为当前连接 w i j w_{ij} wij对误差函数的贡献值。

我们来小结一下每层的偏导数的计算公式。

输出层:
∂ L ∂ w j k = δ k K o j δ k K = o k ( 1 − o k ) ( o k − t k ) \begin{array}{c} \frac{\partial \mathcal{L}}{\partial w_{j k}}=\delta_{k}^{K} o_{j} \\ \delta_{k}^{K}=o_{k}\left(1-o_{k}\right)\left(o_{k}-t_{k}\right) \end{array} wjkL=δkKojδkK=ok(1ok)(oktk)

倒数第二层:
∂ L ∂ w i j = δ j J o i δ j J = o j ( 1 − o j ) ∑ k δ k K w j k \begin{array}{c} \frac{\partial \mathcal{L}}{\partial w_{i j}}=\delta_{j}^{J} o_{i} \\ \delta_{j}^{J}=o_{j}\left(1-o_{j}\right) \sum_{k} \delta_{k}^{K} w_{j k} \end{array} wijL=δjJoiδjJ=oj(1oj)kδkKwjk
倒数第三层:
∂ L ∂ w n i = δ i I o n δ i I = o i ( 1 − o i ) ∑ j δ j J w i j \begin{array}{c} \frac{\partial \mathcal{L}}{\partial w_{n i}}=\delta_{i}^{I} o_{n} \\ \delta_{i}^{I}=o_{i}\left(1-o_{i}\right) \sum_{j} \delta_{j}^{J} w_{i j} \end{array} wniL=δiIonδiI=oi(1oi)jδjJwij

其中 O n O_{n} On为倒数第三层的输入,即倒数第四层的输出。

依照此规律,只需要循环迭代计算每一层每个节点的 δ k K , δ j J , δ i I , … \delta_{k}^{K}, \delta_{j}^{J}, \delta_{i}^{I}, \ldots δkK,δjJ,δiI,等值即可求得当前层的偏导数,从而得到每层权值矩阵 W W W的梯度,再通过梯度下降算法迭代优化网络参数即可。

至此,反向传播算法介绍完毕。

接下来我们会进行两个实战:第一个实战是采用 TensorFlow 提供的自动求导来优化Himmelblau 函数的极小值;第二个实战是基于 Numpy 实现反向传播算法,并完成多层

你可能感兴趣的:(TensorFlow 2.0深度学习算法实战 第七章 反向传播算法)