Anomaly-Free Prior Guided Knowledge Distillation for Industrial Anomaly Detection
Gang Li, Tianjiao Chen, Min Li, Delong Han, Mingle Zhou · 2024
In industrial manufacturing, visual anomaly detection is critical for maintaining product quality by detecting and preventing production anomalies. Anomaly detection methods based on knowledge distillation demonstrate promising performance in addressing the unpredictability and diversity of anomalies. However, they suffer from a lack of effective guidance from anomaly-free priors when handling anomalous features and underutilize multi-scale features during the segmentation scoring stage, yielding suboptimal detection results. To alleviate these issues, we propose an Anomaly-free Prior Guided knowledge distillation (APG) for industrial anomaly detection. Firstly, it filters the abnormal features by training the de-noising target network with knowledge distillation structure. Concurrently, we propose the Prior Perception Propagation Module (P3M), which extracts more efficient anomaly-free features by imposing constraints on anomalous features. Secondly, we propose the Multi-scale Prior Guided Fusion Module (MPGFM) to improve anomaly detection accuracy by utilizing anomaly-free features from the target network as priors to guide the generation and fusion of cross-scale differential features. Finally, the Global Perception Enhancement Module (GPEM) is proposed to construct an anomaly scoring network, leveraging comprehensive scene features to enhance the detection and localization performance of numerous small-target anomalies in industrial manufacturing. Extensive experiments on the MVTecAD and BTAD datasets show that the proposed method demonstrates a consistent and significant outperformance against competing methods.