Abstract:Visual surface defect inspection provides an important tool for product quality assessment in a wide range of industrial applications. In recent years, diverse convolutional neural network (CNN) models have been devel-oped for the high-accuracy image classification of natural objects. However, it remains a challenging task to develop deep-learning-based approaches for surface defect inspection of industrial products. The main challenge is that surface defects of industrial products occur in low probability, therefore it is impractical to build a large image dataset containing different types of defects to train/fine-tune deep CNN models to achieve satisfactory performance when applied to industrial inspection tasks. To overcome the above-mentioned problem, we build a light-weight defect classification model based on the pre-trained SqueezeNet architecture and present three effective techniques to achieve high-accuracy defect classification using very few collected defective training samples. First, we experimentally demonstrate that it is reasonable to freeze shallower convolutional layers and only fine-tune deeper layers in the last convolutional stage when there are not enough training samples. Second, we integrate batch-size independent Group Normalization (GN) into SqueezeNet to compute stable statistics based on a limited number of defective samples for training the classification model. Third, we add an auxiliary task to perform the similarity comparison of two defective samples during the training phase, significantly increasing the size of the training data and generating more distinctive defect-specific representations. The experimental results on the NEU-CLS dataset and USB-FS dataset verify the effectiveness of the proposed method. Our method can obtain accurate defect classification results when very few defective training samples are provided, i.e., about 97.69% and 82.92% average classification accuracy on the NEU-CLS and USB-FS datasets, respectively, under the 5-shot setting (5 images per defect category).

When Wafer Failure Pattern Classification Meets Few-shot Learning and Self-Supervised Learning

Auto-Labeling for Pattern Recognition of Wafer Defect Maps in Semiconductor Manufacturing

Wafer Defect Pattern Labeling and Recognition Using Semi-Supervised Learning

Unsupervised Learning for Wafer Surface Defect Pattern Recognition

Machine Learning Based Wafer Defect Detection

Semi-Supervised Framework for Wafer Defect Pattern Recognition with Enhanced Labeling.

An Effective Industrial Defect Classification Method under the Few-Shot Setting Via Two-Stream Training

Mixed-Type Wafer Failure Pattern Recognition

Few-Shot Classification of Screen Defects with Class-Agnostic Mask and Context-Based Classifier.

Enhancing Defect Diagnosis and Localization in Wafer Map Testing Through Weakly Supervised Learning

Wafer Map Defect Patterns Semi-Supervised Classification Using Latent Vector Representation

Coarse-to-Fine Few-Shot Defect Recognition with Dynamic Weighting and Joint Metric

Advances in machine learning and deep learning applications towards wafer map defect recognition and classification: a review

Inspection and Classification of Semiconductor Wafer Surface Defects Using CNN Deep Learning Networks

Few-shot Defect Recognition of Metal Surfaces Via Attention-Embedding and Self-Supervised Learning

Observational and Experimental Insights into Machine Learning-Based Defect Classification in Wafers

Machine Learning-Based Detection Method for Wafer Test Induced Defects

Few-Shot Classification of Wafer Bin Maps Using Transfer Learning and Ensemble Learning

An Efficient Deep Learning Framework for Mixed-Type Wafer Map Defect Pattern Recognition

Utilizing the Mean Teacher with Supcontrast Loss for Wafer Pattern Recognition

Wafer Defect Localization and Classification Using Deep Learning Techniques