Abstract:Surface defect detection is crucial in the industrial domain, but deploying deep learning models on edge devices faces challenges of memory constraints and real-time requirements. Therefore, model pruning is necessary to reduce the model’s parameter count and improve inference speed. Due to the high detection accuracy demanded in industrial defect detection, it is essential to prune the network while ensuring minimal impact on network accuracy. Different industrial defect scenarios may exhibit defects with varying importance of features and scale representation. Hence, to preserve the network’s capability to recognize various defect features, pruning methods need to be adaptive to the dataset and effectively balance the remaining feature information across different scale layers in various scene contexts. To tackle these challenges, we propose a structured pruning method tailored for industrial detection scenarios, which significantly reduces model computation and parameter count while preserving network accuracy. Specifically, we design a scoring network to guide the allocation of pruning amounts for each layer. This network evaluates the importance of different scale layers globally based on the performance of various defect features within the dataset. Additionally, we employ the L1 norm criterion to prune filters with low weights that minimally affect accuracy. We also consider filter correlation and introduce a comprehensive evaluation method that incorporates both filter contribution and substitutability. This enables pruning of filters with low impact on detection accuracy. Finally, we obtain the most lightweight network through an iterative “prune-retraining” process. Our method is evaluated on four datasets using segmentation networks, achieving over 80% reduction in floating-point operations and 90% reduction in parameters while maintaining accuracy, thus significantly improving model memory usage and inference speed.

Surface Defect Segmentation Algorithm of Steel Plate Based on Geometric Median Filter Pruning

Depth feature fusion based surface defect region identification method for steel plate manufacturing

Adaptive Model Compression for Steel Plate Surface Defect Detection: An Expert Knowledge and Working Condition-Based Approach

Steel Surface Defects Detection Based on Deep Learning

Filter collaborative contribution pruning method based on the importance of different-scale layers for surface defect detection

Surface Defect Detection of Steel Plate Based on SKS-YOLO

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

A lightweight image-level segmentation method for steel surface defects based on cross-layer feature fusion

Steel Surface Defect Detection via Deformable Convolution and Background Suppression

A deep learning model for steel surface defect detection

A New Steel Defect Detection Algorithm Based on Deep Learning

Lightweight Network-Based Surface Defect Detection Method for Steel Plates

Strip steel surface defect detecting method combined with a multi-layer attention mechanism network

Deep Learning-Based Automated Steel Surface Defect Segmentation: a Comparative Experimental Study.

A steel surface defect inspection approach towards smart industrial monitoring

RDDPA: Real-time Defect Detection Via Pruning Algorithm on Steel Surface

Steel surface defect detection based on sparse global attention transformer

Steel Plate Surface Defect Detection Method - Based on Improved YOLO Algorithm

Steel Surface Defect Detection Method Based on Improved YOLOv8

DAssd-Net: A Lightweight Steel Surface Defect Detection Model Based on Multi-Branch Dilated Convolution Aggregation and Multi-Domain Perception Detection Head

Efficient Detection Model of Steel Strip Surface Defects Based on YOLO-V7