Module juice::solvers::sgd[][src]

Expand description

Provides ISolver implementations based on [Stochastic Gradient Descent][2]. [2]: https://en.wikipedia.org/wiki/Stochastic_gradient_descent

One of the steps during backpropagation is determining the gradient for each weight. In theory this can be achieved very well using Gradient Descent (GD). In practice however applying GD when minimizing an objective function for large datasets, quickly becomes computationaly unfeasible.

Luckily GD can be approximated by taking a small random subset from the training dataset, commonly refered to as mini-batch. We then compute the gradients only for the samples from the mini-batch and average over these gradients, resulting in an estimate of the global gradient.
This method is refered to as Stochastic Gradient Descent.

Re-exports

pub use self::momentum::Momentum;

Modules

A [Stochastic Gradient Descent with Momentum][1] [1]: https://en.wikipedia.org/wiki/Stochastic_gradient_descent#Momentum