预测检索文档有用性和答案质量,提升大模型生成效果
Predicting Retrieval Utility and Answer Quality in Retrieval-Augmented Generation
- 用多类特征预测检索文档价值和答案质量
- 融合查询相关与读者感知特征,准确率显著提升
- 适合优化RAG系统性能的研究者与工程师
大型语言模型在检索增强生成(RAG)中的回答质量,主要受检索文档中上下文信息的影响。本文定义了两个预测任务:检索性能预测(RPP),估计检索文档的实用性;生成性能预测(GPP),估计最终答案的正确性与相关性。我们提出,检索文档的主题相关性与其效用正相关,因此可借鉴查询性能预测(QPP)方法用于RPP和GPP。此外,引入基于大模型对上下文的困惑度等读者感知特征,能进一步提升预测精度。同时,不依赖查询的文档质量与可读性特征也提供有效信号。我们在Natural Questions(NQ)数据集上训练线性回归模型,结果表明,整合多类别特征可实现最精确的RAG性能估计。
原文摘要 · Abstract (English)
The quality of answers generated by large language models (LLMs) in retrieval-augmented generation (RAG) is largely influenced by the contextual information contained in the retrieved documents. A key challenge for improving RAG is to predict both the utility of retrieved documents -- quantified as the performance gain from using context over generation without context -- and the quality of the final answers in terms of correctness and relevance. In this paper, we define two prediction tasks within RAG. The first is retrieval performance prediction (RPP), which estimates the utility of retrieved documents. The second is generation performance prediction (GPP), which estimates the final answer quality. We hypothesise that in RAG, the topical relevance of retrieved documents correlates with their utility, suggesting that query performance prediction (QPP) approaches can be adapted for RPP and GPP. Beyond these retriever-centric signals, we argue that reader-centric features, such as the LLM's perplexity of the retrieved context conditioned on the input query, can further enhance prediction accuracy for both RPP and GPP. Finally, we propose that features reflecting query-agnostic document quality and readability can also provide useful signals to the predictions. We train linear regression models with the above categories of predictors for both RPP and GPP. Experiments on the Natural Questions (NQ) dataset show that combining predictors from multiple feature categories yields the most accurate estimates of RAG performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。