An Adaptive Feature Reconstruction Network for the Precise Segmentation of Surface Defects on Printed Circuit Boards

Kang Danqing,Lai Jianhuang,Zhu Junyong,Han Yu
DOI: https://doi.org/10.1007/s10845-022-02008-w
IF: 8.3
2022-01-01
Journal of Intelligent Manufacturing
Abstract:Segmentation networks based on deep learning are widely used in the field of industrial vision inspection, including for the precise segmentation of surface defects on printed circuit boards (PCBs). However, most previous studies have focused only on the utilization of defect samples with visible defects and underestimated the value of template samples without surface defects. In fact, template samples can provide sufficient prior information to identify defects and are not difficult to obtain in many manufacturing scenarios. Therefore, an adaptive feature reconstruction network (AFRNet) is proposed in this paper to utilize these two types of samples. Specifically, AFRNet consists of two main components: a Siamese encoder with shared parameters for extracting features from the input sample pair, and a symmetrical feature reconstruction module for adaptively fusing these extracted features. Similar image-level and feature-level fusion schemes, as well as spatial misalignment caused by unaligned sample pairs have been carefully studied. Extensive experiments on a real-world PCB surface-defect dataset confirm the effectiveness of the proposed method, demonstrating that it can significantly improve the segmentation performance of multiple baselines, such as DANet, PSPNet and DeepLabv3.
What problem does this paper attempt to address?