上一节中我们观察了过拟合现象，即模型的训练误差远小于它在测试集上的误差。虽然增大训练数据集可能会减轻过拟合，但是获取额外的训练数据往往代价高昂。本节介绍应对过拟合问题的常用方法：权重衰减（weight decay）。

方法

权重衰减等价于𝐿2范数正则化（regularization）。正则化通过为模型损失函数添加惩罚项使学出的模型参数值较小，是应对过拟合的常用手段。我们先描述𝐿2范数正则化，再解释它为何又称权重衰减。

𝐿2范数正则化在模型原损失函数基础上添加𝐿2范数惩罚项，从而得到训练所需要最小化的函数。𝐿2范数惩罚项指的是模型权重参数每个元素的平方和与一个正的常数的乘积。以“线性回归”一节中的线性回归损失函数

image.png
为例，其中𝑤1,𝑤2 是权重参数，𝑏 是偏差参数，样本𝑖 的输入为权重衰减 - 图2，标签为权重衰减 - 图3 ，样本数为𝑛 。将权重参数用向量𝑤=[𝑤1,𝑤2] 表示，带有𝐿2 范数惩罚项的新损失函数为

image.png
其中超参数𝜆>0 。当权重参数均为0时，惩罚项最小。当𝜆 较大时，惩罚项在损失函数中的比重较大，这通常会使学到的权重参数的元素较接近0。当𝜆 设为0时，惩罚项完全不起作用。上式中𝐿2 范数平方权重衰减 - 图5 展开后得到 权重衰减 - 图6。有了𝐿2 范数惩罚项后，在小批量随机梯度下降中，我们将“线性回归”一节中权重𝑤1 和𝑤2 的迭代方式更改为

image.png
可见，𝐿2 范数正则化令权重𝑤1 和𝑤2 先自乘小于1的数，再减去不含惩罚项的梯度。因此，𝐿2 范数正则化又叫权重衰减。权重衰减通过惩罚绝对值较大的模型参数为需要学习的模型增加了限制，这可能对过拟合有效。实际场景中，我们有时也在惩罚项中添加偏差元素的平方和。

高维线性回归实验

下面，我们以高维线性回归为例来引入一个过拟合问题，并使用权重衰减来应对过拟合。设数据样本特征的维度为𝑝 。对于训练数据集和测试数据集中特征为𝑥1,𝑥2,…,𝑥𝑝 的任一样本，我们使用如下的线性函数来生成该样本的标签：

image.png
其中噪声项𝜖 服从均值为0、标准差为0.01的正态分布。为了较容易地观察过拟合，我们考虑高维线性回归问题，如设维度𝑝=200 ；同时，我们特意把训练数据集的样本数设低，如20。

  1. %matplotlib inline
  2. import d2lzh as d2l
  3. from mxnet import autograd, gluon, init, nd
  4. from mxnet.gluon import data as gdata, loss as gloss, nn
  5. n_train, n_test, num_inputs = 20, 100, 200
  6. true_w, true_b = nd.ones((num_inputs, 1)) * 0.01, 0.05
  7. features = nd.random.normal(shape=(n_train + n_test, num_inputs))
  8. labels = nd.dot(features, true_w) + true_b
  9. labels += nd.random.normal(scale=0.01, shape=labels.shape)
  10. train_features, test_features = features[:n_train, :], features[n_train:, :]
  11. train_labels, test_labels = labels[:n_train], labels[n_train:]

从零开始实现

下面先介绍从零开始实现权重衰减的方法。我们通过在目标函数后添加𝐿2 范数惩罚项来实现权重衰减。

初始化模型参数

首先，定义随机初始化模型参数的函数。该函数为每个参数都附上梯度。

  1. def init_params():
  2.     w = nd.random.normal(scale=1, shape=(num_inputs, 1))
  3.     b = nd.zeros(shape=(1,))
  4.     w.attach_grad()
  5.     b.attach_grad()
  6.     return [w, b]


定义𝐿2 范数惩罚项

下面定义𝐿2 范数惩罚项。这里只惩罚模型的权重参数。

  1. def l2_penalty(w):
  2.     return (w**2).sum() / 2


定义训练和测试

