
摘要
我们研究了神经网络中检测分布外图像的问题。为此,我们提出了一种简单且有效的方法——ODIN(Out-of-Distribution Detector for Neural Networks),该方法无需对预训练的神经网络进行任何修改。我们的方法基于以下观察:通过温度缩放和对输入添加微小扰动,可以分离出分布内图像和分布外图像的softmax分数分布,从而实现更有效的检测。一系列实验表明,ODIN兼容多种网络架构和数据集,并且在这一任务上始终大幅优于基线方法,确立了新的最先进性能。例如,在DenseNet(应用于CIFAR-10)上,当真阳性率为95%时,ODIN将假阳性率从基线的34.7%降低到4.3%。
代码仓库
facebookresearch/odin
官方
pytorch
GitHub 中提及
remic-othr/openmibood
pytorch
GitHub 中提及
JoonHyung-Park/ODIN
pytorch
GitHub 中提及
kingjamessong/rankfeat
pytorch
GitHub 中提及
guyAmit/GLOD
pytorch
GitHub 中提及
ericjang/odin
pytorch
GitHub 中提及
jun-cen/unified_open_set_recognition
pytorch
GitHub 中提及
ShiyuLiang/odin-pytorch
pytorch
GitHub 中提及
kobybibas/pnml_ood_detection
pytorch
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| out-of-distribution-detection-on-imagenet | ResNet 34 + ODIN | AUROC: 90.8 |
| out-of-distribution-detection-on-ms-1m-vs-ijb | ResNeXt 50 + ODIN | AUROC: 61.3 |