• Laser Journal
  • Vol. 45, Issue 8, 92 (2024)
ZHANG Xiaoyong1,2, WANG Liming1,2,*, LI Xuan1,2, and HAN Xingcheng1,2
Author Affiliations
  • 1College of Information and Communication Engineering, North University of China, Taiyuan 030051, China
  • 2Shanxi Key Laboratory of Signal Capturing and Processing (North University of China), Taiyuan 030051, China
  • show less
    DOI: 10.14016/j.cnki.jgzz.2024.08.092 Cite this Article
    ZHANG Xiaoyong, WANG Liming, LI Xuan, HAN Xingcheng. Research on workpiece anomaly detection algorithm based on inverse knowledge distillation[J]. Laser Journal, 2024, 45(8): 92 Copy Citation Text show less

    Abstract

    Workpiece anomaly detection is a key link in production. Due to the small number of abnormal samples and large randomness, supervised learning can not fully learn all types of anomalies, and there exists the problem of poor model stability. In order to solve the above problems, this paper studies an unsupervised workpiece anomaly detection algorithm based on reverse knowledge distillation, and uses the teacher model and student model designed by ResNet network structure as the backbone network. The teacher model truly extracts the image features, the student model reconstructs the image according to the prior knowledge, and adopts the reverse structure to expand the specificity of the abnormal condition. A memory module and a mask attention module are added to extract the multi-dimensional feature information of the sample to avoid missing the details of the image; after the memory module, a mask attention mechanism is added to integrate the multi-dimensional and multi-level features of the image. the accuracy of detection is further improved. The experimental results on two open industrial anomaly detection data sets show that the proposed algorithm is 5% higher than the general knowledge distillation algorithm AUC by 7%, and the effect of locating subtle anomalies is better.
    ZHANG Xiaoyong, WANG Liming, LI Xuan, HAN Xingcheng. Research on workpiece anomaly detection algorithm based on inverse knowledge distillation[J]. Laser Journal, 2024, 45(8): 92
    Download Citation