Laser Journal, Volume. 45, Issue 8, 92(2024)

Research on workpiece anomaly detection algorithm based on inverse knowledge distillation

ZHANG Xiaoyong1...2, WANG Liming1,2,*, LI Xuan1,2, and HAN Xingcheng12 |Show fewer author(s)
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

    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.

    Tools

    Get Citation

    Copy Citation Text

    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

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Jan. 5, 2024

    Accepted: Dec. 20, 2024

    Published Online: Dec. 20, 2024

    The Author Email: Liming WANG (wlm@nuc.edu.cn)

    DOI:10.14016/j.cnki.jgzz.2024.08.092

    Topics