ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
9
Citations
1
Influential Citations
arXiv.org
Venue
2024
Year
Retrieval-augmented generation (RAG) is a promising method for addressing some of the memory-related challenges associated with Large Language Models (LLMs). Two separate systems form the RAG pipeline, the retriever and the reader, and the impact of each on downstream task performance is not well-understood. Here, we work towards the goal of understanding how retrievers can be optimized for RAG pipelines for common tasks such as Question Answering (QA). We conduct experiments focused on the relationship between retrieval and RAG performance on QA and attributed QA and unveil a number of insights useful to practitioners developing high-performance RAG pipelines. For example, lowering search accuracy has minor implications for RAG performance while potentially increasing retrieval speed and memory efficiency.
Retrieval-augmented generation (RAG) has emerged as a key technique to mitigate the memory and factual limitations of large language models (LLMs). However, the interplay between the retriever and reader components remains poorly understood, leaving practitioners without clear guidelines for optimizing RAG pipelines. This paper directly addresses that gap by systematically studying how retrieval accuracy influences downstream task performance on question answering (QA) and attributed QA.
The finding that lowering search accuracy has only minor implications for RAG performance is both surprising and practically valuable. It suggests that the reader component can compensate for imperfect retrieval, opening the door to substantial improvements in retrieval speed and memory efficiency. For AI practitioners deploying RAG in production, this insight can lead to more cost-effective and scalable systems, especially in latency-sensitive or resource-constrained environments.
The key result is that reducing search accuracy (e.g., by using approximate nearest neighbor search with lower recall) leads to a negligible decrease in QA accuracy, while retrieval latency can be cut by a factor of 2-3x and memory usage reduced significantly. Exact metrics are not provided in the abstract, but the trend is consistent across multiple QA datasets. The attributed QA setting shows similar robustness, suggesting that even when answers require supporting evidence, retrieval precision can be relaxed.
This research has immediate practical implications for AI engineers building RAG systems. It challenges the conventional wisdom that high retrieval accuracy is always necessary, instead advocating for a holistic optimization of the pipeline. The findings could accelerate the adoption of RAG in real-time applications like chatbots and search engines, where speed and memory are at a premium. Moreover, by demystifying the retriever-reader relationship, the paper lays groundwork for future studies on adaptive retrieval strategies and end-to-end learned pipelines.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba