论文标题
清除记忆启动的自动编码器,用于表面缺陷检测
Clear Memory-Augmented Auto-Encoder for Surface Defect Detection
论文作者
论文摘要
在表面缺陷检测中,由于阳性和负样品数量的极度失衡,基于阳性的异常检测方法已受到越来越多的关注。具体而言,基于重建的方法是最受欢迎的方法。但是,现有方法要么难以修复异常的前景,要么重建清晰的背景。因此,我们提出了一个清晰的内存启动自动编码器(CMA-AE)。首先,我们提出了一个新颖的清晰内存调节模块(CMAM),该模块将编码和记忆编码结合在一起,以忘记和输入的方式,从而修复异常的前景并保留清晰的背景。其次,提出了一般人工异常产生算法(GAAGA)来模拟尽可能逼真且特征富含特征的异常。最后,我们提出了一种用于缺陷分割的新型多量表残差检测方法(MSFR),这使缺陷位置更加准确。广泛的比较实验表明,CMA-AE可以达到最新的检测准确性,并在工业应用中显示出巨大的潜力。
In surface defect detection, due to the extreme imbalance in the number of positive and negative samples, positive-samples-based anomaly detection methods have received more and more attention. Specifically, reconstruction-based methods are the most popular. However, existing methods are either difficult to repair abnormal foregrounds or reconstruct clear backgrounds. Therefore, we propose a clear memory-augmented auto-encoder (CMA-AE). At first, we propose a novel clear memory-augmented module (CMAM), which combines the encoding and memoryencoding in a way of forgetting and inputting, thereby repairing abnormal foregrounds and preserving clear backgrounds. Secondly, a general artificial anomaly generation algorithm (GAAGA) is proposed to simulate anomalies that are as realistic and feature-rich as possible. At last, we propose a novel multi scale feature residual detection method (MSFR) for defect segmentation, which makes the defect location more accurate. Extensive comparison experiments demonstrate that CMA-AE achieves state-of-the-art detection accuracy and shows great potential in industrial applications.