参数 X 是图像数据、t 是正确解标签
accuracy(self, x, t) 计算识别精度
numerical_gradient(self, x, t) 通过数值微分计算关于权重参数的梯度(同上一章)
gradient(self, x, t) 通过误差反向传播法计算关于权重参数的梯度

这个类的实现稍微有一点长,但是内容和 4.5 节的学习算法的实现有很多共通的部分,不同点主要在于这里使用了层。通过使用层,获得识别结果的处理(predict())和计算梯度的处理(gradient())只需通过层之间的传递就能完成。下面是 TwoLayerNet 的代码实现。

  1. import sys, os
  2. sys.path.append(os.pardir)
  3. import numpy as np
  4. from common.layers import *
  5. from common.gradient import numerical_gradient
  6. from collections import OrderedDict
  7.  
  8. class TwoLayerNet:
  9.  
  10. def __init__(self, input_size, hidden_size, output_size,
  11. weight_init_std=0.01):
  12. # 初始化权重
  13. self.params = {}
  14. self.params['W1'] = weight_init_std * \
  15. np.random.randn(input_size, hidden_size)
  16. self.params['b1'] = np.zeros(hidden_size)
  17. self.params['W2'] = weight_init_std * \
  18. np.random.randn(hidden_size, output_size)
  19. self.params['b2'] = np.zeros(output_size)
  20.  
  21. # 生成层
  22. self.layers = OrderedDict()
  23. self.layers['Affine1'] = \
  24. Affine(self.params['W1'], self.params['b1'])
  25. self.layers['Relu1'] = Relu()
  26. self.layers['Affine2'] = \
  27. Affine(self.params['W2'], self.params['b2'])
  28.  
  29. self.lastLayer = SoftmaxWithLoss()
  30.  
  31. def predict(self, x):
  32. for layer in self.layers.values():
  33. x = layer.forward(x)
  34.  
  35. return x
  36.  
  37. # x:输入数据, t:监督数据
  38. def loss(self, x, t):
  39. y = self.predict(x)
  40. return self.lastLayer.forward(y, t)
  41.  
  42. def accuracy(self, x, t):
  43. y = self.predict(x)
  44. y = np.argmax(y, axis=1)
  45. if t.ndim != 1 : t = np.argmax(t, axis=1)
  46. accuracy = np.sum(y == t) / float(x.shape[0])
  47. return accuracy
  48.  
  49. # x:输入数据, t:监督数据
  50. def numerical_gradient(self, x, t):
  51. loss_W = lambda W: self.loss(x, t)
  52.  
  53. grads = {}
  54. grads['W1'] = numerical_gradient(loss_W, self.params['W1'])
  55. grads['b1'] = numerical_gradient(loss_W, self.params['b1'])
  56. grads['W2'] = numerical_gradient(loss_W, self.params['W2'])
  57. grads['b2'] = numerical_gradient(loss_W, self.params['b2'])
  58.  
  59. return grads
  60.  
  61. def gradient(self, x, t):
  62. # forward
  63. self.loss(x, t)
  64.  
  65. # backward
  66. dout = 1
  67. dout = self.lastLayer.backward(dout)
  68.  
  69. layers = list(self.layers.values())
  70. layers.reverse()
  71. for layer in layers:
  72. dout = layer.backward(dout)
  73.  
  74. # 设定
  75. grads = {}
  76. grads['W1'] = self.layers['Affine1'].dW
  77. grads['b1'] = self.layers['Affine1'].db
  78. grads['W2'] = self.layers['Affine2'].dW
  79. grads['b2'] = self.layers['Affine2'].db
  80.  
  81. return grads

请注意这个实现中的粗体字代码部分,尤其是将神经网络的层保存为 OrderedDict 这一点非常重要。OrderedDict 是有序字典,“有序”是指它可以记住向字典里添加元素的顺序。因此,神经网络的正向传播只需按照添加元素的顺序调用各层的 forward() 方法就可以完成处理,而反向传播只需要按照相反的顺序调用各层即可。因为 Affine 层和 ReLU 层的内部会正确处理正向传播和反向传播,所以这里要做的事情仅仅是以正确的顺序连接各层,再按顺序(或者逆序)调用各层。

像这样通过将神经网络的组成元素以层的方式实现,可以轻松地构建神经网络。这个用层进行模块化的实现具有很大优点。因为想另外构建一个神经网络(比如 5 层、10 层、20 层……的大的神经网络)时,只需像组装乐高积木那样添加必要的层就可以了。之后,通过各个层内部实现的正向传播和反向传播,就可以正确计算进行识别处理或学习所需的梯度。

5.7.3 误差反向传播法的梯度确认

到目前为止,我们介绍了两种求梯度的方法。一种是基于数值微分的方法,另一种是解析性地求解数学式的方法。后一种方法通过使用误差反向传播法,即使存在大量的参数,也可以高效地计算梯度。因此,后文将不再使用耗费时间的数值微分,而是使用误差反向传播法求梯度。

