arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.14403cs.CLcs.AI
收藏

Adaptive-RAG:通过问题复杂度学习自适应检索增强大语言模型

Adaptive-RAG: Learning to Adapt Retrieval-Augmented Large Language Models through Question Complexity

School of Computing(计算机学院);Graduate School of AI(人工智能研究生院)

Soyeong Jeong, Jinheon Baek, Sukmin Cho, Sung Ju Hwang, Jong C. Park

AI总结:

本文提出Adaptive-RAG框架,通过训练小型分类器预测查询复杂度,在无检索、单步和迭代检索增强LLM之间动态选择最优策略,从而在开放域QA任务中同时提升效率与准确率。

摘要:

检索增强大语言模型(LLMs)将外部知识库中的非参数化知识融入LLMs,已成为在问答(QA)等多项任务中提升回答准确率的一种有前景的方法。然而,尽管已有多种方法处理不同复杂度的查询,它们要么以不必要的计算开销处理简单查询,要么无法充分应对复杂的多步查询;而且,并非所有用户请求都只属于简单或复杂中的某一类。在本工作中,我们提出了一种新颖的自适应QA框架,该框架能够根据查询复杂度,动态地为(检索增强的)LLMs选择从最简单到最复杂的合适策略。此外,这一选择过程由一个分类器实现,该分类器是一个较小的语言模型(LM),经过训练以预测传入查询的复杂度级别,其标签是自动收集的,来源于模型的实际预测结果以及数据集中固有的归纳偏置。该方法提供了一种均衡的策略,能够在迭代式检索增强LLMs、单步检索增强LLMs以及无检索方法之间无缝适配,以应对不同范围的查询复杂度。我们在一组覆盖多种查询复杂度的开放域QA数据集上验证了我们的模型,并表明与包括自适应检索方法在内的相关基线相比,我们的方法提升了QA系统的整体效率和准确率。代码可在以下地址获取:https://github.com/starsuzi/Adaptive-RAG。

英文摘要:

Retrieval-Augmented Large Language Models (LLMs), which incorporate the non-parametric knowledge from external knowledge bases into LLMs, have emerged as a promising approach to enhancing response accuracy in several tasks, such as Question-Answering (QA). However, even though there are various approaches dealing with queries of different complexities, they either handle simple queries with unnecessary computational overhead or fail to adequately address complex multi-step queries; yet, not all user requests fall into only one of the simple or complex categories. In this work, we propose a novel adaptive QA framework, that can dynamically select the most suitable strategy for (retrieval-augmented) LLMs from the simplest to the most sophisticated ones based on the query complexity. Also, this selection process is operationalized with a classifier, which is a smaller LM trained to predict the complexity level of incoming queries with automatically collected labels, obtained from actual predicted outcomes of models and inherent inductive biases in datasets. This approach offers a balanced strategy, seamlessly adapting between the iterative and single-step retrieval-augmented LLMs, as well as the no-retrieval methods, in response to a range of query complexities. We validate our model on a set of open-domain QA datasets, covering multiple query complexities, and show that ours enhances the overall efficiency and accuracy of QA systems, compared to relevant baselines including the adaptive retrieval approaches. Code is available at: https://github.com/starsuzi/Adaptive-RAG.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