Backpropagation

From HandWiki
Short description: Optimization algorithm for artificial neural networks

In machine learning, backpropagation is a gradient estimation method used to train neural network models. The gradient estimate is used by the optimization algorithm to compute the network parameter updates.

It is an efficient application of the Leibniz chain rule (1673)[1] to such networks.[2] It is also known as the reverse mode of automatic differentiation or reverse accumulation, due to Seppo Linnainmaa (1970).[3][4][5][6][7][8][9] The term "back-propagating error correction" was introduced in 1962 by Frank Rosenblatt,[10][2] but he did not know how to implement this, even though Henry J. Kelley had a continuous precursor of backpropagation[11] already in 1960 in the context of control theory.[2]

Backpropagation computes the gradient of a loss function with respect to the weights of the network for a single input–output example, and does so efficiently, computing the gradient one layer at a time, iterating backward from the last layer to avoid redundant calculations of intermediate terms in the chain rule; this can be derived through dynamic programming.[11][12][13] Gradient descent, or variants such as stochastic gradient descent,[14] are commonly used.

Strictly the term backpropagation refers only to the algorithm for computing the gradient, not how the gradient is used; but the term is often used loosely to refer to the entire learning algorithm – including how the gradient is used, such as by stochastic gradient descent.[15] In 1986 David E. Rumelhart et al. published an experimental analysis of the technique.[16] This contributed to the popularization of backpropagation and helped to initiate an active period of research in multilayer perceptrons.

Overview

