Improving language models by retrieving from trillions of tokens

AI-generated keywords: Retrieval-Enhanced Transformer

AI-generated Key Points

  • Language models have made remarkable progress in natural language processing tasks but still have limitations.
  • The proposed Retrieval-Enhanced Transformer (RETRO) enhances auto-regressive language models by conditioning on document chunks retrieved from a large corpus based on local similarity with preceding tokens.
  • RETRO achieves comparable performance to GPT-3 and Jurassic-1 on the Pile dataset despite using 25 times fewer parameters.
  • 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.
  • RETRO's performance translates to downstream knowledge-intensive tasks such as question answering after fine-tuning.
  • The model can be trained from scratch or rapidly retrofit pre-trained transformers with retrieval and still achieve good performance.
  • Improvements do not diminish as they scale the models or increase the number of retrieved chunks.
  • Jurassic-1 outperforms their baseline on all datasets except for books, likely due to the inclusion of books in their training data.
  • This work opens up new avenues for improving language models through explicit memory at unprecedented scale by retrieving information from trillions of tokens.
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Sebastian Borgeaud, Arthur Mensch, Jordan Hoffmann, Trevor Cai, Eliza Rutherford, Katie Millican, George van den Driessche, Jean-Baptiste Lespiau, Bogdan Damoc, Aidan Clark, Diego de Las Casas, Aurelia Guy, Jacob Menick, Roman Ring, Tom Hennigan, Saffron Huang, Loren Maggiore, Chris Jones, Albin Cassirer, Andy Brock, Michela Paganini, Geoffrey Irving, Oriol Vinyals, Simon Osindero, Karen Simonyan, Jack W. Rae, Erich Elsen, Laurent Sifre

License: CC BY 4.0

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.

Submitted to arXiv on 08 Dec. 2021

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2112.04426v1

In recent years, language models have shown remarkable progress in various natural language processing tasks. However, these models still suffer from limitations such as lack of explicit memory and inability to handle long-range dependencies. To address these issues, a team of researchers from DeepMind has proposed a Retrieval-Enhanced Transformer (RETRO) that enhances auto-regressive language models by conditioning on document chunks retrieved from a large corpus based on local similarity with preceding tokens. With a database of 2 trillion tokens, RETRO achieves comparable performance to GPT-3 and Jurassic-1 on the Pile dataset despite using 25 times fewer parameters. The model 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. After fine-tuning, RETRO's performance translates to downstream knowledge-intensive tasks such as question answering. The researchers train RETRO from scratch but can also rapidly retrofit pre-trained transformers with retrieval and still achieve good performance. They evaluate the model's performance by scaling both the model size and the retrieval database at evaluation. They observe that improvements do not diminish as they scale the models or increase the number of retrieved chunks. The team evaluates their 7 billion parameter RETRO model on the Pile test sets and compares it against Jurassic-1 and Gopher models. They find that Jurassic-1 outperforms their baseline on all datasets except for books, likely due to the inclusion of books in their training data. On Curation Corpus, they concatenate the article, "TL;DR:" string, and summary but only evaluate bits-per-byte (bpb) on the summary. Overall, this work opens up new avenues for improving language models through explicit memory at unprecedented scale by retrieving information from trillions of tokens.
Created on 15 Jun. 2023

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.