4 个月前

用于工业异常检测的非对称师生网络

用于工业异常检测的非对称师生网络

摘要

工业缺陷检测通常采用异常检测(AD)方法来解决,这些方法在潜在缺陷的数据不足或完全缺失的情况下尤为适用。本研究发现了学生-教师网络在异常检测中存在的一些先前未知的问题,并提出了一种解决方案,即训练两个神经网络以对无缺陷的训练样本产生相同的输出。学生-教师网络的核心假设是,对于异常样本,由于其在训练过程中未出现,两个网络的输出之间的距离会更大。然而,以往的方法由于学生和教师网络架构相似,导致异常样本的输出距离过小。因此,我们提出了非对称学生-教师网络(Asymmetric Student-Teacher, AST)。我们将一个用于密度估计的归一化流作为教师网络进行训练,并将一个传统的前馈网络作为学生网络进行训练,以触发异常样本时较大的输出距离:归一化流的双射性确保了异常数据与正常数据相比,教师网络的输出具有更大的差异。而由于学生网络的基本架构不同,在训练分布之外它无法模仿这种差异。我们的AST网络通过归一化流补偿了错误估计的可能性值,这在以往的工作中曾被用作异常检测的一种替代方法。我们展示了该方法在当前最相关的两个缺陷检测数据集MVTec AD和MVTec 3D-AD上,在RGB图像和3D数据的图像级异常检测方面取得了最先进的结果。

代码仓库

marco-rudolph/ast
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
anomaly-detection-on-mvtec-3d-ad-1AST
Detection AUROC: 93.7
Segmentation AUROC: 97.6
anomaly-detection-on-mvtec-adAST
Detection AUROC: 99.2
Segmentation AUROC: 95.0
anomaly-detection-on-mvtec-loco-adAST
Detection AUROC (only logical): 79.7
Detection AUROC (only structural): 87.1
Segmentation AU-sPRO (until FPR 5%): 42.7
anomaly-detection-on-visaAST
Detection AUROC: 94.9
Segmentation AUPRO (until 30% FPR): 81.5
rgb-3d-anomaly-detection-and-segmentation-onAST
Detection AUCROC: 0.937
Segmentation AUCROC: 0.976

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于工业异常检测的非对称师生网络 | 论文 | HyperAI超神经