YOLOv8n-Enhanced PCB Defect Detection: A Lightweight Method Integrating Spatial–Channel Reconstruction and Adaptive Feature Selection
Source abstract
Abstract
In response to the challenges of small-size defects and low recognition rates in Printed Circuit Boards (PCBs), as well as the need for lightweight detection models that can be embedded in portable devices, this paper proposes an improved defect detection method based on a lightweight shared convolutional head using YOLOv8n. Firstly, the Spatial and Channel reconstruction Convolution (SCConv) is embedded into the Cross Stage Partial with Convolutional Layer Fusion (C2f) structure of the backbone network, which reduces redundant computations and enhances the model’s learning capacity. Secondly, an adaptive feature selection module is integrated to improve the network’s ability to recognize small targets. Subsequently, a Shared Lightweight Convolutional Detection (SLCD) Head replaces the original Decoupled Head, reducing the model’s computational complexity while increasing detection accuracy. Finally, the Weighted Intersection over Union (WIoU) loss function is introduced to provide more precise evaluation results and improve generalization capability. Comparative experiments conducted on a public PCB dataset demonstrate that the improved algorithm achieves a mean Average Precision (mAP) of 98.6% and an accuracy of 99.8%, representing improvements of 3.8% and 3.1%, respectively, over the original model. The model size is 4.1 M, and its FPS is 144.1, meeting the requirements for real-time and lightweight portable deployment.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Industrial Vision Systems and Defect Detection · Advanced Neural Network Applications · Integrated Circuits and Semiconductor Failure Analysis
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Sino-French Engineer School, Nanjing University of Science and Technology, Nanjing 214400, China
References (25)
Show source-provided references
1. Research on Automatic Inspection System of Printed Circuit Board Based on Computer Vision10.1088/1742-6596/1861/1/012093 ↗
2. Automatic 3-D inspection of solder paste on surface mount printed circuit boards10.1016/0924-0136(91)90138-5 ↗
3. Printed circuit board defect detection based on MobileNet-Yolo-Fast10.1117/1.JEI.30.4.043004 ↗
4. LiDAR-only 3D object detection based on spatial context10.1016/j.jvcir.2023.103805 ↗
5. A lightweight and efficient one-stage detection framework10.1016/j.compeleceng.2022.108520 ↗
6. Nikhil, K., and Pravendra, S. (2023). Small and Dim Target Detection in IR Imagery: A Review. arXiv.
7. Li, J., Gu, J., Huang, Z., and Wen, J. (2019). Application Research of Improved YOLO V3 Algorithm in PCB Electronic Component Detection. Appl. Sci., 9.10.3390/app9183750 ↗
8. Printed Circuit Board Quality Detection Method Integrating Lightweight Network and Dual Attention Mechanism10.1109/ACCESS.2022.3198994 ↗
9. Xia, K., Lv, Z., Liu, K., Lu, Z., Zhou, C., Zhu, H., and Chen, X. (2023). Global contextual attention augmented YOLO with ConvMixer prediction heads for PCB surface defect detection. Sci. Rep., 13.10.1038/s41598-023-36854-2 ↗
10. A Design of Bare Printed Circuit Board Defect Detection System Based on YOLOv810.54097/hset.v57i.10002 ↗
11. PCB Defect Detection Algorithm Based on Improved YOLOv810.54097/ajst.v7i3.13420 ↗
12. Joseph, R., and Ali, F. (2016). YOLO9000: Better, Faster, Stronger. arXiv.
13. Shu, L., Lu, Q., Haifang, Q., Jianping, S., and Jiaya, J. (2018). Path Aggregation Network for Instance Segmentation. arXiv.
14. Adaptive Feature Pyramid Networks for Object Detection10.1109/ACCESS.2021.3100369 ↗
15. Zheng, Z., Wang, P., Liu, W., Li, J., Ye, R., and Ren, D. (2020, January 7–12). Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression. Proceedings of the AAAI Conference on Artificial Intelligence, New York, NY, USA.10.1609/aaai.v34i07.6999 ↗
16. He, J., Zhang, S., Yang, C., Wang, H., Gao, J., Huang, W., Wang, Q., Wang, X., Yuan, W., and Wu, Y. (2024). Pest recognition in microstates state: An improvement of YOLOv7 based on Spatial and Channel Reconstruction Convolution for feature redundancy and vision transformer with Bi-Level Routing Attention. Front. Plant Sci., 15.10.3389/fpls.2024.1327237 ↗
17. Diganta, M., Trikay, N., Ajay Uppili, A., and Qibin, H. (2020). Rotate to Attend: Convolutional Triplet Attention Module. arXiv.
18. Coordinate-Aware Mask R-CNN with Group Normalization: A underwater marine animal instance segmentation framework10.1016/j.neucom.2024.127488 ↗
19. FCOS: A Simple and Strong Anchor-Free Object Detector
20. Zanjia, T., Yuhang, C., Zewei, X., and Rong, Y. (2023). Wise-IoU: Bounding Box Regression Loss with Dynamic Focusing Mechanism. arXiv.
21. Xiao, G., Hou, S., and Zhou, H. (2024). PCB defect detection algorithm based on, CDI-YOLO. Sci. Rep., 14.10.1038/s41598-024-57491-3 ↗
22. Lightweight PCB defect detection algorithm based on MSD-YOLO10.1007/s10586-023-04156-x ↗
23. Yuan, Z., Tang, X., Ning, H., and Yang, Z. (2024). LW-YOLO: Lightweight Deep Learning Model for Fast and Precise Defect Detection in Printed Circuit Boards. Symmetry, 16.10.3390/sym16040418 ↗
24. Tang, J., Liu, S., Zhao, D., Tang, L., Zou, W., and Zheng, B. (2023). PCB-YOLO: An Improved Detection Algorithm of PCB Surface Defects Based on YOLOv5. Sustainability, 15.10.3390/su15075963 ↗
25. Du, B., Wan, F., Lei, G., Xu, L., Xu, C., and Xiong, Y. (2023). YOLO-MBBi: PCB Surface Defect Detection Method Based on Enhanced YOLOv5. Electronics, 12.10.3390/electronics12132821 ↗