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

CutPaste:用于异常检测和定位的自监督学习

CutPaste: Self-Supervised Learning for Anomaly Detection and Localization

Google Cloud AI Research(谷歌云人工智能研究院)

Chun-Liang Li, Kihyuk Sohn, Jinsung Yoon, Tomas Pfister

摘要:

我们的目标是构建一个高性能的缺陷检测模型,检测未知的异常模式的图像没有异常数据。为此,我们提出了一个两阶段的框架,只使用正常的训练数据建立异常检测器。我们首先学习自我监督的深度表示,然后在学习的表示上构建一个生成的单类分类器。我们通过对CutPaste中的正常数据进行分类来学习表示,CutPaste是一种简单的数据增强策略,可以剪切图像补丁并粘贴在大图像的随机位置。我们的MVTec异常检测数据集的实证研究表明,该算法是通用的,能够检测到各种类型的现实世界的缺陷。当从头开始学习表示时,我们将以前的艺术提高了3.1 AUC。通过对ImageNet上的预训练表示进行迁移学习,我们实现了新的96.6 AUC。最后,我们扩展了框架来学习和提取补丁的表示,以允许在训练过程中定位没有注释的缺陷区域。

英文摘要:

We aim at constructing a high performance model for defect detection that detects unknown anomalous patterns of an image without anomalous data. To this end, we propose a two-stage framework for building anomaly detectors using normal training data only. We first learn self-supervised deep representations and then build a generative one-class classifier on learned representations. We learn representations by classifying normal data from the CutPaste, a simple data augmentation strategy that cuts an image patch and pastes at a random location of a large image. Our empirical study on MVTec anomaly detection dataset demonstrates the proposed algorithm is general to be able to detect various types of real-world defects. We bring the improvement upon previous arts by 3.1 AUCs when learning representations from scratch. By transfer learning on pretrained representations on ImageNet, we achieve a new state-of-theart 96.6 AUC. Lastly, we extend the framework to learn and extract representations from patches to allow localizing defective areas without annotations during training.

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

↑