about
Retrieval Meets Long Context Large Language Models (arxiv.org)
2 points by Jimmc414 on Oct 5, 2023 | hide | past | pdf | 1 comment on HN

In plain words: Tests whether feeding a language model only the most relevant passages beats stretching its memory window, and whether combining both works best. A 4K-window model with retrieval matched a 16K fine-tuned one at far lower cost, and retrieval still helped at 32K.

Abstract · Retrieval meets Long Context Large Language Models

Extending the context window of large language models (LLMs) is getting popular recently, while the solution of augmenting LLMs with retrieval has existed for years. The natural questions are: i) Retrieval-augmentation versus long context window, which one is better for downstream tasks? ii) Can both methods be combined to get the best of both worlds? In this work, we answer these questions by studying both solutions using two state-of-the-art pretrained LLMs, i.e., a proprietary 43B GPT and Llama2-70B. Perhaps surprisingly, we find that LLM with 4K context window using simple retrieval-augmentation at generation can achieve comparable performance to finetuned LLM with 16K context window via positional interpolation on long context tasks, while taking much less computation. More importantly, we demonstrate that retrieval can significantly improve the performance of LLMs regardless of their extended context window sizes. Our best model, retrieval-augmented Llama2-70B with 32K context window, outperforms GPT-3.5-turbo-16k and Davinci003 in terms of average score on nine long context tasks including question answering, query-based summarization, and in-context few-shot learning tasks. It also outperforms its non-retrieval Llama2-70B-32k baseline by a margin, while being much faster at generation. Our study provides general insights on the choice of retrieval-augmentation versus long context extension of LLM for practitioners.

Peng Xu, Wei Ping, Xianchao Wu, Lawrence McAfee, Chen Zhu, Zihan Liu, Sandeep Subramanian, Evelina Bakhturina, Mohammad Shoeybi, Bryan Catanzaro
arXiv:2310.03025 · cs.CL, cs.AI, cs.IR, cs.LG · submitted Oct 4, 2023 · updated Jan 23, 2024
abstract · pdf · html · Published at ICLR 2024

add comment on HN

The main findings are:

Retrieval-augmentation can significantly boost the performance of both 4K short context LLMs and 16K/32K long context LLMs.

The 4K context LLMs with simple retrieval-augmentation can perform comparable to 16K long context LLMs, while being more efficient at inference.

After context window extension and retrieval-augmentation, the best model LLaMA2-70B-32k-ret can outperform GPT-3.5-turbo-16k and Davinci003 in terms of average score on a suit of downstream tasks with informative queries.