arXiv:2504.13548cs.CVcs.AI2025-04ICML被引 3

用混合语义生成带可信度标签的合成数据,提升模型校准效果

Beyond One-Hot Labels: Semantic Mixing for Model Calibration

  • 通过扩散模型生成混合类别的图像并标注不同置信度
  • 在ImageNet上校准误差降至0.038,优于现有方法
  • 适合需要高置信度输出的医疗、自动驾驶场景

模型校准旨在使模型的置信度分数准确反映预测正确的实际概率。然而,现有校准方法依赖于一热标签数据集,隐含假设所有标注都完全确定,这类数据虽适合分类任务,却缺乏不确定性信息,难以支持校准。为此,需构建包含数值丰富真实置信度的标注数据集。但因不确定视觉样本稀缺,真实数据难获取。本文提出校准感知数据增强,构建多样化样本及其真实不确定性标注的合成数据集。具体提出校准感知语义混合(CSM)框架,利用扩散模型生成具有混合类别特征的训练样本,并通过反向过程注入精确置信度标注。为解决标注置信度与混合比例之间的偏差,引入校准重标注机制,并探索适配新数据表示范式的损失函数。实验表明,CSM在多个基准上校准性能显著优于当前最优方法,在ImageNet上校准误差低至0.038。

原文摘要 · Abstract (English)

Model calibration seeks to ensure that models produce confidence scores that accurately reflect the true likelihood of their predictions being correct. However, existing calibration approaches are fundamentally tied to datasets of one-hot labels implicitly assuming full certainty in all the annotations. Such datasets are effective for classification but provides insufficient knowledge of uncertainty for model calibration, necessitating the curation of datasets with numerically rich ground-truth confidence values. However, due to the scarcity of uncertain visual examples, such samples are not easily available as real datasets. In this paper, we introduce calibration-aware data augmentation to create synthetic datasets of diverse samples and their ground-truth uncertainty. Specifically, we present \textbf{Calibration-aware Semantic Mixing (CSM)}, a novel framework that generates training samples with mixed class characteristics and annotates them with distinct confidence scores via diffusion models. Based on this framework, we propose calibrated reannotation to tackle the misalignment between the annotated confidence score and the mixing ratio during the diffusion reverse process. Besides, we explore the loss functions that better fit the new data representation paradigm. Experimental results demonstrate that CSM achieves superior calibration compared to the state-of-the-art calibration approaches. Our code is \href{https://github.com/E-Galois/CSM}{available here}.

模型校准扩散模型不确定性建模数据增强

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