通过短文本投票提升大模型生成准确率,效率更高
Efficient Test-Time Retrieval Augmented Generation
- 只生成部分响应就计算共识,降低计算开销
- 在三类任务中显著提升准确率,效果优于传统RAG
- 无需训练,适合快速部署到现有大模型系统
尽管大语言模型展现出强大能力,但其依赖参数化知识常导致错误。检索增强生成(RAG)通过引入外部知识缓解此问题,但可能引入无关文档,造成回答不准确。现有方法虽能过滤错误答案,却缺乏外部知识支持,且成本较高,难以平衡开销与性能。为此,我们提出一种高效测试时检索增强生成框架ET2RAG,实现高性能与高效率的统一。该方法为无训练设计,先检索最相关文档,再通过控制生成长度,高效生成多样化候选响应;随后计算候选响应间的相似性,并采用多数投票机制选出最优输出。关键发现是:仅需部分生成即可捕捉共识所需关键信息,因此无需完整生成即可完成投票。由此可在保留性能的同时,通过调节生成长度和检索文档数量,有效平衡计算成本与效果。实验表明,ET2RAG在开放域问答、菜谱生成和图像描述三项任务中均显著提升性能。
原文摘要 · Abstract (English)
Although Large Language Models (LLMs) demonstrate significant capabilities, their reliance on parametric knowledge often leads to inaccuracies. Retrieval Augmented Generation (RAG) mitigates this by incorporating external knowledge, but these methods may introduce irrelevant retrieved documents, leading to inaccurate responses. While the integration methods filter out incorrect answers from multiple responses, but lack external knowledge like RAG methods, and their high costs require balancing overhead with performance gains. To address these issues, we propose an Efficient Test-Time Retrieval-Augmented Generation Framework named ET2RAG to improve the performance of LLMs while maintaining efficiency. Specifically, ET2RAG is a training-free method, that first retrieves the most relevant documents and augments the LLMs to efficiently generate diverse candidate responses by managing response length. Then we compute the similarity of candidate responses and employ a majority voting mechanism to select the most suitable response as the final output. In particular, we discover that partial generation is sufficient to capture the key information necessary for consensus calculation, allowing us to effectively perform majority voting without the need for fully generated responses. Thus, we can reach a balance between computational cost and performance by managing the response length for the number of retrieved documents for majority voting. Experimental results demonstrate that ET2RAG significantly enhances performance across three tasks, including open-domain question answering, recipe generation and image captioning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。