发现预训练大模型暗中能做无监督置信度校准,提升模型可靠性。
Your Pre-trained LLM is Secretly an Unsupervised Confidence Calibrator
- 基于一致预测样本优化温度参数,避开分歧样本干扰。
- 在多个基准上使置信度校准误差降低最多15.08%。
- 无需标注数据,适合部署后提升各类大模型可信度。
大语言模型的后训练对适配人类偏好和下游任务至关重要。尽管预训练语言模型(PLM)通常具备良好的置信度校准能力,但后训练语言模型(PoLM)常出现过度自信,对正确与错误输出均赋予高置信度,影响关键应用的可靠性。其主要瓶颈在于下游任务缺乏标注数据。为此,我们提出无监督方法DACA(Disagreement-Aware Confidence Alignment),通过温度缩放优化后处理置信度校准参数。该方法受PLM与PoLM在分歧样本上预测不一致导致置信度低估的启发,理论上会引发过大的温度τ值,造成置信度偏低。DACA通过仅使用一致预测样本进行校准,有效解耦分歧样本的影响。实验表明,该方法显著提升开源及API型大模型(如GPT-4o)的平均ECE,最高改善达15.08%。
原文摘要 · Abstract (English)
Post-training of large language models is essential for adapting pre-trained language models (PLMs) to align with human preferences and downstream tasks. While PLMs typically exhibit well-calibrated confidence, post-trained language models (PoLMs) often suffer from over-confidence, assigning high confidence to both correct and incorrect outputs, which can undermine reliability in critical applications. A major obstacle in calibrating PoLMs is the scarcity of labeled data for individual downstream tasks. To address this, we propose Disagreement-Aware Confidence Alignment (DACA), a novel unsupervised method to optimize the parameters (e.g., temperature $τ$) in post-hoc confidence calibration. Our method is motivated by the under-confidence issue caused by prediction disagreement between the PLM and PoLM while aligning their confidence via temperature scaling. Theoretically, the PLM's confidence underestimates PoLM's prediction accuracy on disagreement examples, causing a larger $τ$ and producing under-confident predictions. DACA mitigates this by selectively using only agreement examples for calibration, effectively decoupling the influence of disagreement. In this manner, our method avoids an overly large $τ$ in temperature scaling caused by disagreement examples, improving calibration performance. Extensive experiments demonstrate the effectiveness of our method, improving the average ECE of open-sourced and API-based LLMs (e.g. GPT-4o) by up to 15.08$\%$ on common benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。