PreprintCited by 4,692Topic ModelingAdvanced Neural Network ApplicationsDomain Adaptation and Few-Shot Learning

The DOI matches its DataCite record; 23 of the 23 references it cites were found in a registry or an index; 3 of the 4 sources its claims cite were checked for support.

DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lighter

Open access
Sanh, Victor, Debut, Lysandre, Chaumond, Julien, Wolf, Thomas·2019·arXiv

Abstract

As Transfer Learning from large-scale pre-trained models becomes more prevalent in Natural Language Processing (NLP), operating these large models in on-the-edge and/or under constrained computational training or inference budgets remains challenging. In this work, we propose a method to pre-train a smaller general-purpose language representation model, called DistilBERT, which can then be fine-tuned with good performances on a wide range of tasks like its larger counterparts. While most prior work investigated the use of distillation for building task-specific models, we leverage knowledge distillation during the pre-training phase and show that it is possible to reduce the size of a BERT model by 40%, while retaining 97% of its language understanding capabilities and being 60% faster. To leverage the inductive biases learned by larger models during pre-training, we introduce a triple loss combining language modeling, distillation and cosine-distance losses. Our smaller, faster and lighter model is cheaper to pre-train and we demonstrate its capabilities for on-device computations in a proof-of-concept experiment and a comparative on-device study.

Overview

AI snapshot

Written from the paper’s full text.

TL;DR

DistilBERT is a distilled version of BERT pre-trained with knowledge distillation, reducing parameters by 40% while retaining 97% of BERT’s GLUE performance and running 60% faster at inference. It is presented as a general-purpose, lighter model suitable for constrained and on-device NLP applications.

Key findings

  • DistilBERT has 66M parameters vs. BERT-base’s 110M and takes 410s vs. 668s for a CPU inference pass on STS-B, i.e. 40% smaller and 60% faster.
  • On GLUE dev sets, DistilBERT scores 77.0 macro-score vs. BERT-base’s 79.5, retaining 97% of performance; it is always on par or improving over the ELMo baseline.
  • On IMDb test accuracy, DistilBERT reaches 92.82 vs. BERT-base’s 93.46; on SQuAD v1.1, DistilBERT reaches 77.7/85.8 EM/F1 vs. BERT-base’s 81.2/88.5.
  • Ablations show removing the cosine loss costs -2.96 GLUE macro-score, removing distillation loss costs -1.46, removing MLM loss costs -0.31, and random initialization costs -3.69.

Methods

  • Knowledge distillation during pre-training: a student with BERT’s architecture but half the layers, no token-type embeddings and no pooler, initialized from every other teacher layer.
  • Training objective is a triple loss combining distillation loss \(L_{ce}\), masked language modeling loss \(L_{mlm}\), and cosine embedding loss \(L_{cos}\), using softmax-temperature.
  • Trained on English Wikipedia + Toronto Book Corpus with large batches up to 4K examples, dynamic masking, no next-sentence prediction, on 8 16GB V100 GPUs for approximately 90 hours; evaluated on GLUE, IMDb, SQuAD v1.1, CPU inference, and an iPhone 7 Plus on-device proof-of-concept.

Limitations

  • The paper does not state an explicit limitations section.
  • Evident from the text: evaluation is limited to English benchmarks and a small set of downstream tasks, and the on-device study is described as a proof-of-concept; the model could be further reduced with quantization.
  • Evident from the text: the SQuAD second-distillation result is reported inconsistently—the prose says 79.8 F1 and 70.4 EM, while Table 2 lists DistilBERT (D) as 79.1/86.9.

Model-written summary. Check it against the paper before relying on it.

Integrity signals

Registry

DataCite · 10.48550/arxiv.1910.01108

Cites

23 of 23 references verified0 citation errors · 0 not checkable

Statistics

Nothing to recompute.

Verification

Claims the paper makes

15 Own result · 6 Own interpretation · 2 Own method · 2 No basis given · 1 Supported · 1 Partly supported · 1 Sources not found · accuracy not yet measured

Show all 19 claims

  • Probably not a substantial claim. 7
  • It may state two separate claims as one. 2
9 of 9
  • The paper relies on this without support.

    Operating large pre-trained models on-device in real-time is hampered by their growing computational and memory requirements.

    “the growing computational and memory requirements of these models may hamper wide adoption.”
  • The paper relies on this without support.

    Near-zero probabilities in a trained model's output distribution reflect its generalization capabilities.

    “some of these”
  • The paper relies on this without support.

    Inference speed measured on a single GLUE task (STS-B) on CPU is representative of DistilBERT's speed advantage generally.

    “Inference time of a full pass of GLUE task STS-B (sentiment analysis) on CPU with a batch size of 1.”
    “DistilBERT has 40% fewer parameters than BERT and is 60% faster than BERT.”
3 of 3

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

Each contradiction between two passages listed here was confirmed by an independent check.

2 of 2

Citations

Cites

Show all 18 references

Research works printed but not linked: 1

  1. Cristian Bucila, Rich Caruana, and Alexandru Niculescu-Mizil. Model compression. In KDD , 2006.

Cited by

Sorted by citations

Items 4,501–4,600 of 4,692

Metadata

Authors
Sanh, V., Debut, L., Chaumond, J., & Wolf, T.
Venue
arXiv
Publisher
arXiv
Published
2019
ISSN
2331-8422 (ISSN-L)
OpenAlex
W2978017171
Cited by
4,653
References
18
Citations by year
2026: 67 · 2025: 443 · 2024: 801 · 2023: 1136 · 2022: 736 · 2021: 968 · 2020: 486 · 2019: 15
Type
posted content
Language
en
Open access
Locations
  • arXiv (Cornell University) (submittedVersion)
  • arXiv (Cornell University)
  • Transactions on Graph Data and Knowledge (cc-by)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Topic Modeling
  • Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Neural Network Applications
  • Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
Keywords
DistilBERT, knowledge distillation, BERT, pre-training, model compression, on-device computation, language modeling, transfer learning

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

A language model drew the claims from the paper on 2026-10-23, under version 3 of the claim definition (model deepseek-v4.1-flash, prompt extract-v3).

A second model labelled each claim on 2026-10-23 (model jev-1.13-20260917, definition revision gold-guideline-v0.2).

Cited sources are checked with a language model’s help (gpt-6-sol).

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review