Abstract:In recent years, unsupervised anomaly detection based on knowledge distillation has gained special attention and some promising results have been reported in the literature. However, there is still room to improve the sensitivity of the model to anomalies. To do so, in this paper, a novel two-stage training method in terms of reverse knowledge distillation is proposed for anomaly detection and localization. Firstly, self-supervised mask training is introduced after the initial training of reverse knowledge distillation, which contributes greatly to the model detection against random unknown anomalies by self-simulating anomalies and forcing repair so as to reinforce learning single-category prototype patterns. Then, with the aim to facilitate the anomaly localization, an anomaly feature diffusion module is employed, which strengthens the correlation between pixels and helps spread the anomaly information to the surrounding area by covering the central pixel and reconstructing the representation for features after diffused. Furthermore, inspired by the human memory mechanism, an innovative normalized embedding memory bank is adopted to regulate the low-dimensional representations after embedding the encoding, inhibit the flow of anomalous information to the student decoder, and encourage the high-quality reconstruction of the model. Finally, the contextual similarity loss is used to guide the student model to learn knowledge representations from a contextual perspective, capture higher-order similarities between teachers and students, and delicately evaluate the differences between teachers and students. The empirical experiments conducted on the MVTec dataset show that the proposed SSMRKD method can achieve the best performance compared to other state-of-the-art methods, meanwhile extensive experiments of the ablation study validate the contribution of each component of the model. In addition, the advanced performance achieved on four commonly used datasets verifies the generalizability of the model in the industrial domain. Overall, the proposed SSMRKD method has significant advantages over the state-of-the-art anomaly detection methods.

Multi-Channel Knowledge Distillation Via Efficient Feature Fusion Module for Anomaly Segmentation

Pull & Push: Leveraging Differential Knowledge Distillation for Efficient Unsupervised Anomaly Detection and Localization.

Knowledge Distillation Method for Surface Defect Detection.

VDKD: A ViT-Based Student-Teacher Knowledge Distillation for Multi-Texture Class Anomaly Detection

Enhanced multi-scale features mutual mapping fusion based on reverse knowledge distillation for industrial anomaly detection and localization

Informative knowledge distillation for image anomaly segmentation

Dual-Modeling Decouple Distillation for Unsupervised Anomaly Detection

A Diffusion-Based Framework for Multi-Class Anomaly Detection

Dual-student knowledge distillation for visual anomaly detection

Unsupervised anomaly detection and localization via bidirectional knowledge distillation

RDMS: Reverse distillation with multiple students of different scales for anomaly detection

Two-stage reverse knowledge distillation incorporated and Self-Supervised Masking strategy for industrial anomaly detection

Cosine similarity knowledge distillation for surface anomaly detection

ADPS: Asymmetric Distillation Post-Segmentation for Image Anomaly Detection

[Formulating a diagnosis in psychiatry].

Remembering Normality: Memory-guided Knowledge Distillation for Unsupervised Anomaly Detection

Advancing Pre-trained Teacher: Towards Robust Feature Discrepancy for Anomaly Detection

Anomaly detection based on multi-teacher knowledge distillation

Defect Detection Method Based on Knowledge Distillation

Dual-Student Knowledge Distillation Networks for Unsupervised Anomaly Detection

Mixed Noise-Guided Mutual Constraint Framework for Unsupervised Anomaly Detection in Smart Industries