论文标题

ADP:图像异常检测的分割后不对称蒸馏

ADPS: Asymmetric Distillation Post-Segmentation for Image Anomaly Detection

论文作者

Xing, Peng, Tang, Hao, Tang, Jinhui, Li, Zechao

论文摘要

基于知识蒸馏的异常检测(KDAD)方法依赖于教师学生范式来检测和分段异常区域,以对比两个网络提取的独特功能。 However, existing KDAD methods suffer from two main limitations: 1) the student network can effortlessly replicate the teacher network's representations, and 2) the features of the teacher network serve solely as a ``reference standard" and are not fully leveraged. Toward this end, we depart from the established paradigm and instead propose an innovative approach called Asymmetric Distillation Post-Segmentation (ADPS). Our ADPS employs an asymmetric distillation范式采用与教师 - 学生网络的输入相同形式的不同形式的范式,推动学生网络学习异常区域的歧视表示形式。 同时,提出了一个定制的重量面膜块(WMB)来产生粗大的异常定位掩模,该掩模将从非对称范式获取的蒸馏知识转移到教师网络。拟议的分割后模块(PSM)配备了WMB,能够有效地检测和分段异常区域,并具有良好的结构和清晰的边界。实验结果表明,所提出的ADP的表现优于检测和分割异常的最新方法。出乎意料的是,ADP分别在MVTEC AD和Kolektorsd2数据集上显着提高了平均精度(AP)度量的9%和20%。

Knowledge Distillation-based Anomaly Detection (KDAD) methods rely on the teacher-student paradigm to detect and segment anomalous regions by contrasting the unique features extracted by both networks. However, existing KDAD methods suffer from two main limitations: 1) the student network can effortlessly replicate the teacher network's representations, and 2) the features of the teacher network serve solely as a ``reference standard" and are not fully leveraged. Toward this end, we depart from the established paradigm and instead propose an innovative approach called Asymmetric Distillation Post-Segmentation (ADPS). Our ADPS employs an asymmetric distillation paradigm that takes distinct forms of the same image as the input of the teacher-student networks, driving the student network to learn discriminating representations for anomalous regions. Meanwhile, a customized Weight Mask Block (WMB) is proposed to generate a coarse anomaly localization mask that transfers the distilled knowledge acquired from the asymmetric paradigm to the teacher network. Equipped with WMB, the proposed Post-Segmentation Module (PSM) is able to effectively detect and segment abnormal regions with fine structures and clear boundaries. Experimental results demonstrate that the proposed ADPS outperforms the state-of-the-art methods in detecting and segmenting anomalies. Surprisingly, ADPS significantly improves Average Precision (AP) metric by 9% and 20% on the MVTec AD and KolektorSDD2 datasets, respectively.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源