Batch size

From AI Wiki
(Redirected from Mini-batch size)
See also: machine learning terms

Introduction

Machine learning relies on a hyperparameter called batch size which indicates how many examples should be run before changing internal model parameters. It is the number of examples in a batch. This number can vary based on both machine memory capacity and the needs of each model and dataset.

Example

If the batch size is 50, then the model processes 50 examples per iteration. If the batch size is 200, then the model processes 200 examples per iteration.

Batch Size and Gradient Descent

Gradient descent relies on the batch size as a key parameter that determines how many samples are used in each iteration of the algorithm. Gradient descent works by iteratively updating model parameters in order to minimize cost function costs by computing gradients between them and batch size. Each iteration uses samples that make up this gradient calculation.

Batch Size and Stochastic Gradient Descent

Stochastic Gradient Descent (SGD) is a variant of gradient descent that uses only one example to update model parameters each iteration. Compared to batch gradient descent, which utilizes all examples for this task, SGD requires only one example for parameter updating - making it both computationally more efficient and faster but also prone to noise in gradient estimates.

Explain Like I'm 5 (ELI5)

The batch size is a number that instructs the computer how many examples to consider before it changes its interpretation of data patterns.