arXiv · 2411.02454
Graph-based Confidence Calibration for Large Language Models
Abstract
Reliable confidence estimation is essential for enhancing the trustworthiness of large language models (LLMs), especially in high-stakes scenarios. Despite its importance, accurately estimating confidence in LLM responses remains a significant challenge. In this work, we propose using an auxiliary learning model to assess response correctness based on the self-consistency of multiple outputs generated by the LLM. Our method builds a consistency graph to represent the agreement among multiple responses and uses a graph neural network (GNN) to estimate the likelihood that each response is correct. Experiments demonstrate that this method has strong calibration performance on various benchmark datasets and generalizes well to out-of-domain cases.
Explore related subjects
Keep this discovery
Yukun Li, Sijia Wang, Lifu Huang, Li-Ping Liu. 2024-11-03. Graph-based Confidence Calibration for Large Language Models. https://arxiv.org/abs/2411.02454
Cite the original work for its findings. Save a collection to share your selection of sources.