3 个月前

SimpleNet:一种用于图像异常检测与定位的简单网络

SimpleNet:一种用于图像异常检测与定位的简单网络

摘要

我们提出了一种简单且易于应用的网络(称为 SimpleNet),用于异常检测与定位。SimpleNet 由四个核心组件构成:(1)一个预训练的特征提取器,用于生成局部特征;(2)一个浅层特征适配器,用于将源域的局部特征迁移至目标域;(3)一个简化的异常特征生成器,通过向正常特征添加高斯噪声来合成异常特征;(4)一个二分类异常判别器,用于区分异常特征与正常特征。在推理阶段,异常特征生成器将被移除。本方法基于三个核心直觉:第一,将预训练特征转换为面向目标域的特征,有助于缓解域偏移问题;第二,在特征空间中生成合成异常更为有效,因为缺陷在图像空间中可能缺乏显著共性;第三,采用简洁的判别器结构具有更高的效率与实用性。尽管结构简单,SimpleNet 在定量与定性评估上均优于现有方法。在 MVTec AD 基准测试中,SimpleNet 实现了 99.6% 的异常检测 AUROC,相较于表现最佳的现有模型,错误率降低了 55.5%。此外,SimpleNet 具有极高的运行效率,在 NVIDIA 3080 Ti GPU 上可达到高达 77 FPS 的帧率。同时,该方法在单类新奇检测(One-Class Novelty Detection)任务中也展现出显著的性能提升。代码地址:https://github.com/DonaldRR/SimpleNet

代码仓库

donaldrr/simplenet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
anomaly-classification-on-goodsadSimpleNet
AUPR: 78.7
AUROC: 75.3
anomaly-detection-on-mvtec-adSimpleNet
Detection AUROC: 99.6
FPS: 77(FP32 on 3080ti)
Segmentation AUROC: 98.1
anomaly-detection-on-mvtec-loco-adSimpleNet
Avg. Detection AUROC: 77.6
Detection AUROC (only logical): 71.5
Detection AUROC (only structural): 83.7
Segmentation AU-sPRO (until FPR 5%): 36.3

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SimpleNet:一种用于图像异常检测与定位的简单网络 | 论文 | HyperAI超神经