1 计算图原理

计算图(Computational Graph)是机器学习领域中推导神经网络和其他模型算法，以及软件编程实现的有效工具。

计算图的核心是将模型表示成一张拓扑有序(Topologically Ordered)的有向无环图(Directed Acyclic Graph)，其中每个节点$u_i$包含数值信息(可以是标量、向量、矩阵或张量)和算子信息$f_i$。拓扑有序指当前节点仅在全体指向它的节点被计算后才进行计算。

计算图的优点在于：

• 可以通过基本初等映射 的拓扑联结，形成复合的复杂模型，大多数神经网络模型都可以被计算图表示；
• 便于实现自动微分机(Automatic Differentiation Machine)，对给定计算图可基于链式法则由节点局部梯度进行反向传播。

计算图的基本概念如表所示，基于计算图的基本前向传播和反向传播算法如表

2 基于计算图的传播

基于计算图的前向传播算法如下

基于计算图的反向传播算法如下

以第一节的图为例，可知 E = { ( 1 , 3 ) , ( 2 , 3 ) , ( 2 , 4 ) , ( 3 , 4 ) } E=\left\{ \left( 1,3 \right) ,\left( 2,3 \right) ,\left( 2,4 \right) ,\left( 3,4 \right) \right\} E={(1,3),(2,3),(2,4),(3,4)}。首先进行前向传播：

$$\{\begin{array}{l}{u}_3=u_1+u_2=5\\ u_4=u_2{u}_3=15\end{array}$$

$$\{\begin{array}{l}{\mathbf{J}}_{1\to 3}=\partial u_3/\partial u_1=1\\ {\mathbf{J}}_{2\to 3}=\partial u_3/\partial u_2=1\\ {\mathbf{J}}_{2\to 4}=\partial u_4/\partial u_2=u_3=5\\ {\mathbf{J}}_{3\to 4}=\partial u_4/\partial u_3=u_2=3\end{array}$$

接着进行反向传播：

$$\{\begin{array}{l}{\mathbf{P}}_4=1\\ {\mathbf{P}}_3={\mathbf{P}}_4{\mathbf{J}}_{3\to 4}=3\\ {\mathbf{P}}_2={\mathbf{P}}_4{\mathbf{J}}_{2\to 4}+{\mathbf{P}}_3{\mathbf{J}}_{2\to 3}=8\\ {\mathbf{P}}_1={\mathbf{P}}_3{\mathbf{J}}_{1\to 3}=3\end{array}$$

3 神经网络计算图

一个神经网络的计算图实例如下，所有参数都可以用之前的模型表示

$$L\{\begin{array}{l}{u}_1={\mathbf{W}}^1\in {\mathbb{R}}^{n_1\times n_0}\\ u_2={\mathbf{b}}^1\in {\mathbb{R}}^{n_1}\\ u_3=\mathbf{x}\in {\mathbb{R}}^{n_0}\\ u_4={\mathbf{W}}^2\in {\mathbb{R}}^{n_2\times n_1}\\ u_5={\mathbf{b}}^2\in {\mathbb{R}}^{n_2}\\ u_6=\mathbf{y}\in {\mathbb{R}}^{n_2}\end{array}C\{\begin{array}{l}{u}_7={\mathbf{z}}^1\in {\mathbb{R}}^{n_1}={\mathbf{W}}^1\mathbf{x}+{\mathbf{b}}^1\\ u_8={\mathbf{a}}^1\in {\mathbb{R}}^{n_1}=\sigma \left({\mathbf{z}}^1\right)\\ u_9={\mathbf{z}}^2\in {\mathbb{R}}^{n_2}={\mathbf{W}}^2{\mathbf{a}}^1+{\mathbf{b}}^2\\ u_{10}=\mathbf{y}\in {\mathbb{R}}^{n_2}=\sigma \left({\mathbf{z}}^2\right)\\ u_{11}=E\in \mathbb{R}=\frac{1}{2}{\left(\mathbf{y}-\tilde{\mathbf{y}}\right)}^T\left(\mathbf{y}-\tilde{\mathbf{y}}\right)\end{array}$$

4 自动微分机

自动微分机的基本原理是：

• 跟踪记录从输入张量到输出张量的计算过程，并生成一幅前向传播计算图，计算图中的节点与张量一一对应；
• 基于计算图反向传播原理即可链式地求解输出节点关于各节点的梯度。

必须指出，Pytorch不允许张量对张量求导，故输出节点必须是标量，通常为损失函数或输出向量的加权和；为节约内存，每次反向传播后Pytorch会自动释放前向传播计算图，即销毁中间计算节点的梯度和节点间的连接结构。

5 Pytorch中的自动微分

Tensor在自动微分机中的重要属性如表所示。

完成前向传播后，调用反向传播API即可更新各节点梯度，具体如下

backward(gradient=None, retain_graph=None, create_graph=None)

其中

• gradient是权重向量，当输出节点$y$不为标量时需指定与其同维的gradient，并以标量$gradien{t}^{T}y$为输出进行反向传播
• retain_graph用于缓存前向传播计算图，可应用于一次传播测试多个损失函数等情形；
• creat_graph用于构造导数计算图，可用于进一步求解高阶导数。

5.1 梯度缓存

中间计算节点的梯度需要通过retain_grad()方法进行缓存

w1 = torch.tensor([[2.], [3.]], requires_grad=True)
b1 = torch.tensor([1.], requires_grad=True)
x = torch.tensor([[10.], [20.]])

y = torch.mm(w1.transpose(0, 1), x) + b1
y.retain_grad()	# 若不缓存则y.grad=None
out = 3*y
out.backward()


>> tensor([[30.], [60.]]) tensor([3.]) None tensor([[3.]])

5.2 参数冻结

若希望冻结网络部分参数，只调整优化另一部分参数；或按顺序训练分支网络而屏蔽对主网络梯度的，可使用detach()方法从计算图中分离节点，阻断反向传播。分离的节点与原节点共享值内存，但不具有grad和grad_fn属性。

# 记第一层网络w1-b1为f，第二层网络w2-b2为g
w1 = torch.tensor([[2.], [3.]], requires_grad=True)
w2 = torch.tensor([3.], requires_grad=True)
b1 = torch.tensor([1.], requires_grad=True)
b2 = torch.tensor([2.], requires_grad=True)
x = torch.tensor([[10.], [20.]])

y = torch.mm(w1.transpose(0, 1), x) + b1
y_ = y.detach()
z = w2 * y_ + b2
out = 3*z
out.backward()

print(w1.grad, b1.grad, w2.grad, b2.grad)
>> None None tensor([243.]) tensor([3.]) # f被冻结，梯度不更新
# 若不使用detach冻结y之前的网络，则
>> tensor([[ 90.], [180.]]) tensor([9.]) tensor([243.]) tensor([3.])

🔥 更多精彩专栏：

• 《ROS从入门到精通》
• 《Pytorch深度学习实战》
• 《机器学习强基计划》
• 《运动规划实战精讲》
• …

更多文章请关注《万象专栏》