arXiv:2412.07255cs.CLcs.AI2024-12被引 1

提出新方法提升大模型生成结果的可信度评估

Label-Confidence-Aware Uncertainty Estimation in Natural Language Generation

  • 基于KL散度融合采样与标签源信息,捕捉生成不确定性
  • 实验证明不同标签来源影响分类结果,新方法更稳定可靠
  • 适合关注大模型安全与输出可信度的研究者

大语言模型在生成任务中表现强大,但易产生幻觉内容。不确定性量化(UQ)对于保障AI系统安全与鲁棒性至关重要。现有方法多关注采样条件下输出熵与标签的关系,但忽视了贪婪解码产生的标签源不确定性,导致评估偏差。本文分析贪婪解码引入的偏差,提出标签置信度感知(LCA)不确定性估计方法,通过Kullback-Leibler(KL)散度连接采样结果与标签源,增强评估可靠性。在多个主流LLM和NLP数据集上的实证表明,不同标签源确实影响分类性能,所提方法能有效捕捉采样与标签源差异,实现更优的不确定性估计。

原文摘要 · Abstract (English)

Large Language Models (LLMs) display formidable capabilities in generative tasks but also pose potential risks due to their tendency to generate hallucinatory responses. Uncertainty Quantification (UQ), the evaluation of model output reliability, is crucial for ensuring the safety and robustness of AI systems. Recent studies have concentrated on model uncertainty by analyzing the relationship between output entropy under various sampling conditions and the corresponding labels. However, these methods primarily focus on measuring model entropy with precision to capture response characteristics, often neglecting the uncertainties associated with greedy decoding results-the sources of model labels, which can lead to biased classification outcomes. In this paper, we explore the biases introduced by greedy decoding and propose a label-confidence-aware (LCA) uncertainty estimation based on Kullback-Leibler (KL) divergence bridging between samples and label source, thus enhancing the reliability and stability of uncertainty assessments. Our empirical evaluations across a range of popular LLMs and NLP datasets reveal that different label sources can indeed affect classification, and that our approach can effectively capture differences in sampling results and label sources, demonstrating more effective uncertainty estimation.

大模型不确定性生成评估

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。