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.

AEKD: Unsupervised auto-encoder knowledge distillation for industrial anomaly detection

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

Unsupervised anomaly detection and localization via bidirectional knowledge distillation

UNSUPERVISED ANOMALY DETECTION WITH SELF-TRAINING AND KNOWLEDGE DISTILLATION

Autoencoder-Like Knowledge Distillation Network for Anomaly Detection

Auto-AD: Autonomous Hyperspectral Anomaly Detection Network Based on Fully Convolutional Autoencoder

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

Anomaly Detection via Reverse Distillation from One-Class Embedding

Cosine similarity knowledge distillation for surface anomaly detection

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

Anomaly detection based on multi-teacher knowledge distillation

Unlocking the Potential of Reverse Distillation for Anomaly Detection

Multimodal Industrial Anomaly Detection by Crossmodal Reverse Distillation

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

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

Dual-student knowledge distillation for visual anomaly detection

Dual-Modeling Decouple Distillation for Unsupervised Anomaly Detection

ADPS: Asymmetric Distillation Post-Segmentation for Image Anomaly Detection

ADPS: Asymmetric Distillation Postsegmentation for Image Anomaly Detection

Data-Free Adversarial Distillation

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