下面定义如何在训练数据集和测试数据集上分别训练和测试模型。与前面几节中不同的是，这里在计算最终的损失函数时添加了𝐿2 范数惩罚项。

  1. batch_size, num_epochs, lr = 1, 100, 0.003
  2. net, loss = d2l.linreg, d2l.squared_loss
  3. train_iter = gdata.DataLoader(gdata.ArrayDataset(train_features, train_labels), batch_size, shuffle=True)
  4. def fit_and_plot(lambd):
  5.     w, b = init_params()
  6.     train_ls, test_ls = [], []
  7.     for _ in range(num_epochs):
  8.         for X, y in train_iter:
  9.             with autograd.record():
  10.                 # 添加了L2范数惩罚项，广播机制使其变成长度为batch_size的向量
  11.                 l = loss(net(X, w, b), y) + lambd * l2_penalty(w)
  12.             l.backward()
  13.             d2l.sgd([w, b], lr, batch_size)
  14.         train_ls.append(loss(net(train_features, w, b),train_labels).mean().asscalar())
  15.         test_ls.append(loss(net(test_features, w, b),test_labels).mean().asscalar())
  16.     d2l.semilogy(range(1, num_epochs + 1), train_ls, 'epochs', 'loss',
  17.                  range(1, num_epochs + 1), test_ls, ['train', 'test'])
  18.     print('L2 norm of w:', w.norm().asscalar())


观察过拟合

接下来，让我们训练并测试高维线性回归模型。当lambd设为0时，我们没有使用权重衰减。结果训练误差远小于测试集上的误差。这是典型的过拟合现象。

  1. fit_and_plot(lambd=0)
  4. L2 norm of w: 11.611939

使用权重衰减

下面我们使用权重衰减。可以看出，训练误差虽然有所提高，但测试集上的误差有所下降。过拟合现象得到一定程度的缓解。另外，权重参数的𝐿2 范数比不使用权重衰减时的更小，此时的权重参数更接近0。

  1. fit_and_plot(lambd=3)
  4. L2 norm of w: 0.04092614

简洁实现

这里我们直接在构造Trainer实例时通过wd参数来指定权重衰减超参数。默认下，Gluon会对权重和偏差同时衰减。我们可以分别对权重和偏差构造Trainer实例，从而只对权重衰减。

  1. def fit_and_plot_gluon(wd):
  2.     net = nn.Sequential()
  3.     net.add(nn.Dense(1))
  4.     net.initialize(init.Normal(sigma=1))
  5.     # 对权重参数衰减。权重名称一般是以weight结尾
  6.     trainer_w = gluon.Trainer(net.collect_params('.*weight'), 'sgd',{'learning_rate': lr, 'wd': wd})
  7.     # 不对偏差参数衰减。偏差名称一般是以bias结尾
  8.     trainer_b = gluon.Trainer(net.collect_params('.*bias'), 'sgd',{'learning_rate': lr})
  9.     train_ls, test_ls = [], []
  10.     for _ in range(num_epochs):
  11.         for X, y in train_iter:
  12.             with autograd.record():
  13.                 l = loss(net(X), y)
  14.             l.backward()
  15.             # 对两个Trainer实例分别调用step函数，从而分别更新权重和偏差
  16.             trainer_w.step(batch_size)
  17.             trainer_b.step(batch_size)
  18.         train_ls.append(loss(net(train_features),train_labels).mean().asscalar())
  19.         test_ls.append(loss(net(test_features),test_labels).mean().asscalar())
  20.     d2l.semilogy(range(1, num_epochs + 1), train_ls, 'epochs', 'loss',
  21.                  range(1, num_epochs + 1), test_ls, ['train', 'test'])
  22.     print('L2 norm of w:', net[0].weight.data().norm().asscalar())

与从零开始实现权重衰减的实验现象类似，使用权重衰减可以在一定程度上缓解过拟合问题。

  1. fit_and_plot_gluon(0)
  4. L2 norm of w: 13.311796
  1. fit_and_plot_gluon(3)
  4. L2 norm of w: 0.033146076

小结

  • 正则化通过为模型损失函数添加惩罚项使学出的模型参数值较小，是应对过拟合的常用手段。
  • 权重衰减等价于𝐿2 范数正则化，通常会使学到的权重参数的元素较接近0。
  • 权重衰减可以通过Gluon的wd超参数来指定。
  • 可以定义多个Trainer实例对不同的模型参数使用不同的迭代方法。