数值微分的计算很耗费时间,而且如果有误差反向传播法的(正确的)实现的话,就没有必要使用数值微分的实现了。那么数值微分有什么用呢?实际上,在确认误差反向传播法的实现是否正确时,是需要用到数值微分的。

数值微分的优点是实现简单,因此,一般情况下不太容易出错。而误差反向传播法的实现很复杂,容易出错。所以,经常会比较数值微分的结果和误差反向传播法的结果,以确认误差反向传播法的实现是否正确。确认数值微分求出的梯度结果和误差反向传播法求出的结果是否一致(严格地讲,是非常相近)的操作称为梯度确认(gradient check)。梯度确认的代码实现如下所示(源代码在 ch05/gradient_check.py 中)。

  1. import sys, os
  2. sys.path.append(os.pardir)
  3. import numpy as np
  4. from dataset.mnist import load_mnist
  5. from two_layer_net import TwoLayerNet
  6. # 读入数据
  7. (x_train, t_train), (x_test, t_test) = \ load_mnist(normalize=True, one_
  8. hot_label = True)
  9. network = TwoLayerNet(input_size=784, hidden_size=50, output_size=10)
  10. x_batch = x_train[:3]
  11. t_batch = t_train[:3]
  12. grad_numerical = network.numerical_gradient(x_batch, t_batch)
  13. grad_backprop = network.gradient(x_batch, t_batch)
  14. # 求各个权重的绝对误差的平均值
  15. for key in grad_numerical.keys():
  16. diff = np.average( np.abs(grad_backprop[key] - grad_numerical[key]) )
  17. print(key + ":" + str(diff))

和以前一样,读入 MNIST 数据集。然后,使用训练数据的一部分,确认数值微分求出的梯度和误差反向传播法求出的梯度的误差。这里误差的计算方法是求各个权重参数中对应元素的差的绝对值,并计算其平均值。运行上面的代码后,会输出如下结果。

  1. b1:9.70418809871e-13
  2. W2:8.41139039497e-13
  3. b2:1.1945999745e-10
  4. W1:2.2232446644e-13

从这个结果可以看出,通过数值微分和误差反向传播法求出的梯度的差非常小。比如,第 1 层的偏置的误差是 9.7e-130.00000000000097)。这样一来,我们就知道了通过误差反向传播法求出的梯度是正确的,误差反向传播法的实现没有错误。

读入数据 - 图1 数值微分和误差反向传播法的计算结果之间的误差为 0 是很少见的。这是因为计算机的计算精度有限(比如,32 位浮点数)。受到数值精度的限制,刚才的误差一般不会为 0,但是如果实现正确的话,可以期待这个误差是一个接近 0 的很小的值。如果这个值很大,就说明误差反向传播法的实现存在错误。

5.7.4 使用误差反向传播法的学习

最后,我们来看一下使用了误差反向传播法的神经网络的学习的实现。和之前的实现相比,不同之处仅在于通过误差反向传播法求梯度这一点。这里只列出了代码,省略了说明(源代码在 ch05/train_neuralnet.py 中)。

  1. import sys, os
  2. sys.path.append(os.pardir)
  3. import numpy as np
  4. from dataset.mnist import load_mnist
  5. from two_layer_net import TwoLayerNet
  6.  
  7. # 读入数据
  8. (x_train, t_train), (x_test, t_test) = \
  9. load_mnist(normalize=True, one_hot_label=True)
  10.  
  11. network = TwoLayerNet(input_size=784, hidden_size=50, output_size=10)
  12.  
  13. iters_num = 10000
  14. train_size = x_train.shape[0]
  15. batch_size = 100
  16. learning_rate = 0.1
  17. train_loss_list = []
  18. train_acc_list = []
  19. test_acc_list = []
  20.  
  21. iter_per_epoch = max(train_size / batch_size, 1)
  22.  
  23. for i in range(iters_num):
  24. batch_mask = np.random.choice(train_size, batch_size)
  25. x_batch = x_train[batch_mask]
  26. t_batch = t_train[batch_mask]
  27.  
  28. # 通过误差反向传播法求梯度
  29. grad = network.gradient(x_batch, t_batch)
  30.  
  31. # 更新
  32. for key in ('W1', 'b1', 'W2', 'b2'):
  33. network.params[key] -= learning_rate * grad[key]
  34.  
  35. loss = network.loss(x_batch, t_batch)
  36. train_loss_list.append(loss)
  37.  
  38. if i % iter_per_epoch == 0:
  39. train_acc = network.accuracy(x_train, t_train)
  40. test_acc = network.accuracy(x_test, t_test)
  41. train_acc_list.append(train_acc)
  42. test_acc_list.append(test_acc)
  43. print(train_acc, test_acc)