反向传播算法代码
class Network(object):
...
def backprop(self, x, y):
"""Return a tuple "(nabla_b, nabla_w)" representing the
gradient for the cost function C_x. "nabla_b" and
"nabla_w" are layer-by-layer lists of numpy arrays, similar
to "self.biases" and "self.weights"."""
nabla_b = [np.zeros(b.shape) for b in self.biases]
nabla_w = [np.zeros(w.shape) for w in self.weights]
# feedforward
activations = [x] # list to store all the activations, layer by layer
zs = [] # list to store all the z vectors, layer by layer
for b, w in zip(self.biases, self.weights):
z = np.dot(w, activation)+b
zs.append(z)
activations.append(activation)
# backward pass
delta = self.cost_derivative(activations[-1], y) * \
sigmoid_prime(zs[-1])
nabla_b[-1] = delta
nabla_w[-1] = np.dot(delta, activations[-2].transpose())
# Note that the variable l in the loop below is used a little
# differently to the notation in Chapter 2 of the book. Here,
# l = 1 means the last layer of neurons, l = 2 is the
# second-last layer, and so on. It's a renumbering of the
# scheme in the book, used here to take advantage of the fact
# that Python can use negative indices in lists.
for l in xrange(2, self.num_layers):
z = zs[-l]
delta = np.dot(self.weights[-l+1].transpose(), delta) * sp
nabla_b[-l] = delta
return (nabla_b, nabla_w)
...
def cost_derivative(self, output_activations, y):
"""Return the vector of partial derivatives \partial C_x /
\partial a for the output activations."""
return (output_activations-y)
def sigmoid(z):
"""The sigmoid function."""
return 1.0/(1.0+np.exp(-z))
def sigmoid_prime(z):
"""Derivative of the sigmoid function."""
return sigmoid(z)*(1-sigmoid(z))
- 在一个批次(mini-batch)上应用完全基于矩阵的反向传播方法
在我们的随机梯度下降算法的实现中,我们需要依次遍历一个批次(mini-batch)中的训练样例。我们也可以修改反向传播算法,使得它可以同时为一个批次中的所有训练样例计算梯度。我们在输入时传入一个矩阵(而不是一个向量),这个矩阵的列代表了这一个批次中的向量。前向传播时,每一个节点都将输入乘以权重矩阵、加上偏置矩阵并应用sigmoid
函数来得到输出,反向传播时也用类似的方式计算。明确地写出这种反向传播方法,并修改,令其使用这种完全基于矩阵的方法进行计算。这种方式的优势在于它可以更好地利用现代线性函数库,并且比循环的方式运行得更快。(例如,在我的笔记本电脑上求解与上一章所讨论的问题相类似的MNIST分类问题时,最高可以达到两倍的速度。)在实践中,所有正规的反向传播算法库都使用了这种完全基于矩阵的方法或其变种。