about
Revisiting Small Batch Training for Deep Neural Networks (arxiv.org)
3 points by tempw on Apr 26, 2018 | hide | past | pdf | discuss on HN

In plain words: They retested how batch size changes training by keeping the weight change per gradient fixed, instead of scaling the learning rate with batch size. Batches of 2 to 32 gave the best test accuracy; batches in the thousands needed a narrow learning-rate range.

Abstract

Modern deep neural network training is typically based on mini-batch stochastic gradient optimization. While the use of large mini-batches increases the available computational parallelism, small batch training has been shown to provide improved generalization performance and allows a significantly smaller memory footprint, which might also be exploited to improve machine throughput. In this paper, we review common assumptions on learning rate scaling and training duration, as a basis for an experimental comparison of test performance for different mini-batch sizes. We adopt a learning rate that corresponds to a constant average weight update per gradient calculation (i.e., per unit cost of computation), and point out that this results in a variance of the weight updates that increases linearly with the mini-batch size $m$. The collected experimental results for the CIFAR-10, CIFAR-100 and ImageNet datasets show that increasing the mini-batch size progressively reduces the range of learning rates that provide stable convergence and acceptable test performance. On the other hand, small mini-batch sizes provide more up-to-date gradient calculations, which yields more stable and reliable training. The best performance has been consistently obtained for mini-batch sizes between $m = 2$ and $m = 32$, which contrasts with recent work advocating the use of mini-batch sizes in the thousands.

Dominic Masters, Carlo Luschi
arXiv:1804.07612 · cs.LG, cs.CV, stat.ML · submitted Apr 20, 2018
abstract · pdf · html

add comment on HN
Also discussed: Jun 2020 (1 point, 0 comments) · Apr 2018 (2 points, 0 comments) · Apr 2018 (1 point, 0 comments)