基于回顾蒸馏学习的无监督工业品缺陷检测方法

在自动化工业生产环境中高效地完成产品质检是生产过程中的重要任务之一，提出一种基于回顾蒸馏学习的无监督工业品缺陷检测方法（Retro-KD）.首先，针对缺陷产生的未知性问题，采用无监督的方式训练蒸馏学习模型，同时，为了充分地利用蒸馏学习中的信息传递机制，利用中间层特征提取模块完善教师网络中的特征架构；其次，提出迭代信息融合模块，回顾地传递中间层信息，指导学生网络拟合正样本特征分布，放大缺陷样本差异性；再引入相似性度量（Structural Similarity,SSIM），增强教师与学生网络在图像空间中的相似度；最后，采用基于梯度变化的缺陷分割方法得到像素级的定位图.在MVTec-AD和Magnetic-Tile两个工业数据集上验证了该方法的有效性，其AUROC(Area under ROC)与ACC(Accuracy)指标分别提升了1.9%与1.3%.