SMD Anomaly Detection: A Self-Supervised Texture–Structure Anomaly Detection Framework
Jiaxiang Luo, Junbin Lin, Zhiyu Yang, Haiming Liu
Abstract
In electronic manufacturing, anomaly detection of surface mount devices (SMDs) through computer vision is an important task to control the production quality of SMDs. The difficulty of the detection is that some anomalous regions on the surfaces of SMDs are very minor and with variable shapes, which leads to poor detection efficiency. To solve this problem, based on the assumption that normal samples can be reconstructed more accurately than anomalous samples, a self-supervised image anomaly detection framework with a multi-scale two-branch feature fusion strategy is proposed. Specifically, it adopts autoencoder as the basic framework, and to enhance the reconstruction error between input anomalous samples and the reconstructed ones, a self-supervised learning task of reconstructing images is introduced to have the model neglect the encoding of the suspected anomalous regions found by a contextual attention mask module. Meanwhile, a multi-scale feature fusion strategy is developed to fuse texture and structure features in the decoder to reconstruct samples. Moreover, a multi-level anomalous score criterion is proposed to enlarge the scores for the samples with very minor anomalies. At last, an SMD-Capacitor anomaly detection dataset (SMDC-DET) is built to evaluate the proposed method. The experiments show that the proposed method achieves an average AUC accuracy of 98.82%, much better when compared to the start-of-art existing anomaly detection methods.