ALBERT
ALBERT (A Lite BERT) is a transformer-based language model introduced by Google Research in 2019 that reduces BERT's parameter count through factorized embedding parameterization and cross-layer parameter sharing, while introducing a…
Definition
ALBERT (A Lite BERT) is a transformer-based language model introduced by Google Research in 2019 that reduces BERT's parameter count through factorized embedding parameterization and cross-layer parameter sharing, while introducing a sentence-order prediction pretraining task.
Overview
ALBERT was introduced in the 2019 paper "ALBERT: A Lite BERT for Self-supervised Learning of Language Representations," with the goal of making BERT-style models more parameter-efficient without sacrificing performance. It introduces two main parameter-reduction techniques. First, factorized embedding parameterization decomposes the large vocabulary embedding matrix into two smaller matrices, decoupling the size of the hidden layers from the size of the vocabulary embedding, which substantially reduces parameter count with minimal performance impact. Second, cross-layer parameter sharing has all transformer layers in the model share the same parameters, dramatically shrinking the total model size compared to BERT, where each layer has its own independent parameters. ALBERT also replaces BERT's next sentence prediction (NSP) pretraining task, which subsequent research found to be relatively ineffective, with a sentence-order prediction (SOP) task that specifically targets coherence between consecutive text segments, requiring the model to distinguish correctly ordered sentence pairs from swapped ones. This change proved more effective at improving downstream performance on tasks requiring an understanding of inter-sentence coherence. The combination of these techniques let ALBERT achieve comparable or better results than BERT on benchmarks like GLUE, RACE, and SQuAD while using far fewer parameters — for example, ALBERT-xxlarge achieves similar or better accuracy than BERT-large with fewer parameters, though it's worth noting that parameter sharing reduces memory footprint without proportionally reducing training or inference compute, since the same shared layer still runs multiple times per forward pass. ALBERT became a popular choice in settings where memory and storage constraints mattered more than raw compute, such as deploying models on devices with limited memory, and its design ideas around parameter efficiency influenced later work on compact and efficient transformer architectures.
Key Concepts
- Factorized embedding parameterization to reduce vocabulary embedding size
- Cross-layer parameter sharing across all transformer layers
- Sentence-order prediction (SOP) replacing BERT's next sentence prediction
- Significantly fewer parameters than BERT at comparable or better accuracy
- Reduces memory footprint, though not proportionally training/inference compute
- Released in Base, Large, xLarge, and xxLarge configurations