arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.02576cs.CV

基于原型的神经网络用于图像异常检测与定位

A Prototype-Based Neural Network for Image Anomaly Detection and Localization

School of Computer Science and Engineering, University of Electronic Science and Technology of China(电子科技大学计算机科学与工程学院)

Chao Huang, Zhao Kang, Hong Wu

AI总结:

本文提出ProtoAD,一种基于原型的神经网络,通过预训练特征提取、非参数聚类学习原型,并以端到端方式实现图像异常检测与定位,在MVTec AD和BTAD数据集上以更高推理速度取得竞争性能。

摘要:

图像异常检测与定位不仅执行图像级别的异常分类,还定位像素级别的异常区域。近年来,由于其在各个领域的广泛应用,该方向受到了大量研究关注。本文提出了ProtoAD,一种基于原型的神经网络,用于图像异常检测与定位。首先,通过一个在自然图像上预训练的深度网络提取正常图像的块特征。然后,通过非参数聚类学习正常块特征的原型。最后,我们通过将特征提取网络与L2特征归一化、一个1×1卷积层、通道最大池化以及减法操作相结合,构建了一个图像异常定位网络(ProtoAD)。我们将原型用作1×1卷积层的卷积核;因此,我们的神经网络不需要训练阶段,可以以端到端的方式进行异常检测与定位。在两个具有挑战性的工业异常检测数据集MVTec AD和BTAD上进行的大量实验表明,与最先进的方法相比,ProtoAD在更高的推理速度下实现了具有竞争力的性能。源代码可在以下网址获取:https://github.com/98chao/ProtoAD。

英文摘要:

Image anomaly detection and localization perform not only image-level anomaly classification but also locate pixel-level anomaly regions. Recently, it has received much research attention due to its wide application in various fields. This paper proposes ProtoAD, a prototype-based neural network for image anomaly detection and localization. First, the patch features of normal images are extracted by a deep network pre-trained on nature images. Then, the prototypes of the normal patch features are learned by non-parametric clustering. Finally, we construct an image anomaly localization network (ProtoAD) by appending the feature extraction network with $L2$ feature normalization, a $1\times1$ convolutional layer, a channel max-pooling, and a subtraction operation. We use the prototypes as the kernels of the $1\times1$ convolutional layer; therefore, our neural network does not need a training phase and can conduct anomaly detection and localization in an end-to-end manner. Extensive experiments on two challenging industrial anomaly detection datasets, MVTec AD and BTAD, demonstrate that ProtoAD achieves competitive performance compared to the state-of-the-art methods with a higher inference speed. The source code is available at: https://github.com/98chao/ProtoAD.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