解决大模型对齐后校准能力下降问题,提升预测可信度。
Restoring Calibration for Aligned Large Language Models: A Calibration-Aware Fine-Tuning Approach
- 通过校准感知微调,缓解对齐导致的过度自信
- 在可校准范围内实现低ECE(<0.03)且不损失性能
- 提出基于EM算法的正则化方法,适用于难校准模型
大型语言模型(LLM)的成功依赖于偏好对齐技术,但该过程会带来校准能力下降:预训练模型通常校准良好,而对齐后模型常表现出过度自信。本文揭示,对齐中的偏好坍缩问题会泛化至校准场景,导致模型失准。研究发现,引入领域知识微调可缓解过度自信。进一步按期望校准误差(ECE)划分模型为可校准与不可校准两类。在可校准情形下,提出校准感知微调方法,在保持性能前提下实现低校准误差(平均ECE < 0.03)。对于进入不可校准区的模型,设计基于EM算法的ECE正则化项加入损失函数,有效维持低校准误差。大量实验验证了方法的有效性。
原文摘要 · Abstract (English)
One of the key technologies for the success of Large Language Models (LLMs) is preference alignment. However, a notable side effect of preference alignment is poor calibration: while the pre-trained models are typically well-calibrated, LLMs tend to become poorly calibrated after alignment with human preferences. In this paper, we investigate why preference alignment affects calibration and how to address this issue. For the first question, we observe that the preference collapse issue in alignment undesirably generalizes to the calibration scenario, causing LLMs to exhibit overconfidence and poor calibration. To address this, we demonstrate the importance of fine-tuning with domain-specific knowledge to alleviate the overconfidence issue. To further analyze whether this affects the model's performance, we categorize models into two regimes: calibratable and non-calibratable, defined by bounds of Expected Calibration Error (ECE). In the calibratable regime, we propose a calibration-aware fine-tuning approach to achieve proper calibration without compromising LLMs' performance. However, as models are further fine-tuned for better performance, they enter the non-calibratable regime. For this case, we develop an EM-algorithm-based ECE regularization for the fine-tuning loss to maintain low calibration error. Extensive experiments validate the effectiveness of the proposed methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。