【2023工业异常检测文献】SimpleNet

news2024/9/20 12:11:24

SimpleNet:ASimpleNetworkforImageAnomalyDetectionandLocalization

1、Background

图像异常检测和定位主要任务是识别并定位图像中异常区域。

工业异常检测最大的难题在于异常样本少，一般采用无监督方法，在训练过程中只使用正常样本。

解决工业异常检测主要有三种方法，分别是 基于重建的方法 、 基于合成的方法 、 基于嵌入的方法 。

基于重建（生成）的 AD

通过比较原始数据和重建后的数据之间的差异，可以检测出异常区域。
自编码器（Autoencoder）或生成对抗网络（GAN）。
会导致 过拟合 现象，重建或生成异常的数据，与原始数据无法进行区分，以至于误检。

基于合成的 AD

通过在正常的训练数据上添加噪声合成异常样本来训练模型。
在正常图像数据上应用某种形式的变换或噪声，以模拟异常。这些变换可能包括随机裁剪和粘贴图像块（CutPaste）、添加高斯噪声、进行图像扭曲等。
主要组成模块：合成异常样本、特征提取（CNN等）、训练分类器（SVM、神经网络等）。
基于合成的异常检测方法通过训练在无异常图像上生成的合成异常来估计正常和异常之间的决策边界，但是， 合成的异常图像不够真实 。来自合成数据的特征可能偏离正常特征很远，使用这种负样本进行训练可能导致正态特征空间的边界松散，这意味着分布中特征空间可能包含不明显的缺陷。

基于嵌入的 AD

通过将图像特征映射到一个嵌入空间来检测异常。使用ImageNet预训练卷积神经网络(CNN)提取广义正态特征。然后采用多元高斯分布、归一化流、记忆库等统计算法嵌入正态特征分布。通过将输入特征与学习分布或记忆特征进行比较来检测异常。
预训练模型：使用大规模数据集（如ImageNet）预训练一个深度神经网络，如卷积神经网络（CNN）。这个网络用于从图像中提取特征。
特征提取：将预训练的模型应用于新的图像数据，提取高级特征。这些特征捕捉了图像的重要视觉信息。
特征嵌入：使用特定的技术（如主成分分析PCA、t-SNE、自编码器等）将提取的特征映射到一个低维嵌入空间。在这个空间中，相似的数据点（如正常图像的特征）彼此靠近，而不同的数据点（如异常图像的特征）则相距较远。
学习正常数据分布：在训练阶段，只使用正常图像的数据来训练模型，从而学习正常数据在嵌入空间中的分布。这可以通过多种方式实现，例如：
- 统计方法：如多元高斯分布，用于建模正常特征的分布。
- 内存库方法：存储正常样本的特征，以便在测试时比较新样本与存储的特征。
- 流模型：如归一化流，将数据映射到一个新的分布，通常假设为高斯分布。
然而，工业图像通常具有与ImageNet不同的分布，直接使用这些有偏差的特征可能会导致不匹配问题。此外，统计算法往往存在 计算复杂度高、内存消耗大 的问题。

为了缓解上述问题，我们提出了一种新的异常检测和定位网络，称为 SimpleNet 。 SimpleNet 利用了基于合成和基于嵌入的方式，并做了一些改进。

首先，我们建议使用特征适配器来产生面向目标的特征，而不是直接使用预训练的特征来减少域偏差。

其次，我们提出在特征空间中对正常特征施加噪声来生成异常特征，而不是直接在图像上合成异常特征。

我们认为，通过适当校准噪声的尺度，可以获得紧密有界的正态特征空间。

第三，我们通过训练一个简单的鉴别器来简化异常检测过程，该鉴别器比上述基于嵌入的方法采用的复杂统计算法计算效率高得多。

具体来说，SimpleNet 使用预训练的主干进行常规特征提取，然后使用特征适配器将特征转移到目标域。然后，通过在适应的正态特征上加入高斯噪声，简单地生成异常特征，在这些特征上训练一个由几层 MLP 组成的简单判别器来判别异常。