CLAIDec 18, 2024

Smarter, Better, Faster, Longer: A Modern Bidirectional Encoder for Fast, Memory Efficient, and Long Context Finetuning and Inference

arXiv:2412.13663v2622 citationsh-index: 20
Originality Highly original
AI Analysis

This work provides a major Pareto improvement for encoder-only models, benefiting production pipelines in retrieval and classification tasks with enhanced efficiency and performance.

The authors tackled the limited improvements in encoder-only transformer models like BERT by introducing ModernBERT, which achieved state-of-the-art results on diverse classification and retrieval tasks while being the most speed and memory efficient encoder for inference on common GPUs.

Encoder-only transformer models such as BERT offer a great performance-size tradeoff for retrieval and classification tasks with respect to larger decoder-only models. Despite being the workhorse of numerous production pipelines, there have been limited Pareto improvements to BERT since its release. In this paper, we introduce ModernBERT, bringing modern model optimizations to encoder-only models and representing a major Pareto improvement over older encoders. Trained on 2 trillion tokens with a native 8192 sequence length, ModernBERT models exhibit state-of-the-art results on a large pool of evaluations encompassing diverse classification tasks and both single and multi-vector retrieval on different domains (including code). In addition to strong downstream performance, ModernBERT is also the most speed and memory efficient encoder and is designed for inference on common GPUs.

Code Implementations2 repos
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes