Defect detection of low-resolution ceramic substrate image based on knowledge distillation

被引:0
|
作者
Guo F. [1 ]
Sun X. [1 ]
Zhu Q. [1 ]
Huang M. [1 ]
Xu X. [2 ]
机构
[1] Key Laboratory of Advanced Process Control for Light Industry, Ministry of Education, Jiangnan University, Wuxi
[2] Wuxi CK Electric Control Equipment Co., Ltd, Wuxi
关键词
ceramic substrate; defect detection; knowledge distillation; YOLOv5;
D O I
10.37188/OPE.20233120.3065
中图分类号
学科分类号
摘要
Ceramic substrate is a vital foundational material of electronic devices, and implementing defect detection for ceramic substrates using machine vision technology combined with deep learning strategies holds significant importance in ensuring product quality. Increasing the field of view of the imaging equipment to make simultaneous imaging of multiple ceramic substrates possible can significantly improve the detection speed of a ceramic substrate. However, it also results in decreased image resolution and subsequently reduces the accuracy of defect detection. To solve these problems, a low-resolution ceramic substrate defect automatic detection method based on knowledge distillation is proposed. The method utilizes the YOLOv5 framework to construct a teacher network and a student network. Based on the idea of knowledge distillation, high-resolution image feature information obtained by the teacher network is used to guide the training of the student network to improve the defect detection ability of the student network for low-resolution ceramic substrate images. Moreover, a feature fusion module based on the coordinate attention (CA) idea is introduced into the teacher network, enabling it to learn features that adapt to both high-resolution and low-resolution image information, thus better guiding the training of the student network. Finally, a confidence loss function based on the gradient harmonizing mechanism (GHM) is introduced to enhance the defect detection rate. Experimental results demonstrate that the proposed ceramic substrate defect detection method based on knowledge distillation achieves an average accuracy and average recall of 96.80% and 90.01%, respectively, for the detection of five types of defect-stain, foreign matter, gold edge bulge, ceramic gap, and damage-in low-resolution (224×224) input images. Compared with current mainstream object detection algorithms, the proposed algorithm achieves better detection results. © 2023 Chinese Academy of Sciences. All rights reserved.
引用
收藏
页码:3065 / 3076
页数:11
相关论文
共 22 条
  • [1] GUO F,, ZHU Q B,, HUANG M,, Et al., Defect detection in ceramic substrate based on improved YOLOV4[J], Opt. Precision Eng, 30, 13, pp. 1631-1641, (2022)
  • [2] LI J C,, FANG F M,, MEI K F,, Et al., Multi-Scale Residual Network for Image Super-Resolution[M], Computer Vision - ECCV 2018, pp. 527-542, (2018)
  • [3] QI L,, KUEN J,, GU J X,, Et al., Multi-scale aligned distillation for low-resolution detection[C], 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 14438-14448, (2021)
  • [4] HINTON G,, VINYALS O, DEAN J., Distilling The Knowledge in a Neural Network[EB/OL], (2015)
  • [5] HE T,, SHEN C H,, TIAN Z,, Et al., Knowledge adaptation for efficient semantic segmentation[C], 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 578-587, (2019)
  • [6] LIU Y F,, SHU C Y,, WANG J D,, Et al., Structured knowledge distillation for dense prediction[J], IEEE Transactions on Pattern Analysis and Machine Intelligence, 45, 6, pp. 7035-7049, (2023)
  • [7] CHEN G B,, CHOI W,, YU X,, Et al., Learning efficient object detection models with knowledge distillation[C], Proceedings of the 31st International Conference on Neural Information Processing Systems, pp. 742-751, (2017)
  • [8] WANG T,, YUAN L,, ZHANG X P,, Et al., Distilling object detectors with fine-grained feature imitation[C], 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 4928-4937, (2019)
  • [9] ZHANG L,, MA K., Improve object detection with feature-based knowledge distillation: towards accurate and efficient detectors[C], International Conference on Learning Representations, (2020)
  • [10] CHU J H,, SHI L D,, JING P G,, Et al., Context-aware knowledge distillation network for object detection[J], Journal of Zhejiang University (Engineering Science, 56, 3, pp. 503-509, (2022)