大型语言模型中的可信不确定性:置信度校准与风险控制拒绝的统一框架
Trusted Uncertainty in Large Language Models: A Unified Framework for Confidence Calibration and Risk-Controlled Refusal
AI总结:
提出UniCR框架,融合多种不确定性证据为校准的正确概率,并通过原则性拒绝满足用户指定的错误预算,无需微调基础模型,在短问答、代码生成和检索增强长问答中提升校准指标并降低风险-覆盖曲线下面积。
摘要:
部署的语言模型必须决定回答什么以及何时不回答。我们提出UniCR,一个统一框架,将异构的不确定性证据(包括序列似然、自一致性离散度、检索兼容性以及工具或验证器反馈)转化为校准的正确概率,然后通过原则性拒绝强制执行用户指定的错误预算。UniCR学习一个轻量级的校准头,采用温度缩放和适当评分,通过黑盒特征支持仅API模型,并使用共形风险控制提供无分布保证。对于长文本生成,我们通过监督从检索证据中得出的原子事实性分数,将置信度与语义保真度对齐,在保持覆盖的同时减少自信幻觉。在短问答、带执行测试的代码生成和检索增强长问答上的实验显示,与熵或logit阈值、后处理校准器和端到端选择性基线相比,校准指标持续改善,风险-覆盖曲线下面积更低,固定风险下覆盖率更高。分析表明,证据矛盾、语义离散度和工具不一致是弃权的主要驱动因素,产生信息丰富的面向用户的拒绝消息。结果是一种可移植的证据融合到校准概率再到风险控制决策的配方,无需微调基础模型即可提高可信度,并在分布偏移下保持有效。
英文摘要:
Deployed language models must decide not only what to answer but also when not to answer. We present UniCR, a unified framework that turns heterogeneous uncertainty evidence including sequence likelihoods, self-consistency dispersion, retrieval compatibility, and tool or verifier feedback into a calibrated probability of correctness and then enforces a user-specified error budget via principled refusal. UniCR learns a lightweight calibration head with temperature scaling and proper scoring, supports API-only models through black-box features, and offers distribution-free guarantees using conformal risk control. For long-form generation, we align confidence with semantic fidelity by supervising on atomic factuality scores derived from retrieved evidence, reducing confident hallucinations while preserving coverage. Experiments on short-form QA, code generation with execution tests, and retrieval-augmented long-form QA show consistent improvements in calibration metrics, lower area under the risk-coverage curve, and higher coverage at fixed risk compared to entropy or logit thresholds, post-hoc calibrators, and end-to-end selective baselines. Analyses reveal that evidence contradiction, semantic dispersion, and tool inconsistency are the dominant drivers of abstention, yielding informative user-facing refusal messages. The result is a portable recipe of evidence fusion to calibrated probability to risk-controlled decision that improves trustworthiness without fine-tuning the base model and remains valid under distribution shift.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。