模型能力提升却更盲目自信,新方法让模型既强又靠谱。
The Illusion of Certainty: Decoupling Capability and Calibration in On-Policy Distillation

- 用学生自身推理结果替代教师信心,避免信息错配
- 在多个模型和任务上实现准确率与可信度双优
- 适合追求可靠推理的落地应用,如医疗、金融
在线策略蒸馏(OPD)是后训练语言模型的重要范式。我们发现一种普遍存在的校准失效规律:尽管OPD能有效提升任务准确率,却使模型陷入严重过度自信。根源在于信息错配——教师监督依赖训练时的特权上下文,而部署模型只能使用运行时信息报告置信度。理论上证明,教师条件下的成功通常不适用于部署时的置信度判断,且有利的特权上下文会导致熵坍缩和系统性乐观偏差。为此,我们提出校准感知的OPD框架CaOPD:通过模型自举回滚估计经验置信度,替换为学生基础的目标置信度,并沿用相同自蒸馏流程进行知识迁移。跨多种模型与领域实验表明,CaOPD在保持竞争力的同时实现了帕累托最优校准,对分布外数据和持续学习具有鲁棒性。研究揭示:能力提升不等于可信度提升,置信度应作为后训练的核心目标。代码已开源。
原文摘要 · Abstract (English)
On-policy distillation (OPD) is an increasingly important paradigm for post-training language models. However, we identify a pervasive Scaling Law of Miscalibration: while OPD effectively improves task accuracy, it systematically traps models in severe overconfidence. We trace this failure to an information mismatch: teacher supervision is formed under privileged context available during training, whereas the deployed model must report confidence using only deployment-time information. We formalize this perspective theoretically, showing that teacher-conditioned success is generally not a valid target for deployment-time confidence and that helpful privileged context induces entropy collapse and a systematic optimism bias. To address this, we propose a calibration-aware OPD framework, CaOPD, that estimates empirical confidence from model rollouts, replaces self-reported confidence with this student-grounded target, and distills the revised response through the same self-distillation pipeline. Experiments across various models and domains show that CaOPD achieves Pareto-optimal calibration while maintaining competitive capability, generalizing robustly under out-of-distribution and continual learning. Our findings highlight that capability distillation does not imply calibrated confidence, and that confidence should be treated as an essential objective in post-training. Code: https://github.com/SalesforceAIResearch/CaOPD
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。