Backpropagation computes the gradient in weight space of a feedforward neural network, with respect to a loss function. Denote:

  • [math]\displaystyle{ x }[/math]: input (vector of features)
  • [math]\displaystyle{ y }[/math]: target output
    For classification, output will be a vector of class probabilities (e.g., [math]\displaystyle{ (0.1, 0.7, 0.2) }[/math], and target output is a specific class, encoded by the one-hot/dummy variable (e.g., [math]\displaystyle{ (0, 1, 0) }[/math]).
  • [math]\displaystyle{ C }[/math]: loss function or "cost function"[lower-alpha 1]
    For classification, this is usually cross-entropy (XC, log loss), while for regression it is usually squared error loss (SEL).
  • [math]\displaystyle{ L }[/math]: the number of layers
  • [math]\displaystyle{ W^l = (w^l_{jk}) }[/math]: the weights between layer [math]\displaystyle{ l - 1 }[/math] and [math]\displaystyle{ l }[/math], where [math]\displaystyle{ w^l_{jk} }[/math] is the weight between the [math]\displaystyle{ k }[/math]-th node in layer [math]\displaystyle{ l - 1 }[/math] and the [math]\displaystyle{ j }[/math]-th node in layer [math]\displaystyle{ l }[/math][lower-alpha 2]
  • [math]\displaystyle{ f^l }[/math]: activation functions at layer [math]\displaystyle{ l }[/math]
    For classification the last layer is usually the logistic function for binary classification, and softmax (softargmax) for multi-class classification, while for the hidden layers this was traditionally a sigmoid function (logistic function or others) on each node (coordinate), but today is more varied, with rectifier (ramp, ReLU) being common.
  • [math]\displaystyle{ a^l_j }[/math]: activation of the [math]\displaystyle{ j }[/math]-th node in layer [math]\displaystyle{ l }[/math].

In the derivation of backpropagation, other intermediate quantities are used by introducing them as needed below. Bias terms are not treated specially since they correspond to a weight with a fixed input of 1. For backpropagation the specific loss function and activation functions do not matter as long as they and their derivatives can be evaluated efficiently. Traditional activation functions include sigmoid, tanh, and ReLU. swish[17] mish,[18] and other activation functions have since been proposed as well.

The overall network is a combination of function composition and matrix multiplication:

[math]\displaystyle{ g(x) := f^L(W^L f^{L-1}(W^{L-1} \cdots f^1(W^1 x)\cdots)) }[/math]

For a training set there will be a set of input–output pairs, [math]\displaystyle{ \left\{(x_i, y_i)\right\} }[/math]. For each input–output pair [math]\displaystyle{ (x_i, y_i) }[/math] in the training set, the loss of the model on that pair is the cost of the difference between the predicted output [math]\displaystyle{ g(x_i) }[/math] and the target output [math]\displaystyle{ y_i }[/math]:

[math]\displaystyle{ C(y_i, g(x_i)) }[/math]

Note the distinction: during model evaluation the weights are fixed while the inputs vary (and the target output may be unknown), and the network ends with the output layer (it does not include the loss function). During model training the input–output pair is fixed while the weights vary, and the network ends with the loss function.

Backpropagation computes the gradient for a fixed input–output pair [math]\displaystyle{ (x_i, y_i) }[/math], where the weights [math]\displaystyle{ w^l_{jk} }[/math] can vary. Each individual component of the gradient, [math]\displaystyle{ \partial C/\partial w^l_{jk}, }[/math] can be computed by the chain rule; but doing this separately for each weight is inefficient. Backpropagation efficiently computes the gradient by avoiding duplicate calculations and not computing unnecessary intermediate values, by computing the gradient of each layer – specifically the gradient of the weighted input of each layer, denoted by [math]\displaystyle{ \delta^l }[/math] – from back to front.

Informally, the key point is that since the only way a weight in [math]\displaystyle{ W^l }[/math] affects the loss is through its effect on the next layer, and it does so linearly, [math]\displaystyle{ \delta^l }[/math] are the only data you need to compute the gradients of the weights at layer [math]\displaystyle{ l }[/math], and then the previous layer can be computed [math]\displaystyle{ \delta^{l-1} }[/math] and repeated recursively. This avoids inefficiency in two ways. First, it avoids duplication because when computing the gradient at layer [math]\displaystyle{ l }[/math] – it is unnecessary to recompute all derivatives on later layers [math]\displaystyle{ l+1, l+2, \ldots }[/math] each time. Second, it avoids unnecessary intermediate calculations, because at each stage it directly computes the gradient of the weights with respect to the ultimate output (the loss), rather than unnecessarily computing the derivatives of the values of hidden layers with respect to changes in weights [math]\displaystyle{ \partial a^{l'}_{j'}/\partial w^l_{jk} }[/math].

Backpropagation can be expressed for simple feedforward networks in terms of matrix multiplication, or more generally in terms of the adjoint graph.

Matrix multiplication

For the basic case of a feedforward network, where nodes in each layer are connected only to nodes in the immediate next layer (without skipping any layers), and there is a loss function that computes a scalar loss for the final output, backpropagation can be understood simply by matrix multiplication.[lower-alpha 3] Essentially, backpropagation evaluates the expression for the derivative of the cost function as a product of derivatives between each layer from right to left – "backwards" – with the gradient of the weights between each layer being a simple modification of the partial products (the "backwards propagated error").

Given an input–output pair [math]\displaystyle{ (x, y) }[/math], the loss is:

[math]\displaystyle{ C(y, f^L(W^L f^{L-1}(W^{L-1} \cdots f^2(W^2 f^1(W^1 x))\cdots))) }[/math]

To compute this, one starts with the input [math]\displaystyle{ x }[/math] and works forward; denote the weighted input of each hidden layer as [math]\displaystyle{ z^l }[/math] and the output of hidden layer [math]\displaystyle{ l }[/math] as the activation [math]\displaystyle{ a^l }[/math]. For backpropagation, the activation [math]\displaystyle{ a^l }[/math] as well as the derivatives [math]\displaystyle{ (f^l)' }[/math] (evaluated at [math]\displaystyle{ z^l }[/math]) must be cached for use during the backwards pass.

The derivative of the loss in terms of the inputs is given by the chain rule; note that each term is a total derivative, evaluated at the value of the network (at each node) on the input [math]\displaystyle{ x }[/math]:

[math]\displaystyle{ \frac{d C}{d a^L}\cdot \frac{d a^L}{d z^L} \cdot \frac{d z^L}{d a^{L-1}} \cdot \frac{d a^{L-1}}{d z^{L-1}}\cdot \frac{d z^{L-1}}{d a^{L-2}} \cdot \ldots \cdot \frac{d a^1}{d z^1} \cdot \frac{\partial z^1}{\partial x}, }[/math]

where [math]\displaystyle{ \frac{d a^L}{d z^L} }[/math] is a diagonal matrix.

These terms are: the derivative of the loss function;[lower-alpha 4] the derivatives of the activation functions;[lower-alpha 5] and the matrices of weights:[lower-alpha 6]

[math]\displaystyle{ \frac{d C}{d a^L}\circ (f^L)' \cdot W^L \circ (f^{L-1})' \cdot W^{L-1} \circ \cdots \circ (f^1)' \cdot W^1. }[/math]

The gradient [math]\displaystyle{ \nabla }[/math] is the transpose of the derivative of the output in terms of the input, so the matrices are transposed and the order of multiplication is reversed, but the entries are the same:

[math]\displaystyle{ \nabla_x C = (W^1)^T \cdot (f^1)' \circ \ldots \circ (W^{L-1})^T \cdot (f^{L-1})' \circ (W^L)^T \cdot (f^L)' \circ \nabla_{a^L} C. }[/math]

Backpropagation then consists essentially of evaluating this expression from right to left (equivalently, multiplying the previous expression for the derivative from left to right), computing the gradient at each layer on the way; there is an added step, because the gradient of the weights is not just a subexpression: there's an extra multiplication.

Introducing the auxiliary quantity [math]\displaystyle{ \delta^l }[/math] for the partial products (multiplying from right to left), interpreted as the "error at level [math]\displaystyle{ l }[/math]" and defined as the gradient of the input values at level [math]\displaystyle{ l }[/math]:

[math]\displaystyle{ \delta^l := (f^l)' \circ (W^{l+1})^T\cdot(f^{l+1})' \circ \cdots \circ (W^{L-1})^T \cdot (f^{L-1})' \circ (W^L)^T \cdot (f^L)' \circ \nabla_{a^L} C. }[/math]

Note that [math]\displaystyle{ \delta^l }[/math] is a vector, of length equal to the number of nodes in level [math]\displaystyle{ l }[/math]; each component is interpreted as the "cost attributable to (the value of) that node".

The gradient of the weights in layer [math]\displaystyle{ l }[/math] is then:

[math]\displaystyle{ \nabla_{W^l} C = \delta^l(a^{l-1})^T. }[/math]

The factor of [math]\displaystyle{ a^{l-1} }[/math] is because the weights [math]\displaystyle{ W^l }[/math] between level [math]\displaystyle{ l - 1 }[/math] and [math]\displaystyle{ l }[/math] affect level [math]\displaystyle{ l }[/math] proportionally to the inputs (activations): the inputs are fixed, the weights vary.

The [math]\displaystyle{ \delta^l }[/math] can easily be computed recursively, going from right to left, as:

[math]\displaystyle{ \delta^{l-1} := (f^{l-1})' \circ (W^l)^T \cdot \delta^l. }[/math]

The gradients of the weights can thus be computed using a few matrix multiplications for each level; this is backpropagation.

Compared with naively computing forwards (using the [math]\displaystyle{ \delta^l }[/math] for illustration):

[math]\displaystyle{ \begin{align} \delta^1 &= (f^1)' \circ (W^2)^T \cdot (f^2)' \circ \cdots \circ (W^{L-1})^T \cdot (f^{L-1})' \circ (W^L)^T \cdot (f^L)' \circ \nabla_{a^L} C\\ \delta^2 &= (f^2)' \circ \cdots \circ (W^{L-1})^T \cdot (f^{L-1})' \circ (W^L)^T \cdot (f^L)' \circ \nabla_{a^L} C\\ &\vdots\\ \delta^{L-1} &= (f^{L-1})' \circ (W^L)^T \cdot (f^L)' \circ \nabla_{a^L} C\\ \delta^L &= (f^L)' \circ \nabla_{a^L} C, \end{align} }[/math]

there are two key differences with backpropagation:

  1. Computing [math]\displaystyle{ \delta^{l-1} }[/math] in terms of [math]\displaystyle{ \delta^l }[/math] avoids the obvious duplicate multiplication of layers [math]\displaystyle{ l }[/math] and beyond.
  2. Multiplying starting from [math]\displaystyle{ \nabla_{a^L} C }[/math] – propagating the error backwards – means that each step simply multiplies a vector ([math]\displaystyle{ \delta^l }[/math]) by the matrices of weights [math]\displaystyle{ (W^l)^T }[/math] and derivatives of activations [math]\displaystyle{ (f^{l-1})' }[/math]. By contrast, multiplying forwards, starting from the changes at an earlier layer, means that each multiplication multiplies a matrix by a matrix. This is much more expensive, and corresponds to tracking every possible path of a change in one layer [math]\displaystyle{ l }[/math] forward to changes in the layer [math]\displaystyle{ l+2 }[/math] (for multiplying [math]\displaystyle{ W^{l+1} }[/math] by [math]\displaystyle{ W^{l+2} }[/math], with additional multiplications for the derivatives of the activations), which unnecessarily computes the intermediate quantities of how weight changes affect the values of hidden nodes.

Adjoint graph

For more general graphs, and other advanced variations, backpropagation can be understood in terms of automatic differentiation, where backpropagation is a special case of reverse accumulation (or "reverse mode").[9]

Intuition

Motivation

The goal of any supervised learning algorithm is to find a function that best maps a set of inputs to their correct output. The motivation for backpropagation is to train a multi-layered neural network such that it can learn the appropriate internal representations to allow it to learn any arbitrary mapping of input to output.[19]

Learning as an optimization problem

To understand the mathematical derivation of the backpropagation algorithm, it helps to first develop some intuition about the relationship between the actual output of a neuron and the correct output for a particular training example. Consider a simple neural network with two input units, one output unit and no hidden units, and in which each neuron uses a linear output (unlike most work on neural networks, in which mapping from inputs to outputs is non-linear)[lower-alpha 7] that is the weighted sum of its input.

A simple neural network with two input units (each with a single input) and one output unit (with two inputs)

Initially, before training, the weights will be set randomly. Then the neuron learns from training examples, which in this case consist of a set of tuples [math]\displaystyle{ (x_1, x_2, t) }[/math] where [math]\displaystyle{ x_1 }[/math] and [math]\displaystyle{ x_2 }[/math] are the inputs to the network and t is the correct output (the output the network should produce given those inputs, when it has been trained). The initial network, given [math]\displaystyle{ x_1 }[/math] and [math]\displaystyle{ x_2 }[/math], will compute an output y that likely differs from t (given random weights). A loss function [math]\displaystyle{ L(t, y) }[/math] is used for measuring the discrepancy between the target output t and the computed output y. For regression analysis problems the squared error can be used as a loss function, for classification the categorical cross-entropy can be used.

As an example consider a regression problem using the square error as a loss:

[math]\displaystyle{ L(t, y)= (t-y)^2 = E, }[/math]

where E is the discrepancy or error.

Consider the network on a single training case: [math]\displaystyle{ (1, 1, 0) }[/math]. Thus, the input [math]\displaystyle{ x_1 }[/math] and [math]\displaystyle{ x_2 }[/math] are 1 and 1 respectively and the correct output, t is 0. Now if the relation is plotted between the network's output y on the horizontal axis and the error E on the vertical axis, the result is a parabola. The minimum of the parabola corresponds to the output y which minimizes the error E. For a single training case, the minimum also touches the horizontal axis, which means the error will be zero and the network can produce an output y that exactly matches the target output t. Therefore, the problem of mapping inputs to outputs can be reduced to an optimization problem of finding a function that will produce the minimal error.

Error surface of a linear neuron for a single training case

However, the output of a neuron depends on the weighted sum of all its inputs:

[math]\displaystyle{ y=x_1w_1 + x_2w_2, }[/math]

where [math]\displaystyle{ w_1 }[/math] and [math]\displaystyle{ w_2 }[/math] are the weights on the connection from the input units to the output unit. Therefore, the error also depends on the incoming weights to the neuron, which is ultimately what needs to be changed in the network to enable learning.

In this example, upon injecting the training data [math]\displaystyle{ (1, 1, 0) }[/math], the loss function becomes

[math]\displaystyle{ E = (t-y)^2 = y^2 = (x_1w_1 + x_2w_2)^2 = (w_1 + w_2)^2. }[/math]

Then, the loss function [math]\displaystyle{ E }[/math] takes the form of a parabolic cylinder with its base directed along [math]\displaystyle{ w_1 = -w_2 }[/math]. Since all sets of weights that satisfy [math]\displaystyle{ w_1 = -w_2 }[/math] minimize the loss function, in this case additional constraints are required to converge to a unique solution. Additional constraints could either be generated by setting specific conditions to the weights, or by injecting additional training data.

One commonly used algorithm to find the set of weights that minimizes the error is gradient descent. By backpropagation, the steepest descent direction is calculated of the loss function versus the present synaptic weights. Then, the weights can be modified along the steepest descent direction, and the error is minimized in an efficient way.

Derivation

The gradient descent method involves calculating the derivative of the loss function with respect to the weights of the network. This is normally done using backpropagation. Assuming one output neuron,[lower-alpha 8] the squared error function is

[math]\displaystyle{ E = L(t, y) }[/math]

where

[math]\displaystyle{ L }[/math] is the loss for the output [math]\displaystyle{ y }[/math] and target value [math]\displaystyle{ t }[/math],
[math]\displaystyle{ t }[/math] is the target output for a training sample, and
[math]\displaystyle{ y }[/math] is the actual output of the output neuron.

For each neuron [math]\displaystyle{ j }[/math], its output [math]\displaystyle{ o_j }[/math] is defined as

[math]\displaystyle{ o_j = \varphi(\text{net}_j) = \varphi\left(\sum_{k=1}^n w_{kj}x_k\right), }[/math]

where the activation function [math]\displaystyle{ \varphi }[/math] is non-linear and differentiable over the activation region (the ReLU is not differentiable at one point). A historically used activation function is the logistic function:

[math]\displaystyle{ \varphi(z) = \frac 1 {1+e^{-z}} }[/math]

which has a convenient derivative of:

[math]\displaystyle{ \frac {d \varphi}{d z} = \varphi(z)(1-\varphi(z)) }[/math]

The input [math]\displaystyle{ \text{net}_j }[/math] to a neuron is the weighted sum of outputs [math]\displaystyle{ o_k }[/math] of previous neurons. If the neuron is in the first layer after the input layer, the [math]\displaystyle{ o_k }[/math] of the input layer are simply the inputs [math]\displaystyle{ x_k }[/math] to the network. The number of input units to the neuron is [math]\displaystyle{ n }[/math]. The variable [math]\displaystyle{ w_{kj} }[/math] denotes the weight between neuron [math]\displaystyle{ k }[/math] of the previous layer and neuron [math]\displaystyle{ j }[/math] of the current layer.

Finding the derivative of the error

Diagram of an artificial neural network to illustrate the notation used here

Calculating the partial derivative of the error with respect to a weight [math]\displaystyle{ w_{ij} }[/math] is done using the chain rule twice:

[math]\displaystyle{ \frac{\partial E}{\partial w_{ij}} = \frac{\partial E}{\partial o_{j}} \frac{\partial o_{j}}{\partial w_{ij}} = \frac{\partial E}{\partial o_j} \frac{\partial o_j}{\partial\text{net}_j} \frac{\partial \text{net}_j}{\partial w_{ij}} }[/math]

 

 

 

 

(Eq. 1)

In the last factor of the right-hand side of the above, only one term in the sum [math]\displaystyle{ \text{net}_j }[/math] depends on [math]\displaystyle{ w_{ij} }[/math], so that

[math]\displaystyle{ \frac{\partial \text{net}_j}{\partial w_{ij}} = \frac{\partial}{\partial w_{ij}} \left(\sum_{k=1}^n w_{kj} o_k\right) = \frac{\partial}{\partial w_{ij}} w_{ij} o_i= o_i. }[/math]

 

 

 

 

(Eq. 2)

If the neuron is in the first layer after the input layer, [math]\displaystyle{ o_i }[/math] is just [math]\displaystyle{ x_i }[/math].

The derivative of the output of neuron [math]\displaystyle{ j }[/math] with respect to its input is simply the partial derivative of the activation function:

[math]\displaystyle{ \frac{\partial o_j}{\partial\text{net}_j} = \frac {\partial \varphi(\text{net}_j)}{\partial \text{net}_j} }[/math]

 

 

 

 

(Eq. 3)

which for the logistic activation function

[math]\displaystyle{ \frac{\partial o_j}{\partial\text{net}_j} = \frac {\partial}{\partial \text{net}_j} \varphi(\text{net}_j) = \varphi(\text{net}_j)(1-\varphi(\text{net}_j)) = o_j(1-o_j) }[/math]

This is the reason why backpropagation requires that the activation function be differentiable. (Nevertheless, the ReLU activation function, which is non-differentiable at 0, has become quite popular, e.g. in AlexNet)

The first factor is straightforward to evaluate if the neuron is in the output layer, because then [math]\displaystyle{ o_j = y }[/math] and

[math]\displaystyle{ \frac{\partial E}{\partial o_j} = \frac{\partial E}{\partial y} }[/math]

 

 

 

 

(Eq. 4)

If half of the square error is used as loss function we can rewrite it as

[math]\displaystyle{ \frac{\partial E}{\partial o_j} = \frac{\partial E}{\partial y} = \frac{\partial}{\partial y} \frac{1}{2}(t - y)^2 = y - t }[/math]

However, if [math]\displaystyle{ j }[/math] is in an arbitrary inner layer of the network, finding the derivative [math]\displaystyle{ E }[/math] with respect to [math]\displaystyle{ o_j }[/math] is less obvious.

Considering [math]\displaystyle{ E }[/math] as a function with the inputs being all neurons [math]\displaystyle{ L = \{u, v, \dots, w\} }[/math] receiving input from neuron [math]\displaystyle{ j }[/math],

[math]\displaystyle{ \frac{\partial E(o_j)}{\partial o_j} = \frac{\partial E(\mathrm{net}_u, \text{net}_v, \dots, \mathrm{net}_w)}{\partial o_j} }[/math]

and taking the total derivative with respect to [math]\displaystyle{ o_j }[/math], a recursive expression for the derivative is obtained:

[math]\displaystyle{ \frac{\partial E}{\partial o_j} = \sum_{\ell \in L} \left(\frac{\partial E}{\partial \text{net}_\ell}\frac{\partial \text{net}_\ell}{\partial o_j}\right) = \sum_{\ell \in L} \left(\frac{\partial E}{\partial o_\ell}\frac{\partial o_\ell}{\partial \text{net}_\ell}\frac{\partial \text{net}_\ell}{\partial o_j}\right) = \sum_{\ell \in L} \left(\frac{\partial E}{\partial o_\ell}\frac{\partial o_\ell}{\partial \text{net}_\ell}w_{j \ell}\right) }[/math]

 

 

 

 

(Eq. 5)

Therefore, the derivative with respect to [math]\displaystyle{ o_j }[/math] can be calculated if all the derivatives with respect to the outputs [math]\displaystyle{ o_\ell }[/math] of the next layer – the ones closer to the output neuron – are known. [Note, if any of the neurons in set [math]\displaystyle{ L }[/math] were not connected to neuron [math]\displaystyle{ j }[/math], they would be independent of [math]\displaystyle{ w_{ij} }[/math] and the corresponding partial derivative under the summation would vanish to 0.]

Substituting Eq. 2, Eq. 3 Eq.4 and Eq. 5 in Eq. 1 we obtain:

[math]\displaystyle{ \frac{\partial E}{\partial w_{ij}} = \frac{\partial E}{\partial o_{j}} \frac{\partial o_{j}}{\partial \text{net}_{j}} \frac{\partial \text{net}_{j}}{\partial w_{ij}} = \frac{\partial E}{\partial o_{j}} \frac{\partial o_{j}}{\partial \text{net}_{j}} o_i }[/math]
[math]\displaystyle{ \frac{\partial E}{\partial w_{ij}} = o_i \delta_j }[/math]

with

[math]\displaystyle{ \delta_j = \frac{\partial E}{\partial o_j} \frac{\partial o_j}{\partial\text{net}_j} = \begin{cases} \frac{\partial L(t, o_j)}{\partial o_j} \frac {d \varphi(\text{net}_j)}{d \text{net}_j} & \text{if } j \text{ is an output neuron,}\\ (\sum_{\ell\in L} w_{j \ell} \delta_\ell)\frac {d \varphi(\text{net}_j)}{d \text{net}_j} & \text{if } j \text{ is an inner neuron.} \end{cases} }[/math]

if [math]\displaystyle{ \varphi }[/math] is the logistic function, and the error is the square error:

[math]\displaystyle{ \delta_j = \frac{\partial E}{\partial o_j} \frac{\partial o_j}{\partial\text{net}_j} = \begin{cases} (o_j-t_j)o_j(1-o_{j}) & \text{if } j \text{ is an output neuron,}\\ (\sum_{\ell\in L} w_{j \ell} \delta_\ell)o_j(1-o_j) & \text{if } j \text{ is an inner neuron.} \end{cases} }[/math]

To update the weight [math]\displaystyle{ w_{ij} }[/math] using gradient descent, one must choose a learning rate, [math]\displaystyle{ \eta \gt 0 }[/math]. The change in weight needs to reflect the impact on [math]\displaystyle{ E }[/math] of an increase or decrease in [math]\displaystyle{ w_{ij} }[/math]. If [math]\displaystyle{ \frac{\partial E}{\partial w_{ij}} \gt 0 }[/math], an increase in [math]\displaystyle{ w_{ij} }[/math] increases [math]\displaystyle{ E }[/math]; conversely, if [math]\displaystyle{ \frac{\partial E}{\partial w_{ij}} \lt 0 }[/math], an increase in [math]\displaystyle{ w_{ij} }[/math] decreases [math]\displaystyle{ E }[/math]. The new [math]\displaystyle{ \Delta w_{ij} }[/math] is added to the old weight, and the product of the learning rate and the gradient, multiplied by [math]\displaystyle{ -1 }[/math] guarantees that [math]\displaystyle{ w_{ij} }[/math] changes in a way that always decreases [math]\displaystyle{ E }[/math]. In other words, in the equation immediately below, [math]\displaystyle{ - \eta \frac{\partial E}{\partial w_{ij}} }[/math] always changes [math]\displaystyle{ w_{ij} }[/math] in such a way that [math]\displaystyle{ E }[/math] is decreased:

[math]\displaystyle{ \Delta w_{ij} = - \eta \frac{\partial E}{\partial w_{ij}} = - \eta o_i \delta_j }[/math]

Second-order gradient descent

Using a Hessian matrix of second-order derivatives of the error function, the Levenberg–Marquardt algorithm often converges faster than first-order gradient descent, especially when the topology of the error function is complicated.[20][21] It may also find solutions in smaller node counts for which other methods might not converge.[21] The Hessian can be approximated by the Fisher information matrix.[22]

Loss function

The loss function is a function that maps values of one or more variables onto a real number intuitively representing some "cost" associated with those values. For backpropagation, the loss function calculates the difference between the network output and its expected output, after a training example has propagated through the network.

Assumptions

The mathematical expression of the loss function must fulfill two conditions in order for it to be possibly used in backpropagation.[23] The first is that it can be written as an average [math]\displaystyle{ E=\frac{1}{n}\sum_xE_x }[/math] over error functions [math]\displaystyle{ E_x }[/math], for [math]\displaystyle{ n }[/math] individual training examples, [math]\displaystyle{ x }[/math]. The reason for this assumption is that the backpropagation algorithm calculates the gradient of the error function for a single training example, which needs to be generalized to the overall error function. The second assumption is that it can be written as a function of the outputs from the neural network.

Example loss function

Let [math]\displaystyle{ y,y' }[/math] be vectors in [math]\displaystyle{ \mathbb{R}^n }[/math].

Select an error function [math]\displaystyle{ E(y,y') }[/math] measuring the difference between two outputs. The standard choice is the square of the Euclidean distance between the vectors [math]\displaystyle{ y }[/math] and [math]\displaystyle{ y' }[/math]:[math]\displaystyle{ E(y,y') = \tfrac{1}{2} \lVert y-y'\rVert^2 }[/math]The error function over [math]\displaystyle{ n }[/math] training examples can then be written as an average of losses over individual examples:[math]\displaystyle{ E=\frac{1}{2n}\sum_x\lVert (y(x)-y'(x)) \rVert^2 }[/math]

Limitations

Gradient descent may find a local minimum instead of the global minimum.
  • Gradient descent with backpropagation is not guaranteed to find the global minimum of the error function, but only a local minimum; also, it has trouble crossing plateaus in the error function landscape. This issue, caused by the non-convexity of error functions in neural networks, was long thought to be a major drawback, but Yann LeCun et al. argue that in many practical problems, it is not.[24]
  • Backpropagation learning does not require normalization of input vectors; however, normalization could improve performance.[25]
  • Backpropagation requires the derivatives of activation functions to be known at network design time.

History

Precursors

Backpropagation had been derived repeatedly, as it is essentially an efficient application of the chain rule (first written down by Gottfried Wilhelm Leibniz in 1676[1][26]) to neural networks.

The terminology "back-propagating error correction" was introduced in 1962 by Frank Rosenblatt, but he did not know how to implement this.[27] In any case, he only studied neurons whose outputs were discrete levels, which only had zero derivatives, making backpropagation impossible.

Precursors to backpropagation appeared in optimal control theory since 1950s. Yann leCun et al credits 1950s work by Pontryagin and others, in optimal control theory, especially the adjoint state method, for being a continuous-time version of backpropagation.[28] Hecht-Nielsen[29] credits Robbins–Monro algorithm (1951) and Arthur Bryson and Yu-Chi Ho's Applied Optimal Control (1969) as presages of backpropagation. Other precursors were Henry J. Kelley 1960,[11] and Arthur E. Bryson (1961).[12] In 1962, Stuart Dreyfus published a simpler derivation based only on the chain rule.[30][31][32] In 1973, he adapted parameters of controllers in proportion to error gradients.[33] Unlike modern backpropagation, these precursors used standard Jacobian matrix calculations from one stage to the previous one, neither addressing direct links across several stages nor potential additional efficiency gains due to network sparsity.[2]

The ADALINE (1960) learning algorithm was gradient descent with a squared error loss for a single layer. The first multilayer perceptron (MLP) with more than one layer trained by stochastic gradient descent[14] was published in 1967 by Shun'ichi Amari.[34][2] The MLP had 5 layers, with 2 learnable layers, and it learned to classify patterns not linearly separable.[2]

Modern backpropagation

Modern backpropagation was first published by Seppo Linnainmaa as "reverse mode of automatic differentiation" (1970)[3] for discrete connected networks of nested differentiable functions.[4][5]

In 1982, Paul Werbos applied backpropagation to MLPs in the way that has become standard.[35][36] Werbos described how he developed backpropagation in an interview. In 1971, during his PhD work, he developed backpropagation to mathematicize Freud's "flow of psychic energy". He faced repeated difficulty in publishing the work, only managing in 1981.[37]

Around 1982,[37]:376 David E. Rumelhart independently developed[38]:252 backpropagation and taught the algorithm to others in his research circle. He did not cite previous work as he was unaware of them. He published the algorithm first in a 1985 paper, then in a 1986 Nature paper an experimental analysis of the technique.[16] These papers became highly cited, contributed to the popularization of backpropagation, and coincided with the resurging research interest in neural networks during the 1980s.[19][39][40]

In 1985, the method was also described by David Parker.[41][42] Yann LeCun proposed an alternative form of backpropagation for neural networks in his PhD thesis in 1987.[43]

Gradient descent took a considerable amount of time to reach acceptance. Some early objections were: there were no guarantees that gradient descent could reach a global minimum, only local minimum; neurons were "known" by physiologists as making discrete signals (0/1), not continuous ones, and with discrete signals, there is no gradient to take. See the interview with Geoffrey Hinton.[37]

Early successes

Contributing to the acceptance were several applications in training neural networks via backpropagation, sometimes achieving popularity outside the research circles.

In 1987, NETtalk learned to convert English text into pronunciation. Sejnowski tried training it with both backpropagation and Boltzmann machine, but found the backpropagation significantly faster, so he used it for the final NETtalk.[37]:{{{1}}} The NETtalk program became a popular success, appearing on the Today show.[44]

In 1989, Dean A. Pomerleau published ALVINN, a neural network trained to drive autonomously using backpropagation.[45]

The LeNet was published in 1989 to recognize handwritten zip codes.

In 1992, TD-Gammon achieved top human level play in backgammon. It was a reinforcement learning agent with a neural network with two layers, trained by backpropagation.[46]

In 1993, Eric Wan won an international pattern recognition contest through backpropagation.[7][47]

After backpropagation

During the 2000s it fell out of favour[citation needed], but returned in the 2010s, benefiting from cheap, powerful GPU-based computing systems. This has been especially so in speech recognition, machine vision, natural language processing, and language structure learning research (in which it has been used to explain a variety of phenomena related to first[48] and second language learning.[49]).

Error backpropagation has been suggested to explain human brain ERP components like the N400 and P600.[50]

In 2023, a backpropagation algorithm was implemented on a photonic processor by a team at Stanford University.[51]

See also

Notes

  1. Use [math]\displaystyle{ C }[/math] for the loss function to allow [math]\displaystyle{ L }[/math] to be used for the number of layers
  2. This follows (Nielsen 2015), and means (left) multiplication by the matrix [math]\displaystyle{ W^l }[/math] corresponds to converting output values of layer [math]\displaystyle{ l - 1 }[/math] to input values of layer [math]\displaystyle{ l }[/math]: columns correspond to input coordinates, rows correspond to output coordinates.
  3. This section largely follows and summarizes (Nielsen 2015).
  4. The derivative of the loss function is a covector, since the loss function is a scalar-valued function of several variables.
  5. The activation function is applied to each node separately, so the derivative is just the diagonal matrix of the derivative on each node. This is often represented as the Hadamard product with the vector of derivatives, denoted by [math]\displaystyle{ (f^l)'\odot }[/math], which is mathematically identical but better matches the internal representation of the derivatives as a vector, rather than a diagonal matrix.
  6. Since matrix multiplication is linear, the derivative of multiplying by a matrix is just the matrix: [math]\displaystyle{ (Wx)' = W }[/math].
  7. One may notice that multi-layer neural networks use non-linear activation functions, so an example with linear neurons seems obscure. However, even though the error surface of multi-layer networks are much more complicated, locally they can be approximated by a paraboloid. Therefore, linear neurons are used for simplicity and easier understanding.
  8. There can be multiple output neurons, in which case the error is the squared norm of the difference vector.

References

  1. 1.0 1.1 Leibniz, Gottfried Wilhelm Freiherr von (1920) (in en). The Early Mathematical Manuscripts of Leibniz: Translated from the Latin Texts Published by Carl Immanuel Gerhardt with Critical and Historical Notes (Leibniz published the chain rule in a 1676 memoir). Open court publishing Company. ISBN 9780598818461. https://books.google.com/books?id=bOIGAAAAYAAJ&q=leibniz+altered+manuscripts&pg=PA90. 
  2. 2.0 2.1 2.2 2.3 2.4 2.5 Schmidhuber, Juergen (2022). "Annotated History of Modern AI and Deep Learning". arXiv:2212.11279 [cs.NE].
  3. 3.0 3.1 Linnainmaa, Seppo (1970). The representation of the cumulative rounding error of an algorithm as a Taylor expansion of the local rounding errors (Masters) (in suomi). University of Helsinki. pp. 6–7.
  4. 4.0 4.1 Linnainmaa, Seppo (1976). "Taylor expansion of the accumulated rounding error". BIT Numerical Mathematics 16 (2): 146–160. doi:10.1007/bf01931367. 
  5. 5.0 5.1 Griewank, Andreas (2012). "Who Invented the Reverse Mode of Differentiation?". Optimization Stories. Documenta Matematica, Extra Volume ISMP. pp. 389–400. 
  6. Griewank, Andreas; Walther, Andrea (2008). Evaluating Derivatives: Principles and Techniques of Algorithmic Differentiation, Second Edition. SIAM. ISBN 978-0-89871-776-1. https://books.google.com/books?id=xoiiLaRxcbEC. 
  7. 7.0 7.1 Schmidhuber, Jürgen (2015). "Deep learning in neural networks: An overview". Neural Networks 61: 85–117. doi:10.1016/j.neunet.2014.09.003. PMID 25462637. 
  8. Schmidhuber, Jürgen (2015). "Deep Learning". Scholarpedia 10 (11): 32832. doi:10.4249/scholarpedia.32832. Bibcode2015SchpJ..1032832S. 
  9. 9.0 9.1 (Goodfellow Bengio), "The back-propagation algorithm described here is only one approach to automatic differentiation. It is a special case of a broader class of techniques called reverse mode accumulation."
  10. Rosenblatt, Frank (1962). Principles of Neurodynamics: Perceptrons and the Theory of Brain Mechanisms Cornell Aeronautical Laboratory. Report no. VG-1196-G-8 Report (Cornell Aeronautical Laboratory). Spartan. pp. Page XIII Table of contents, Page 292 "13.3 Back-Propagating Error Correction Procedures", Page 301 "figure 39 BACK-PROPAGATING ERROR-CORRECTION EXPERIMENTS". 
  11. 11.0 11.1 11.2 Kelley, Henry J. (1960). "Gradient theory of optimal flight paths". ARS Journal 30 (10): 947–954. doi:10.2514/8.5282. 
  12. 12.0 12.1 Bryson, Arthur E. (1962). "A gradient method for optimizing multi-stage allocation processes". Proceedings of the Harvard Univ. Symposium on digital computers and their applications, 3–6 April 1961. Cambridge: Harvard University Press. OCLC 498866871. 
  13. Goodfellow, Bengio & Courville 2016, p. 214, "This table-filling strategy is sometimes called dynamic programming."
  14. 14.0 14.1 Robbins, H.; Monro, S. (1951). "A Stochastic Approximation Method". The Annals of Mathematical Statistics 22 (3): 400. doi:10.1214/aoms/1177729586. 
  15. Goodfellow, Bengio & Courville 2016, p. 200, "The term back-propagation is often misunderstood as meaning the whole learning algorithm for multilayer neural networks. Backpropagation refers only to the method for computing the gradient, while other algorithms, such as stochastic gradient descent, is used to perform learning using this gradient."
  16. 16.0 16.1 Rumelhart; Hinton; Williams (1986). "Learning representations by back-propagating errors". Nature 323 (6088): 533–536. doi:10.1038/323533a0. Bibcode1986Natur.323..533R. http://www.cs.toronto.edu/~hinton/absps/naturebp.pdf. 
  17. Ramachandran, Prajit; Zoph, Barret; Le, Quoc V. (2017-10-27). "Searching for Activation Functions". arXiv:1710.05941 [cs.NE].
  18. Misra, Diganta (2019-08-23). "Mish: A Self Regularized Non-Monotonic Activation Function". arXiv:1908.08681 [cs.LG].
  19. 19.0 19.1 Rumelhart, David E.; Hinton, Geoffrey E.; Williams, Ronald J. (1986a). "Learning representations by back-propagating errors". Nature 323 (6088): 533–536. doi:10.1038/323533a0. Bibcode1986Natur.323..533R. 
  20. Tan, Hong Hui; Lim, King Han (2019). "Review of second-order optimization techniques in artificial neural networks backpropagation". IOP Conference Series: Materials Science and Engineering 495 (1): 012003. doi:10.1088/1757-899X/495/1/012003. Bibcode2019MS&E..495a2003T. 
  21. 21.0 21.1 Wiliamowski, Bogdan; Yu, Hao (June 2010). "Improved Computation for Levenberg–Marquardt Training". IEEE Transactions on Neural Networks and Learning Systems 21 (6). https://www.eng.auburn.edu/~wilambm/pap/2010/Improved%20Computation%20for%20LM%20Training.pdf. 
  22. Martens, James (August 2020). "New Insights and Perspectives on the Natural Gradient Method". Journal of Machine Learning Research (21). 
  23. (Nielsen 2015), "[W]hat assumptions do we need to make about our cost function ... in order that backpropagation can be applied? The first assumption we need is that the cost function can be written as an average ... over cost functions ... for individual training examples ... The second assumption we make about the cost is that it can be written as a function of the outputs from the neural network ..."
  24. LeCun, Yann; Bengio, Yoshua; Hinton, Geoffrey (2015). "Deep learning". Nature 521 (7553): 436–444. doi:10.1038/nature14539. PMID 26017442. Bibcode2015Natur.521..436L. 
  25. Buckland, Matt; Collins, Mark (2002). AI Techniques for Game Programming. Boston: Premier Press. ISBN 1-931841-08-X. 
  26. Rodríguez, Omar Hernández; López Fernández, Jorge M. (2010). "A Semiotic Reflection on the Didactics of the Chain Rule". The Mathematics Enthusiast 7 (2): 321–332. doi:10.54870/1551-3440.1191. https://scholarworks.umt.edu/tme/vol7/iss2/10/. Retrieved 2019-08-04. 
  27. Rosenblatt, Frank (1962). Principles of Neurodynamics. Spartan, New York. pp. 287–298. 
  28. LeCun, Yann, et al. "A theoretical framework for back-propagation." Proceedings of the 1988 connectionist models summer school. Vol. 1. 1988.
  29. Hecht-Nielsen, Robert (1990). Neurocomputing. Internet Archive. Reading, Mass. : Addison-Wesley Pub. Co.. pp. 124–125. ISBN 978-0-201-09355-1. http://archive.org/details/neurocomputing0000hech. 
  30. Dreyfus, Stuart (1962). "The numerical solution of variational problems". Journal of Mathematical Analysis and Applications 5 (1): 30–45. doi:10.1016/0022-247x(62)90004-5. 
  31. Dreyfus, Stuart E. (1990). "Artificial Neural Networks, Back Propagation, and the Kelley-Bryson Gradient Procedure". Journal of Guidance, Control, and Dynamics 13 (5): 926–928. doi:10.2514/3.25422. Bibcode1990JGCD...13..926D. 
  32. Mizutani, Eiji; Dreyfus, Stuart; Nishio, Kenichi (July 2000). "On derivation of MLP backpropagation from the Kelley-Bryson optimal-control gradient formula and its application". Proceedings of the IEEE International Joint Conference on Neural Networks. https://coeieor.wpengine.com/wp-content/uploads/2019/03/ijcnn2k.pdf. 
  33. Dreyfus, Stuart (1973). "The computational solution of optimal control problems with time lag". IEEE Transactions on Automatic Control 18 (4): 383–385. doi:10.1109/tac.1973.1100330. 
  34. Amari, Shun'ichi (1967). "A theory of adaptive pattern classifier". IEEE Transactions EC (16): 279–307. 
  35. Werbos, Paul (1982). "Applications of advances in nonlinear sensitivity analysis". System modeling and optimization. Springer. pp. 762–770. http://werbos.com/Neural/SensitivityIFIPSeptember1981.pdf. Retrieved 2 July 2017. 
  36. Werbos, Paul J. (1994). The Roots of Backpropagation : From Ordered Derivatives to Neural Networks and Political Forecasting. New York: John Wiley & Sons. ISBN 0-471-59897-6. 
  37. 37.0 37.1 37.2 37.3 Anderson, James A., ed (2000) (in en). Talking Nets: An Oral History of Neural Networks. The MIT Press. doi:10.7551/mitpress/6626.003.0016. ISBN 978-0-262-26715-1. https://direct.mit.edu/books/book/4886/Talking-NetsAn-Oral-History-of-Neural-Networks. 
  38. Olazaran Rodriguez, Jose Miguel. A historical sociology of neural network research. PhD Dissertation. University of Edinburgh, 1991.
  39. Rumelhart, David E.; Hinton, Geoffrey E.; Williams, Ronald J. (1986b). "8. Learning Internal Representations by Error Propagation". in Rumelhart, David E.; McClelland, James L.. Parallel Distributed Processing : Explorations in the Microstructure of Cognition. 1 : Foundations. Cambridge: MIT Press. ISBN 0-262-18120-7. https://archive.org/details/paralleldistribu00rume. 
  40. Alpaydin, Ethem (2010). Introduction to Machine Learning. MIT Press. ISBN 978-0-262-01243-0. https://books.google.com/books?id=4j9GAQAAIAAJ. 
  41. Parker, D.B. (1985). Learning Logic: Casting the Cortex of the Human Brain in Silicon (Report). Cambridge MA: Massachusetts Institute of Technology. Technical Report TR-47. 
  42. Hertz, John (1991). Introduction to the theory of neural computation. Krogh, Anders., Palmer, Richard G.. Redwood City, Calif.: Addison-Wesley. pp. 8. ISBN 0-201-50395-6. OCLC 21522159. 
  43. Le Cun, Yann (1987). Modèles connexionnistes de l'apprentissage (Thèse de doctorat d'état thesis). Paris, France: Université Pierre et Marie Curie.
  44. Sejnowski, Terrence J. (2018). The deep learning revolution. Cambridge, Massachusetts London, England: The MIT Press. ISBN 978-0-262-03803-4. 
  45. Pomerleau, Dean A. (1988). "ALVINN: An Autonomous Land Vehicle in a Neural Network". Advances in Neural Information Processing Systems (Morgan-Kaufmann) 1. https://proceedings.neurips.cc/paper/1988/hash/812b4ba287f5ee0bc9d43bbf5bbe87fb-Abstract.html. 
  46. Sutton, Richard S.; Barto, Andrew G. (2018). "11.1 TD-Gammon". Reinforcement Learning: An Introduction (2nd ed.). Cambridge, MA: MIT Press. http://www.incompleteideas.net/book/11/node2.html. 
  47. Wan, Eric A. (1994). "Time Series Prediction by Using a Connectionist Network with Internal Delay Lines". in Weigend, Andreas S.; Gershenfeld, Neil A.. Time Series Prediction : Forecasting the Future and Understanding the Past. Proceedings of the NATO Advanced Research Workshop on Comparative Time Series Analysis. 15. Reading: Addison-Wesley. pp. 195–217. ISBN 0-201-62601-2. 
  48. Chang, Franklin; Dell, Gary S.; Bock, Kathryn (2006). "Becoming syntactic.". Psychological Review 113 (2): 234–272. doi:10.1037/0033-295x.113.2.234. PMID 16637761. 
  49. Janciauskas, Marius; Chang, Franklin (2018). "Input and Age-Dependent Variation in Second Language Learning: A Connectionist Account". Cognitive Science 42 (Suppl Suppl 2): 519–554. doi:10.1111/cogs.12519. PMID 28744901. 
  50. Fitz, Hartmut; Chang, Franklin (2019). "Language ERPs reflect learning through prediction error propagation" (in en). Cognitive Psychology 111: 15–52. doi:10.1016/j.cogpsych.2019.03.002. PMID 30921626. 
  51. "Photonic Chips Curb AI Training's Energy Appetite - IEEE Spectrum" (in en). https://spectrum.ieee.org/backpropagation-optical-ai. 

Further reading

External links