arXiv:2605.21552cs.LGstat.ML2026-05中稿 · ICML

提出新方法提升数据分布变化时分类模型的置信度准确性

Expectation Consistency Loss: Rethink Confidence Calibration under Covariate Shift

论文配图:Expectation Consistency Loss: Rethink Confidence Calibration under Covariate Shift
图 1 · 摘自论文原文
  • 基于期望一致性条件设计无监督域适应损失
  • 在模拟与真实数据集上显著改善置信度校准效果
  • 适用于各类校准场景,理论可训练且样本效率高

分类模型的置信度校准在安全关键决策中至关重要,传统方法假设训练与测试数据同分布,难以应对协变量偏移。现有方法在协变量偏移下常面临类别或标准校准难题,且依赖不稳定的权重估计。本文重新思考该问题,推导出校准的充要条件——期望一致性条件,表明协变量偏移不一定导致置信度失准,且弱于全局分布对齐要求。基于此,提出期望一致性损失(ECL),一种无需标签的域自适应校准损失,兼容标准、类别级及顶标签校准。证明计算ECL的样本复杂度与预期校准误差(ECE)相当,并给出理论支持的批量可训练方案。在模拟与真实协变量偏移数据集上验证了方法有效性。

原文摘要 · Abstract (English)

Confidence calibration for classification models is vital in safety-critical decision-making scenarios and has received extensive attention. General confidence calibration methods assume training and test data are independent and identically distributed, limiting their effectiveness under covariate shifts. Previous calibration methods under covariate shift struggle with class-wise or canonical calibrations and often rely on unstable importance weighting when density ratios are large or unbounded. Given the above limitations, this paper rethinks confidence calibration under covariate shifts. First, we derive a necessary and sufficient condition for confidence calibration under covariate shifts, named Expectation consistency condition, which reveals covariate shifts do not necessarily lead to uncalibrated confidence and provides a weaker condition for confidence calibration than global covariate distribution alignment. Then, utilizing Expectation consistency condition, this paper proposes an unsupervised domain adaptation loss to calibrate confidence of the target domain, named Expectation consistency loss (ECL), which is compatible with canonical calibration, class-wise calibration, and top-label calibration. Third, we prove that computing ECL loss has the same sample complexity as Expected Calibration Error (ECE) and provide a theoretically grounded mini-batch trainable scheme for ECL loss. Finally, we validate the effectiveness of our method on both simulated and real-world covariate shift datasets.

置信度校准协变量偏移无监督学习深度学习可靠性

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