Run this notebook online:Binder or Colab: Colab

4.8. 数值稳定性和模型初始化

到目前为止,我们实现的每个模型都是根据某个预先指定的分布来初始化模型的参数。直到现在,我们认为初始化方案是理所当然的,忽略了如何做出这些选择的细节。你甚至可能会觉得,初始化方案的选择并不是特别重要。相反,初始化方案的选择在神经网络学习中起着非常重要的作用,它对保持数值稳定性至关重要。此外,这些选择可以与非线性激活函数的选择以有趣的方式结合在一起。我们选择哪个函数以及如何初始化参数可以决定优化算法收敛的速度有多快。糟糕选择可能会导致我们在训练时遇到梯度爆炸或梯度消失。在本节中,我们将更详细地探讨这些主题,并讨论一些有用的启发式方法。你会发现这些启发式方法在你的整个深度学习生涯中都很有用。

4.8.1. 梯度消失和梯度爆炸

考虑一个具有\(L\)层、输入\(\mathbf{x}\)和输出\(\mathbf{o}\)的深层网络。每一层\(l\)由变换\(f_l\)定义,该变换的参数为权重\(\mathbf{W}^{(l)}\),其隐藏变量是\(\mathbf{h}^{(l)}\)(令 \(\mathbf{h}^{(0)} = \mathbf{x}\))。我们的网络可以表示为:

(4.8.1)\[\mathbf{h}^{(l)} = f_l (\mathbf{h}^{(l-1)}) \text{ 因此 } \mathbf{o} = f_L \circ \ldots \circ f_1(\mathbf{x}).\]

如果所有隐藏变量和输入都是向量,我们可以将\(\mathbf{o}\)关于任何一组参数\(\mathbf{W}^{(l)}\)的梯度写为下式:

(4.8.2)\[\partial_{\mathbf{W}^{(l)}} \mathbf{o} = \underbrace{\partial_{\mathbf{h}^{(L-1)}} \mathbf{h}^{(L)}}_{ \mathbf{M}^{(L)} \stackrel{\mathrm{def}}{=}} \cdot \ldots \cdot \underbrace{\partial_{\mathbf{h}^{(l)}} \mathbf{h}^{(l+1)}}_{ \mathbf{M}^{(l+1)} \stackrel{\mathrm{def}}{=}} \underbrace{\partial_{\mathbf{W}^{(l)}} \mathbf{h}^{(l)}}_{ \mathbf{v}^{(l)} \stackrel{\mathrm{def}}{=}}.\]

换言之,该梯度是\(L-l\)个矩阵\(\mathbf{M}^{(L)} \cdot \ldots \cdot \mathbf{M}^{(l+1)}\)与梯度向量 \(\mathbf{v}^{(l)}\)的乘积。因此,我们容易受到数值下溢问题的影响,当将太多的概率乘在一起时,这些问题经常会出现。在处理概率时,一个常见的技巧是切换到对数空间,即将数值表示的压力从尾数转移到指数。不幸的是,我们上面的问题更为严重:最初,矩阵 \(\mathbf{M}^{(l)}\) 可能具有各种各样的特征值。他们可能很小,也可能很大,他们的乘积可能非常大,也可能非常小

不稳定梯度带来的风险不止在于数值表示。不稳定梯度也威胁到我们优化算法的稳定性。我们可能面临一些问题。 要么是 梯度爆炸(gradient exploding)问题:参数更新过大,破坏了模型的稳定收敛; 要么是 梯度消失(gradient vanishing)问题:参数更新过小,在每次更新时几乎不会移动,导致无法学习。

4.8.1.1. 梯度消失

导致梯度消失问题的一个常见的原因是跟在每层的线性运算之后的激活函数\(\sigma\)。从历史上看,sigmoid函数\(1/(1 + \exp(-x))\)Section 4.1 提到过)很流行,因为它类似于阈值函数。由于早期的人工神经网络受到生物神经网络的启发,神经元要么完全激活要么完全不激活(就像生物神经元)的想法很有吸引力。让我们仔细看看sigmoid函数为什么会导致梯度消失。

%load ../utils/djl-imports
%load ../utils/plot-utils
%load ../utils/DataPoints.java
%load ../utils/Training.java
import org.apache.commons.lang3.ArrayUtils;
NDManager manager = NDManager.newBaseManager();
NDArray x = manager.arange(-8.0f, 8.0f, 0.1f);
x.setRequiresGradient(true);
NDArray y = null;
try(GradientCollector gc = Engine.getInstance().newGradientCollector()){
    y = Activation.sigmoid(x);
    gc.backward(y);
}

NDArray res = x.getGradient();

int xLength = (int) x.size();
int yLength = (int) y.size();

float[] X = new float[xLength];
float[] Y = new float[yLength];
float[] Z = new float[yLength];

X = x.toFloatArray();
Y = y.toFloatArray();
Z = res.toFloatArray();

