A semi-supervised deep learning framework for efficient PCB defect detection confidence-thresholded self-training with YOLOv5
View Abstract View PDF Download PDF

Keywords

Deep learning, Printed circuit board, Defect detection, Pseudo-labeling, Semi-supervised learning, YOLOv5.

Abstract

Printed Circuit Board (PCB) defect inspection is critical in electronics manufacturing, yet annotating large training sets is labor-intensive and costly. This paper proposes a semi-supervised PCB defect detection framework based on YOLOv5, which leverages a small labeled dataset and a larger pool of unlabeled images through an iterative self-training pipeline. A YOLOv5 detector is first trained on the limited labeled data, then used to generate pseudo-labels on unlabeled images; high-confidence detections are retained as defect annotations, and the model is retrained on the combination of labeled and pseudo-labeled data. This pseudo-labeling and retraining cycle is repeated for multiple iterations to progressively refine the detector. The approach is evaluated on a PCB defect dataset with 100 labeled and 1,000 unlabeled images, and shows significant gains over a fully supervised baseline. The proposed semi-supervised YOLOv5 achieves 91.6% mAP@0.5 with only 100 labeled images, outperforming both the baseline (87.0% mAP) and prior semi-supervised methods, while substantially improving recall and precision. The results demonstrate that the method effectively reduces annotation effort while maintaining high detection accuracy, providing a simple, confidence-thresholded self-training strategy for deploying PCB defect detectors under limited labeling resources. This work directly supports SDG 9 by enabling cost-efficient, high-accuracy AI-based PCB inspection that strengthens intelligent and sustainable manufacturing systems.

https://doi.org/10.55493/5003.v16i2.5901
View Abstract View PDF Download PDF

Downloads

Download data is not yet available.