探究检索增强模型是否知道自己的知识边界,发现其常过度拒绝回答。
Do Retrieval Augmented Language Models Know When They Don't Know?
- 通过分析检索结果与内部知识状态,测试模型拒答能力的校准性。
- 即使能正确回答,模型在无关检索下仍频繁拒绝,存在明显过拒现象。
- 提出结合不确定度与拒答机制,提升回答质量与平衡性,适合安全敏感场景使用。
现有大语言模型(LLMs)常生成看似合理但事实错误的回答,即幻觉问题。主流缓解方法包括检索增强语言模型(RALMs)和拒答后训练。然而,当前研究多关注单一方法效果,忽视对RALMs拒答能力的评估。理想情况下,若RALMs知道自己不知道,应主动拒绝回答。本研究提出核心问题:RALMs是否知道何时不知道?首先,考察模型在不同内外部知识状态下的校准性,发现当所有检索文档无关时,即便可正确作答,模型仍倾向于拒绝。其次,针对明显的过拒行为,研究拒答能力与校准质量的关系,结果显示通过上下文微调可缓解过拒,但拒答改善不必然带来更好校准或更高准确率。最后,探索将拒答感知的RALMs与基于不确定性的答案回避结合,设计一种简单有效的拒答机制,使拒绝与正确回答更平衡,从而提升整体回答质量。研究揭示了影响RALM行为的多重因素,强调不确定性估计仍是亟待深入的开放问题。
原文摘要 · Abstract (English)
Existing large language models (LLMs) occasionally generate plausible yet factually incorrect responses, known as hallucinations. Two main approaches have been proposed to mitigate hallucinations: retrieval-augmented language models (RALMs) and refusal post-training. However, current research predominantly focuses on their individual effectiveness while overlooking the evaluation of the refusal capability of RALMs. Ideally, if RALMs know when they do not know, they should refuse to answer.In this study, we ask the fundamental question: Do RALMs know when they don't know? Specifically, we investigate three questions. First, are RALMs well calibrated with respect to different internal and external knowledge states? We examine the influence of various factors. Contrary to expectations, when all retrieved documents are irrelevant, RALMs still tend to refuse questions they could have answered correctly. Next, given the model's pronounced \textbf{over-refusal} behavior, we raise a second question: How does a RALM's refusal ability align with its calibration quality? Our results show that the over-refusal problem can be mitigated through in-context fine-tuning. However, we observe that improved refusal behavior does not necessarily imply better calibration or higher overall accuracy. Finally, we ask: Can we combine refusal-aware RALMs with uncertainty-based answer abstention to mitigate over-refusal? We develop a simple yet effective refusal mechanism for refusal-post-trained RALMs that improves their overall answer quality by balancing refusal and correct answers. Our study provides a more comprehensive understanding of the factors influencing RALM behavior. Meanwhile, we emphasize that uncertainty estimation for RALMs remains an open problem deserving deeper investigation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。