3 个月前

CSI:通过分布外实例的对比学习进行新奇检测

CSI:通过分布外实例的对比学习进行新奇检测

摘要

异常检测(Novelty Detection),即判断给定样本是否来自训练数据分布之外,对于实现可靠的机器学习至关重要。为此,研究者们提出了多种方法,旨在学习适用于异常检测的表示,并基于此类表示设计相应的检测评分。本文提出了一种简单而有效的新型方法——对比位移样本(Contrasting Shifted Instances, CSI),该方法受到视觉表示对比学习近期成功应用的启发。具体而言,与传统对比学习仅将样本与其它实例进行对比不同,我们的训练策略额外引入了对样本自身分布位移增强版本的对比。基于这一训练机制,我们进一步设计了一种专为该方法量身定制的新型检测评分。实验结果表明,无论在无标签单类、无标签多类还是有标签多类等不同异常检测场景下,所提方法在多个图像基准数据集上均展现出显著优势。相关代码与预训练模型已开源,地址为:https://github.com/alinlab/CSI。

代码仓库

alinlab/CSI
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
anomaly-detection-on-anomaly-detection-onCSI
Network: ResNet-18
ROC-AUC: 90.3
anomaly-detection-on-anomaly-detection-on-1CSI
Network: ResNet-18
ROC-AUC: 71.5
anomaly-detection-on-anomaly-detection-on-2CSI
Network: ResNet-18
ROC-AUC: 94.7
anomaly-detection-on-one-class-cifar-10CSI
AUROC: 94.3
anomaly-detection-on-one-class-cifar-100CSI
AUROC: 89.6
anomaly-detection-on-one-class-imagenet-30CSI
AUROC: 91.6
anomaly-detection-on-unlabeled-cifar-10-vsCSI
AUROC: 89.3
Network: ResNet-18

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
CSI:通过分布外实例的对比学习进行新奇检测 | 论文 | HyperAI超神经