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

An imbalanced small sample slab defect recognition method based on image generation

Research on steel surface defect classification method based on deep learning

SDD-YOLO: A Lightweight, High-Generalization Methodology for Real-Time Detection of Strip Surface Defects

A deep-learning-based approach for fast and robust steel surface defects classification

Dual-Path GAN: A Method for Enhancing Small-scale Defect Detection on Metal Images

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

Steel Surface Defect Classification Based on Small Sample Learning

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

Intelligent Small Sample Defect Detection of Concrete Surface Using Novel Deep Learning Integrating Improved YOLOv5

Steel Strip Surface Defect Detection Method Based on Improved YOLOv5s

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

A High-Quality Sample Generation Method for Improving Steel Surface Defect Inspection

Steel Surface Defect Detection Based on SSAM-YOLO

TL-SDD: A Transfer Learning-Based Method for Surface Defect Detection with Few Samples

A small sample nonstandard gear surface defect detection method

Defect Identification of 316L Stainless Steel in Selective Laser Melting Process Based on Deep Learning

A Lightweight Strip Steel Surface Defect Detection Network Based on Improved YOLOv8

YOLOv5s‐GC‐Based Surface Defect Detection Method of Strip Steel

Steel Surface Defect Detection Using Improved Deep Learning Algorithm: ECA-SimSPPF-SIoU-Yolov5

Strip steel surface defect detection based on lightweight YOLOv5

Surface Defect Detection of Steel Plate Based on SKS-YOLO