Large-Batch Training

AI and Machine Learning · Training and Optimisation · 2017 · large-batch-training.yaml

Scales the learning rate linearly with batch size so that many devices can share one step. Holds up to surprisingly large batches, and only with a warmup in front of it.

Colour is the family; a dashed line is the second member of it.

Drag to pan · scroll to zoom · click a node to open it
G data-parallelism Data Parallelism large-batch-training Large-Batch Training large-batch-training->data-parallelism the batch is large because it is split across devices learning-rate-warmup Learning-Rate Warmup large-batch-training->learning-rate-warmup the scaled rate is unstable over the first epochs

This node

References