String[] groups = new String[xLength*2];
Arrays.fill(groups, 0, xLength, "sigmoid");
Arrays.fill(groups, xLength, xLength * 2, "gradient");

Table data = Table.create("Data").addColumns(
    FloatColumn.create("X", ArrayUtils.addAll(X, X)),
    FloatColumn.create("grad of relu", ArrayUtils.addAll(Y, Z)),
    StringColumn.create("groups", groups)
);
render(LinePlot.create("", data, "x", "grad of relu", "groups"), "text/html");

正如你所看到的,当它的输入很大或是很小时,sigmoid函数的梯度都会消失。此外,当反向传播通过许多层时,除非我们在刚刚好的地方,这些地方sigmoid函数的输入接近于零,否则整个乘积的梯度可能会消失。当我们的网络有很多层时,除非我们很小心,否则在某一层可能会切断梯度。事实上,这个问题曾经困扰着深度网络的训练。因此,更稳定(但在神经科学的角度看起来不太合理)的ReLU系列函数已经成为从业者的默认选择。

4.8.1.2. 梯度爆炸

相反的问题,当梯度爆炸时,可能同样令人烦恼。为了更好地说明这一点,我们生成100个高斯随机矩阵,并将它们与某个初始矩阵相乘。对于我们选择的尺度(方差\(\sigma^2=1\)),矩阵乘积发生爆炸。当这种情况是由于深度网络的初始化所导致时,我们没有机会让梯度下降优化器收敛。

NDArray M = manager.randomNormal(new Shape(4,4));
System.out.println("一个矩阵: " + M);
for(int i = 0; i < 100; i++){
    M = M.dot(manager.randomNormal(new Shape(4,4)));
}

System.out.println("a乘以100个矩阵后: " + M);
一个矩阵: ND: (4, 4) gpu(0) float32
[[ 0.2925, -0.7184,  0.1   , -0.3932],
 [ 2.547 , -0.0034,  0.0083, -0.251 ],
 [ 0.129 ,  0.3728,  1.0822, -0.665 ],
 [ 0.5434, -0.7168, -1.4913,  1.4805],
]

a乘以100个矩阵后: ND: (4, 4) gpu(0) float32
[[-1.60097634e+23, -1.36331671e+23,  1.22807521e+22, -4.03570949e+23],
 [-4.56637224e+23, -3.88849763e+23,  3.50268902e+22, -1.15108159e+24],
 [-1.67982393e+23, -1.43046052e+23,  1.28854639e+22, -4.23446884e+23],
 [ 1.81735413e+23,  1.54757590e+23, -1.39405301e+22,  4.58115233e+23],
]

4.8.1.3. 打破对称性

神经网络设计中的另一个问题是其参数化所固有的对称性。假设我们有一个简单的多层感知机,它有一个隐藏层和两个隐藏单元。在这种情况下,我们可以对第一层的权重\(\mathbf{W}^{(1)}\)进行重排列,并且同样对输出层的权重进行重排列,可以获得相同的函数。第一个隐藏单元与第二个隐藏单元没有什么特别的区别。换句话说,我们在每一层的隐藏单元之间具有排列对称性。

这不仅仅是理论上的麻烦。考虑前述具有两个隐藏单元的单隐藏层多层感知机。为便于说明,假设输出层将两个隐藏单元转换为仅一个输出单元。想象一下,如果我们将隐藏层的所有参数初始化为\(\mathbf{W}^{(1)} = c\)\(c\)为常量,会发生什么情况。在这种情况下,在正向传播期间,两个隐藏单元采用相同的输入和参数,产生相同的激活,该激活被送到输出单元。在反向传播期间,根据参数\(\mathbf{W}^{(1)}\)对输出单元进行微分,得到一个梯度,其元素都取相同的值。因此,在基于梯度的迭代(例如,小批量随机梯度下降)之后,\(\mathbf{W}^{(1)}\)的所有元素仍然采用相同的值。这样的迭代永远不会打破对称性,我们可能永远也无法实现网络的表达能力。隐藏层的行为就好像只有一个单元。请注意,虽然小批量随机梯度下降不会打破这种对称性,但dropout正则化可以。

4.8.2. 参数初始化

解决(或至少减轻)上述问题的一种方法是仔细地进行初始化。优化期间的注意和适当的正则化可以进一步提高稳定性。

4.8.2.1. 默认初始化

在前面的部分中,例如在 Section 3.3 中,我们使用正态分布来初始化权重值。如果我们不指定初始化方法,框架将使用默认的随机初始化方法,对于中等规模的问题,这种方法通常很有效。

4.8.2.2. Xavier初始化

让我们看看某些没有非线性的全连接层输出(例如,隐藏变量)\(o_{i}\)的尺度分布。 对于该层\(n_\mathrm{in}\)输入\(x_j\)及其相关权重\(w_{ij}\),输出由下式给出

