Gradient descent (original) (raw)

Gradient descent is an incremental hill-climbing algorithm that approaches a minimum or maximum of a function by taking steps proportional to the gradient (or the approximate gradient) at the current point.

There are two main forms of gradient descent commonly used in machine learning : batch and on-line.

In batch gradient descent, the true gradient is used to update the parameters of a model. The true gradient is usually the sum of the gradients caused by each individual training example. Therefore, batch gradient descent requires one sweep through the training set before any parameters can be changed.

In on-line gradient descent, the true gradient is approximated by the gradient of the cost function only evaluate on a single training example. Therefore, the parameters of the model are updated after each training example. For large data sets, on-line gradient descent can be much faster than batch gradient descent.

There is a compromise between the two forms, which is often called "mini-batches", where the true gradient is approximated by a sum over a small number of training examples.

On-line gradient descent is a form of stochastic approximation. The theory of stochastic approximation gives conditions on when on-line gradient descent converges.