摘要
arXiv:2405.15047v2 宣告类型: replace-cross
摘要:本文提出了一种称为可信度包装器的新颖方法,将其应用于贝叶斯神经网络(BNNs)和深度集成(DEs)的模型平均表示形式,以改进分类任务中的不确定性估计。给定来自BNNs或DEs的一系列有限单预测分布,所提出的可信度包装器方法将为每个类别提取一个上界和一个下界概率,这承认了由于可用于的分布有限而导致的不确定性。这样的类概率区间可以映射到一个概率凸集(一个可信度集),从中可以使用一种称为交概率转换的方法获得一个唯一预测。在本文中,我们在多个离分布(OOD)检测基准上进行了广泛的实验,涵盖了各种数据集对(CIFAR10/100 vs SVHN/Tiny-ImageNet, CIFAR10 vs CIFAR10-C, CIFAR100 vs CIFAR100-C 和 ImageNet vs ImageNet-O)和不同的网络架构(如 VGG16, ResNet-18/50, EfficientNet B2, 和 ViT Base)。与BNN和DE Baselines相比,所提出的可信度包装器方法在不确定性估计方面表现更优,并在受损数据上实现了较低的期望校准误差。