论文标题
重建的学生教师和判别网络用于异常检测
Reconstructed Student-Teacher and Discriminative Networks for Anomaly Detection
论文作者
论文摘要
异常检测是计算机视觉中的重要问题;但是,异常样本的稀缺使这项任务变得困难。因此,最近的异常检测方法仅使用了没有异常区域训练的正常图像。在这项工作中,提出了一种强大的异常检测方法,该方法是根据学生和教师网络组成的学生教师特征金字塔匹配(STPM)提出的。生成模型是检测异常的另一种方法。他们从输入中重建正常图像,并计算预测的正常和输入之间的差异。不幸的是,STPM没有生成正常图像的能力。为了提高STPM的准确性,这项工作使用学生网络(如生成模型)来重建正常功能。这提高了准确性;但是,正常图像的异常图不干净,因为STPM不使用异常图像进行训练,这降低了图像级异常检测的准确性。为了进一步提高准确性,在我们的方法中使用了一个来自异常图的伪异常训练的歧视性网络,该网络由两对学生教师网络和一个歧视性网络组成。该方法在MVTEC异常检测数据集上表现出很高的精度。
Anomaly detection is an important problem in computer vision; however, the scarcity of anomalous samples makes this task difficult. Thus, recent anomaly detection methods have used only normal images with no abnormal areas for training. In this work, a powerful anomaly detection method is proposed based on student-teacher feature pyramid matching (STPM), which consists of a student and teacher network. Generative models are another approach to anomaly detection. They reconstruct normal images from an input and compute the difference between the predicted normal and the input. Unfortunately, STPM does not have the ability to generate normal images. To improve the accuracy of STPM, this work uses a student network, as in generative models, to reconstruct normal features. This improves the accuracy; however, the anomaly maps for normal images are not clean because STPM does not use anomaly images for training, which decreases the accuracy of the image-level anomaly detection. To further improve accuracy, a discriminative network trained with pseudo-anomalies from anomaly maps is used in our method, which consists of two pairs of student-teacher networks and a discriminative network. The method displayed high accuracy on the MVTec anomaly detection dataset.