Uncertainty Estimation of Large Language Models in Medical Question Answering
This addresses the critical need for reliable uncertainty estimation in deploying LLMs for medical applications, where factual errors can have serious consequences, though it is an incremental improvement over existing methods.
The paper tackles the problem of unreliable uncertainty estimation (UE) for large language models (LLMs) in medical question answering, which risks hallucinations, and proposes a Two-phase Verification method that achieves the best overall accuracy and stability across datasets and model sizes, with performance scaling as model size increases.
Large Language Models (LLMs) show promise for natural language generation in healthcare, but risk hallucinating factually incorrect information. Deploying LLMs for medical question answering necessitates reliable uncertainty estimation (UE) methods to detect hallucinations. In this work, we benchmark popular UE methods with different model sizes on medical question-answering datasets. Our results show that current approaches generally perform poorly in this domain, highlighting the challenge of UE for medical applications. We also observe that larger models tend to yield better results, suggesting a correlation between model size and the reliability of UE. To address these challenges, we propose Two-phase Verification, a probability-free Uncertainty Estimation approach. First, an LLM generates a step-by-step explanation alongside its initial answer, followed by formulating verification questions to check the factual claims in the explanation. The model then answers these questions twice: first independently, and then referencing the explanation. Inconsistencies between the two sets of answers measure the uncertainty in the original response. We evaluate our approach on three biomedical question-answering datasets using Llama 2 Chat models and compare it against the benchmarked baseline methods. The results show that our Two-phase Verification method achieves the best overall accuracy and stability across various datasets and model sizes, and its performance scales as the model size increases.