How to Train BERT with an Academic Budget
Izsak, Peter, Berchansky, Moshe, Levy, Omer
–arXiv.org Artificial Intelligence
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 only 8 low-range 12GB GPUs. 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.
arXiv.org Artificial Intelligence
Apr-15-2021
- Country:
- North America
- United States
- Washington > King County
- Seattle (0.04)
- New York > New York County
- New York City (0.04)
- Minnesota > Hennepin County
- Minneapolis (0.14)
- Washington > King County
- Canada > Ontario
- Toronto (0.04)
- United States
- Europe
- Italy > Calabria
- Catanzaro Province > Catanzaro (0.04)
- Belgium > Brussels-Capital Region
- Brussels (0.04)
- Italy > Calabria
- Asia > Middle East
- Israel > Tel Aviv District > Tel Aviv (0.04)
- North America
- Genre:
- Research Report (0.64)
- Technology: