研究RAG在医疗领域如何影响大模型的置信度,发现部分模型能自判断检索内容相关性。
Understanding the Impact of Confidence in Retrieval Augmented Generation: A Case Study in the Medical Domain
- 通过分析输出概率,评估大模型对检索文档相关性的自我判断能力。
- 实验表明某些模型在医疗数据上可提升响应置信度,且与准确性正相关。
- 适合关注医疗AI可靠性、模型可信度评估的研究者参考。
检索增强生成(RAG)通过引入外部信息弥补大语言模型(LLM)的知识不足,提升查询回答的准确性,已在多个领域广泛应用,尤其在需要最新信息的高风险场景中备受关注。然而,现有研究对RAG输出置信度的内在机制仍缺乏深入探索。本文聚焦于医疗领域,系统分析RAG对LLM输出置信度的影响,涵盖多种模型配置与数据集。通过将模型预测概率视为其置信度,并计算校准误差、熵、最高概率和准确率等指标进行评估。实验结果表明,部分模型具备自我判断所插入文档是否与答案相关的潜力,且其置信度与回答准确性呈正相关。该研究揭示了基于输出概率评估模型在RAG框架下是否有效处理检索内容的重要性。
原文摘要 · Abstract (English)
Retrieval Augmented Generation (RAG) complements the knowledge of Large Language Models (LLMs) by leveraging external information to enhance response accuracy for queries. This approach is widely applied in several fields by taking its advantage of injecting the most up-to-date information, and researchers are focusing on understanding and improving this aspect to unlock the full potential of RAG in such high-stakes applications. However, despite the potential of RAG to address these needs, the mechanisms behind the confidence levels of its outputs remain underexplored. Our study focuses on the impact of RAG, specifically examining whether RAG improves the confidence of LLM outputs in the medical domain. We conduct this analysis across various configurations and models. We evaluate confidence by treating the model's predicted probability as its output and calculating several evaluation metrics which include calibration error method, entropy, the best probability, and accuracy. Experimental results across multiple datasets confirmed that certain models possess the capability to judge for themselves whether an inserted document relates to the correct answer. These results suggest that evaluating models based on their output probabilities determine whether they function as generators in the RAG framework. Our approach allows us to evaluate whether the models handle retrieved documents.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。