arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.02454cs.CLcs.AIcs.IRcs.LG
收藏

基于图的置信度校准用于大型语言模型

Graph-based Confidence Calibration for Large Language Models

Department of Computer Science Tufts University(塔夫茨大学计算机科学系);Department of Computer Science Virginia Tech(弗吉尼亚理工大学计算机科学系);Department of Computer Science Virginia Tech, University of California, Davis(弗吉尼亚理工大学计算机科学系,加州大学戴维斯分校计算机科学系)

Yukun Li, Sijia Wang, Lifu Huang, Li-Ping Liu

AI总结:

针对大型语言模型置信度估计不可靠的问题,提出基于自一致性构建一致性图并用图神经网络评估响应正确性的方法,在多个基准上校准性能强且泛化良好。

摘要:

可靠的置信度估计对于增强大型语言模型(LLMs)的可信度至关重要,尤其是在高风险场景中。尽管其重要性不言而喻,但准确估计LLM响应的置信度仍然是一个重大挑战。在这项工作中,我们提出使用一个辅助学习模型,基于LLM生成的多个输出的自一致性来评估响应的正确性。我们的方法构建一个一致性图来表示多个响应之间的一致性,并使用图神经网络(GNN)来估计每个响应正确的可能性。实验表明,该方法在各种基准数据集上具有强大的校准性能,并且能够很好地泛化到域外情况。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