Understanding the Impact of Confidence in Retrieval Augmented Generation: A Case Study in the Medical Domain

Shintaro Ozaki, Yuta Kato, Siyuan Feng, Masayo Tomita, Kazuki Hayashi +6 more
12/29/2024
cs.CL

No official code found. Best implementation: naist-nlp/CC_RAG (0 stars, MIT, last commit 2025-06-07).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

naist-nlp/CC_RAGCommunity100%
MIT

The repository of our paper about confidence calibration on RAG.

00CSS, HTMLNov 26, 20241 years agoMIT
confidence-calibrationlarge-language-modelsllmragretrieval-augmented-generation

Similar Papers

Cite this paper

@article{ozaki2024understanding,
  title  = {Understanding the Impact of Confidence in Retrieval Augmented Generation: A Case Study in the Medical Domain},
  author = {Shintaro Ozaki and Yuta Kato and Siyuan Feng and Masayo Tomita and Kazuki Hayashi and Wataru Hashimoto and Ryoma Obara and Masafumi Oyamada and Katsuhiko Hayashi and Hidetaka Kamigaito and Taro Watanabe},
  year   = {2024},
  eprint = {2412.20309},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2412.20309v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2412.20309.svg)](https://codeofpaper.com/paper/2412.20309)

Discussion