4 个月前

提高神经网络中异常图像检测的可靠性

提高神经网络中异常图像检测的可靠性

摘要

我们研究了神经网络中检测分布外图像的问题。为此,我们提出了一种简单且有效的方法——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 中提及
ShiyuLiang/odin-pytorch
pytorch
GitHub 中提及

基准测试

基准方法指标
out-of-distribution-detection-on-imagenetResNet 34 + ODIN
AUROC: 90.8
out-of-distribution-detection-on-ms-1m-vs-ijbResNeXt 50 + ODIN
AUROC: 61.3

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
提高神经网络中异常图像检测的可靠性 | 论文 | HyperAI超神经