3 个月前

PaDiM:一种用于异常检测与定位的局部分布建模框架

PaDiM:一种用于异常检测与定位的局部分布建模框架

摘要

我们提出了一种新的图像异常检测与定位框架——Patch Distribution Modeling(PaDiM),该方法在单类学习(one-class learning)设定下,能够同时实现图像中异常的检测与精确定位。PaDiM利用预训练的卷积神经网络(CNN)提取图像块(patch)的嵌入表示,并采用多变量高斯分布对正常类样本进行概率建模。此外,该方法还充分利用了CNN不同语义层级之间的相关性,从而提升异常定位的精度。在MVTec AD和STC数据集上,PaDiM在异常检测与定位任务中均优于当前最先进的方法。为更贴近真实工业视觉检测场景,我们进一步扩展了评估协议,以在非对齐(non-aligned)数据集上评估异常定位算法的性能。得益于其卓越的性能表现与较低的计算复杂度,PaDiM成为众多工业应用场景中的理想候选方案。

基准测试

基准方法指标
anomaly-detection-on-hyper-kvasir-datasetPaDiM
AUC: 0.923
anomaly-detection-on-lagPaDiM
AUC: 0.688
anomaly-detection-on-mvtec-adPaDiM-R18-Rd100
Segmentation AUROC: 96.7
anomaly-detection-on-mvtec-adPaDiM
Detection AUROC: 97.9
anomaly-detection-on-mvtec-adPaDiM-WR50-Rd550
Detection AUROC: 95.3
FPS: 4.4
Segmentation AUROC: 97.5
anomaly-detection-on-visaPaDiM
Segmentation AUPRO (until 30% FPR): 85.9

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
PaDiM:一种用于异常检测与定位的局部分布建模框架 | 论文 | HyperAI超神经