(4.8.3)\[o_{i} = \sum_{j=1}^{n_\mathrm{in}} w_{ij} x_j.\]

权重\(w_{ij}\)都是从同一分布中独立抽取的。此外,让我们假设该分布具有零均值和方差\(\sigma^2\)。请注意,这并不意味着分布必须是高斯的,只是均值和方差需要存在。现在,让我们假设层\(x_j\)的输入也具有零均值和方差\(\gamma^2\),并且它们独立于\(w_{ij}\)并且彼此独立。在这种情况下,我们可以按如下方式计算\(o_i\)的平均值和方差:

(4.8.4)\[\begin{split}\begin{aligned} E[o_i] & = \sum_{j=1}^{n_\mathrm{in}} E[w_{ij} x_j] \\&= \sum_{j=1}^{n_\mathrm{in}} E[w_{ij}] E[x_j] \\&= 0, \\ \mathrm{Var}[o_i] & = E[o_i^2] - (E[o_i])^2 \\ & = \sum_{j=1}^{n_\mathrm{in}} E[w^2_{ij} x^2_j] - 0 \\ & = \sum_{j=1}^{n_\mathrm{in}} E[w^2_{ij}] E[x^2_j] \\ & = n_\mathrm{in} \sigma^2 \gamma^2. \end{aligned}\end{split}\]

保持方差不变的一种方法是设置\(n_\mathrm{in} \sigma^2 = 1\)。现在考虑反向传播过程,我们面临着类似的问题,尽管梯度是从更靠近输出的层传播的。使用与正向传播相同的推理,我们可以看到,除非\(n_\mathrm{out} \sigma^2 = 1\),否则梯度的方差可能会增大,其中\(n_\mathrm{out}\)是该层的输出的数量。这使我们进退两难:我们不可能同时满足这两个条件。相反,我们只需满足:

(4.8.5)\[\begin{aligned} \frac{1}{2} (n_\mathrm{in} + n_\mathrm{out}) \sigma^2 = 1 \text{ 或等价于 } \sigma = \sqrt{\frac{2}{n_\mathrm{in} + n_\mathrm{out}}}. \end{aligned}\]

这就是现在标准且实用的Xavier初始化的基础,它以其提出者 [Glorot & Bengio, 2010] 第一作者的名字命名。通常,Xavier初始化从均值为零,方差\(\sigma^2 = \frac{2}{n_\mathrm{in} + n_\mathrm{out}}\)的高斯分布中采样权重。我们也可以利用Xavier的直觉来选择从均匀分布中抽取权重时的方差。注意均匀分布\(U(-a, a)\)的方差为\(\frac{a^2}{3}\)。将\(\frac{a^2}{3}\)代入到\(\sigma^2\)的条件中,将得到初始化的建议:

(4.8.6)\[U\left(-\sqrt{\frac{6}{n_\mathrm{in} + n_\mathrm{out}}}, \sqrt{\frac{6}{n_\mathrm{in} + n_\mathrm{out}}}\right).\]

尽管上述数学推理中,不存在非线性的假设在神经网络中很容易被违反,但Xavier初始化方法在实践中被证明是有效的。

4.8.2.3. 额外内容

上面的推理仅仅触及了现代参数初始化方法的皮毛。深度学习框架通常实现十几种不同的启发式方法。此外,参数初始化一直是深度学习基础研究的热点领域。其中包括专门用于参数绑定(共享)、超分辨率、序列模型和其他情况的启发式算法。例如,Xiao等人演示了通过使用精心设计的初始化方法 [Xiao et al., 2018] ,可以无需结构上的技巧而训练10000层神经网络的可能性。

如果你对该主题感兴趣,我们建议你深入研究本模块的内容,阅读提出并分析每种启发式方法的论文,然后探索有关该主题的最新出版物。也许你会偶然发现甚至发明一个聪明的想法,并为深度学习框架提供一个实现。

4.8.3. 小结

  • 梯度消失和爆炸是深度网络中常见的问题。在参数初始化时需要非常小心,以确保梯度和参数可以得到很好的控制。

  • 需要用启发式的初始化方法来确保初始梯度既不太大也不太小。

  • ReLU激活函数缓解了梯度消失问题,这样可以加速收敛。

  • 随机初始化是保证在进行优化前打破对称性的关键。

  • Xavier初始化表明,对于每一层,输出的方差不受输入数量的影响,任何梯度的方差不受输出数量的影响。

4.8.4. 练习

  1. 除了多层感知机的排列对称性之外,你能设计出其他神经网络可能会表现出对称性且需要被打破的情况吗?

  2. 我们是否可以将线性回归或softmax回归中的所有权重参数初始化为相同的值?

  3. 在相关资料中查找两个矩阵乘积特征值的解析界。这对确保梯度条件合适有什么启示?

  4. 如果我们知道某些项是发散的,我们能在事后修正吗?看看关于分层自适应速率缩放的论文 [You et al., 2017]