Andy Brock

Improving language models by retrieving from trillions of tokens

2021 · arXiv (Cornell University)

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 …