Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

How to Train BERT with an Academic Budget

About

While large language models a la BERT are used ubiquitously in NLP, pretraining them is considered a luxury that only a few well-funded industry labs can afford. How can one train such models with a more modest budget? We present a recipe for pretraining a masked language model in 24 hours using a single low-end deep learning server. We demonstrate that through a combination of software optimizations, design choices, and hyperparameter tuning, it is possible to produce models that are competitive with BERT-base on GLUE tasks at a fraction of the original pretraining cost.

Peter Izsak, Moshe Berchansky, Omer Levy• 2021

Related benchmarks

TaskDatasetResultRank
Natural Language UnderstandingGLUE (test)
QNLI90.6
64
Showing 1 of 1 rows

Other info

Code

Follow for update