PreprintCited by 382Topic ModelingNatural Language Processing TechniquesInformation Retrieval and Search Behavior

Improving language models by retrieving from trillions of tokens

Open access
Borgeaud, Sebastian, Mensch, Arthur, Hoffmann, Jordan, Cai, Trevor, Rutherford, Eliza, Millican, Katie, Driessche, George van den, Lespiau, Jean-Baptiste +20 more·2021·arXiv

Abstract

We enhance auto-regressive language models by conditioning on document chunks retrieved from a large corpus, based on local similarity with preceding tokens. With a $2$ trillion token database, our Retrieval-Enhanced Transformer (RETRO) obtains comparable performance to GPT-3 and Jurassic-1 on the Pile, despite using 25$\times$ fewer parameters. After fine-tuning, RETRO performance translates to downstream knowledge-intensive tasks such as question answering. RETRO combines a frozen Bert retriever, a differentiable encoder and a chunked cross-attention mechanism to predict tokens based on an order of magnitude more data than what is typically consumed during training. We typically train RETRO from scratch, yet can also rapidly RETROfit pre-trained transformers with retrieval and still achieve good performance. Our work opens up new avenues for improving language models through explicit memory at unprecedented scale.

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Integrity signals

Registry

DataCite · 10.48550/arxiv.2112.04426

Cites

1 of 1 references verified0 citation errors · 0 not checkable

Statistics

Claims the paper makes

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

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?

  • GMAI can retrieve relevant context from existing databases, in the form of articles, images or entire previous cases.

    “Furthermore, building on recent retrieval-based approaches, GMAI can retrieve relevant context from existing databases, in the form of articles, images or entire previous cases 19,20 .”

    From Foundation models for generalist medical artificial intelligence (2023) · cites this paper as [19]

    The Introduction (p. 1) describes Retro retrieving similar text from a large database, and the training-dataset section (p. 3) includes Wikipedia articles among its text sources. The paper does not establish that GMAI can retrieve images or entire previous medical cases; it supports the retrieval-based approach, but not those extensions. (LLM, conf=0.88)

    Experimental — accuracy not yet measured
  • Longer contexts enable moving forward to smart context composition techniques, including fill-in-the-middle and retrieval-augmented generation methods.

    “Longer contexts enable us to move forward to”

    From Context Composing for Full Line Code Completion (2024) · cites this paper as [2]

    The Introduction (p. 1) and Method (p. 2) describe a retrieval-enhanced language model that incorporates retrieved text chunks, supporting the retrieval part of the claim. The paper does not say that longer contexts enable such methods or discuss fill-in-the-middle as a context-composition technique. (LLM, conf=0.85)

    Experimental — accuracy not yet measured
  • Models augmented by retrieving information from large databases outperform larger models trained on larger datasets, effectively leveraging available information and providing added benefits such as interpretability.

    “Along these lines, recent studies demonstrated that models augmented by retrieving information from large databases outperform larger models trained on larger datasets, effectively leveraging available information and also providing added benefits such as interpretability 124,125”

    From Multimodal biomedical AI (2022) · cites this paper as [125]

    The Conclusion (p. 15) says Retro outperforms previous models trained on large-scale datasets, and §3.2 (p. 10) says retrieved neighbours make its outputs more interpretable. However, the performance comparison is dataset-dependent: §4.1 (p. 11) reports that Retro outperforms the much larger Jurassic-1 on a majority of Pile test sets, not that it outperforms larger models generally. (LLM, conf=0.86)

    Experimental — accuracy not yet measured
  • REALM integrates retrieval into language-model pretraining, while RETRO conditions language models on retrieved chunks from very large corpora.

    “REALM integrates retrieval into language-model pretraining, while RETRO conditions language models on retrieved chunks from very large corpora [6, 4].”
  • RAG, REALM, and RETRO retrieve from large corpora.

    “RAG [7], REALM [10], and RETRO [11] retrieve from large corpora.”

Self-check

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

The self-check needs the paper’s full text.

Cites

This paper’s reference list could not be split into single entries.

Flagged first, then by citations

Cited by

Sorted by citations

Authors
Borgeaud, S., Mensch, A., Hoffmann, J., Cai, T., Rutherford, E., Millican, K., Driessche, G. V. D., Lespiau, J.-B., Damoc, B., Clark, A., Casas, D. D. L., Guy, A., Menick, J., Ring, R., Hennigan, T., Huang, S., Maggiore, L., Jones, C., Cassirer, A., Brock, A., Paganini, M., Irving, G., Vinyals, O., Osindero, S., Simonyan, K., Rae, J. W., Elsen, E., & Sifre, L.
Venue
arXiv
Publisher
arXiv
Published
2021
ISSN
2331-8422 (ISSN-L)
OpenAlex
W4226082499
Cited by
382
References
1
Citations by year
2026: 7 · 2025: 86 · 2024: 113 · 2023: 141 · 2022: 35
Type
posted content
Language
en
Open access
Locations
  • arXiv (Cornell University) (submittedVersion)
  • arXiv (Cornell University) (cc-by)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Topic Modeling
  • Physical Sciences › Computer Science › Artificial Intelligence › Natural Language Processing Techniques
  • Physical Sciences › Computer Science › Information Systems › Information Retrieval and Search Behavior
Keywords
retrieval-augmented language models, retro style, cross-attention, declarative memory, parameter-efficient fine-tuning, retrofitting, question answering

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.

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