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).

Few-shot incremental learning in the context of solar cell quality inspection

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

Solar Cells Surface Defects Detection Based on Deep Learning

Small-scale defect detection in industrial environment based on lightweight deep learning network

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

A lightweight network for photovoltaic cell defect detection in electroluminescence images based on neural architecture search and knowledge distillation

Few-shot Industrial Defect Image Classification Based on Lightweight Model with Attention Mechanism

Few-shot defect detection using feature enhancement and image generation for manufacturing quality inspection

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

A New Knowledge Distillation Network for Incremental Few-Shot Surface Defect Detection

Few-Shot Defect Segmentation Leveraging Abundant Normal Training Samples Through Normal Background Regularization and Crop-and-Paste Operation

AnomalySeg: Deep Learning-Based Fast Anomaly Segmentation Approach for Surface Defect Detection

Anomaly detection and automatic labeling for solar cell quality inspection based on Generative Adversarial Network

Classification and Fast Few-Shot Learning of Steel Surface Defects with Randomized Network

Deep-Learning-Based Automatic Detection of Photovoltaic Cell Defects in Electroluminescence Images

Deep Learning-Based Solar-Cell Manufacturing Defect Detection With Complementary Attention Network

CellDefectNet: A Machine-designed Attention Condenser Network for Electroluminescence-based Photovoltaic Cell Defect Inspection

Quality Control of Carbon Look Components via Surface Defect Classification with Deep Neural Networks

Deep learning‐based automatic detection of multitype defects in photovoltaic modules and application in real production line

Solar photovoltaic module defect detection based on deep learning

Improving Deep Learning-based Defect Detection on Window Frames with Image Processing Strategies