arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2110.03396eess.IVcs.CV
收藏

AnoSeg:基于自监督学习的异常分割网络

AnoSeg: Anomaly Segmentation Network Using Self-Supervised Learning

Sogang University(西江大学);LG Display(乐金显示);Pukyong National University(釜庆国立大学)

Jouwon Song, Kyeongbo Kong, Ye-In Park, Seong-Gyun Kim, Suk-Ju Kang

摘要:

异常分割是大规模工业制造中的一个重要组成部分,它定位缺陷区域。然而,最近的研究主要集中在异常检测。本文提出了一种新的异常分割网络(AnoSeg),可以使用自监督学习直接生成准确的异常图。对于高度准确的异常分割,提出的AnoSeg考虑了三种新技术:基于硬增强的异常数据生成,具有像素和对抗性损失的自监督学习,以及坐标通道级联。首先,为了生成正常数据的合成异常图像和参考掩模,所提出的方法使用硬增强来改变正态样本分布。然后,所提出的AnoSeg训练在一个自监督学习的方式从合成异常数据和正常数据。最后,将表示像素位置信息的坐标通道连接到AnoSeg的输入,以考虑图像中每个像素的位置关系。所估计的异常图还可以用于提高异常检测的性能。我们的实验表明,该方法优于国家的最先进的异常检测和异常分割方法的MVTec AD数据集。此外,我们比较了所提出的方法与现有的方法,通过在分割任务中常用的交集(IoU)度量,并证明了我们的方法异常分割的优越性。

英文摘要:

Anomaly segmentation, which localizes defective areas, is an important component in large-scale industrial manufacturing. However, most recent researches have focused on anomaly detection. This paper proposes a novel anomaly segmentation network (AnoSeg) that can directly generate an accurate anomaly map using self-supervised learning. For highly accurate anomaly segmentation, the proposed AnoSeg considers three novel techniques: Anomaly data generation based on hard augmentation, self-supervised learning with pixel-wise and adversarial losses, and coordinate channel concatenation. First, to generate synthetic anomaly images and reference masks for normal data, the proposed method uses hard augmentation to change the normal sample distribution. Then, the proposed AnoSeg is trained in a self-supervised learning manner from the synthetic anomaly data and normal data. Finally, the coordinate channel, which represents the pixel location information, is concatenated to an input of AnoSeg to consider the positional relationship of each pixel in the image. The estimated anomaly map can also be utilized to improve the performance of anomaly detection. Our experiments show that the proposed method outperforms the state-of-the-art anomaly detection and anomaly segmentation methods for the MVTec AD dataset. In addition, we compared the proposed method with the existing methods through the intersection over union (IoU) metric commonly used in segmentation tasks and demonstrated the superiority of our method for anomaly segmentation.

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

↑