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

Yibo Chen,Ke Zhu,Haolong Peng,Jianming Zhang
DOI: https://doi.org/10.1109/DDCLS61622.2024.10606616
2024-05-17
Abstract:“With the progression of network perception and detection capabilities, multi-class anomaly detection and localization have become promising research areas. The primary challenge lies in mitigating minor textural defects while maintaining adaptability to a variable external environment. In response to this, we introduce a state-of-the-art pipeline, the ViT-based Discriminative Knowledge Distillation (VDKD). Contrary to existing lightweight models, VDKD uniquely leverages a robust teacher network to train a compact student model using innovative modules. Furthermore, the integrated Semantic Alignment Module (SAM) and Abnormal Assessment Module (AAM) assist in distilling essential information. We also propose the Discriminative Feature Judgment Module (DFJM), designed to emulate the teacher's output within the student model with limited sample intervention. Extensive self-supervised experiments on texture datasets, including KSDD2, MT, and MVTec-AD, underscore the effectiveness of our approach with VDKD.
Computer Science
What problem does this paper attempt to address?