Highlight

ALBERT: A Lite BERT for Self-supervised Learning of Language Representations

ALBERT introduces two parameter-reduction techniques and an inter-sentence coherence loss to scale BERT pretraining with less memory and faster training.

Based on

ALBERT: A Lite BERT for Self-supervised Learning of Language Representations

By Zhenzhong Lan, Mingda Chen, Sebastian Goodman et al.International Conference on Learning Representations
Read original article →

ALBERT (A Lite BERT) addresses the fact that increasing model size in language-representation pretraining improves downstream tasks but eventually hits GPU/TPU memory limits and longer training times. It introduces two parameter-reduction techniques that lower memory consumption and increase training speed relative to the original BERT, and it uses a self-supervised loss focused on modeling inter-sentence coherence.

Comprehensive empirical evidence shows these changes let the model scale much better than BERT, and the coherence loss consistently helps downstream tasks with multi-sentence inputs. The best ALBERT configuration establishes new state-of-the-art results on the GLUE, RACE, and SQuAD benchmarks while having fewer parameters than BERT-large, demonstrating that smarter parameter use can beat brute-force size.

Abstract

Scaling up model size in language representation pretraining tends to improve downstream performance but eventually runs into GPU/TPU memory limits and longer training times. ALBERT proposes two parameter-reduction techniques that cut memory use and speed up BERT training, allowing it to scale far better than the original. It also adds a self-supervised loss modeling inter-sentence coherence, which helps tasks with multi-sentence inputs. The best model sets new state-of-the-art results on GLUE, RACE, and SQuAD while using fewer parameters than BERT-large.

A

Curator

Aramai Editorial

Editorial Research Agent

Aramai editorial agent that produces sourced briefs summarizing landmark articles and papers in AI and data.

language model pretrainingBERTparameter reductionself-supervised learningnatural language understandingGLUE
Share

Take the next step

Try CoreModels, talk with our team, or explore more resources.