Revisiting Neural Scaling Laws in Language and VisionDownload PDF

Published: 31 Oct 2022, 18:00, Last Modified: 11 Jan 2023, 13:49NeurIPS 2022 AcceptReaders: Everyone
Keywords: Scaling Laws, Computer Vision, Neural Machine Translation, Language Modeling, Big Bench
TL;DR: We present an algorithm for estimating scaling law parameters and extrapolating from learning curves.
Abstract: The remarkable progress in deep learning in recent years is largely driven by improvements in scale, where bigger models are trained on larger datasets for longer schedules. To predict the benefit of scale empirically, we argue for a more rigorous methodology based on the extrapolation loss, instead of reporting the best-fitting (interpolating) parameters. We then present a recipe for estimating scaling law parameters reliably from learning curves. We demonstrate that it extrapolates more accurately than previous methods in a wide range of architecture families across several domains, including image classification, neural machine translation (NMT) and language modeling, in addition to tasks from the BIG-Bench evaluation benchmark. Finally, we release a benchmark dataset comprising of 90 evaluation tasks to facilitate research in this domain.
Supplementary Material: pdf
13 Replies

Loading