arXiv:2410.05407cs.LGcs.AI2024-10被引 6

通过筛选数据提升模型置信度校准效果,让可信预测更可靠。

Improving Predictor Reliability with Selective Recalibration

  • 用选择模型剔除部分数据,让校准器专注易拟合区域
  • 在医疗影像与零样本分类任务中校准误差显著降低
  • 适合对置信度可靠性要求高的应用场景

可靠的深度学习系统应能准确表达预测的置信度,这一特性称为校准。现有后处理校准方法(如温度缩放)通常在少量数据上训练,仅在模型输出空间操作,参数少(通常仅一个或少数几个),难以拟合复杂的分布。为此,我们提出「选择性校准」:通过一个选择模型,主动剔除用户指定比例的数据,使校准器聚焦于输入空间中可被简单函数良好捕捉的区域。我们提供了理论分析支持该算法,并在具有挑战性的医学影像和零样本分类任务上进行了全面实验。结果表明,选择性校准在多种基线方法上均显著降低校准误差。

原文摘要 · Abstract (English)

A reliable deep learning system should be able to accurately express its confidence with respect to its predictions, a quality known as calibration. One of the most effective ways to produce reliable confidence estimates with a pre-trained model is by applying a post-hoc recalibration method. Popular recalibration methods like temperature scaling are typically fit on a small amount of data and work in the model's output space, as opposed to the more expressive feature embedding space, and thus usually have only one or a handful of parameters. However, the target distribution to which they are applied is often complex and difficult to fit well with such a function. To this end we propose \textit{selective recalibration}, where a selection model learns to reject some user-chosen proportion of the data in order to allow the recalibrator to focus on regions of the input space that can be well-captured by such a model. We provide theoretical analysis to motivate our algorithm, and test our method through comprehensive experiments on difficult medical imaging and zero-shot classification tasks. Our results show that selective recalibration consistently leads to significantly lower calibration error than a wide range of selection and recalibration baselines.

模型校准置信度医疗图像零样本

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