赞
踩
深度学习论文: SimpleNet: A Simple Network for Image Anomaly Detection and Localization及其PyTorch实现
SimpleNet: A Simple Network for Image Anomaly Detection and Localization
PDF: https://arxiv.org/pdf/2303.15140.pdf
PyTorch代码: https://github.com/shanglianlm0525/CvPytorch
PyTorch代码: https://github.com/shanglianlm0525/PyTorch-Networks
本文提出了一种简单有效的无监督异常检测和定位方法—SimpleNet,SimpleNet由几个简单的神经网络模块组成,易于训练和应用于工业场景。虽然简单,但SimpleNet在MVtec AD基准测试中实现了最高的性能和最快的推理速度。
SimpleNet使用特征适配器在目标数据集上执行迁移学习,以减轻预训练cnn的偏差。SimpleNet提出在特征空间中合成畸变,而不是直接在图像上合成畸变。SimpleNet在推理时遵循单流方式,完全由传统的CNN块构建,便于快速训练,推理和工业应用。
SimpleNet由四个部分组成:
(1)生成局部特征的预训练特征提取器;
(2)将局部特征转移到目标域的浅特征适配器;特征适配器可以由简单的神经块组成,如全连接层或多层感知器(MLP)。通过实验发现,单层全连接产生了良好的性能。
(3)通过向正常特征添加高斯噪声来伪造异常特征的简单异常特征生成器;从下图可以看出,适应特征的每个维度的标准差趋于一致,因此,在区分异常特征和正常特征时,特征空间趋于紧凑。
(4)区分异常特征和正常特征的二进制异常鉴别器;使用2层多层感知器(MLP)结构作为普通分类器。
在推理过程中,异常特征生成器将被丢弃。
提出的方法基于三个直觉:
训练过程如下:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。