With the widespread application of deep learning in industrial surface defect detection, YOLO-series object detection models have become mainstream due to their compact structure and high inference efficiency. However, deployment on embedded devices is limited by computational resources. Therefore, effective model compression is required for lightweight deployment. In this study, a joint compression strategy combining channel pruning and feature distillation is proposed for the YOLOv7-Tiny model. Specifically, the Group_slim method is first applied to perform channel-level structural compression, which removes redundant channels and reduces model complexity. Then, the original uncompressed model is used as the teacher network. The student network is guided using the Channel-Wise Distillation (CWD) method to learn key feature representations after pruning. Extensive experiments on the MVTecAD-Screw industrial defect dataset show that the proposed strategy reduces model parameters and computational cost while maintaining high accuracy. A good balance between inference speed and precision is achieved. Compared with other single pruning or knowledge distillation methods, the Group_slim + CWD scheme demonstrates superior overall performance across multiple evaluation metrics.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

An Industrial Inspection Model Compression Framework Combining Group Slimming and Channel-Wise Distillation

  • Xinyu Zhang,
  • Bei Wang

摘要

With the widespread application of deep learning in industrial surface defect detection, YOLO-series object detection models have become mainstream due to their compact structure and high inference efficiency. However, deployment on embedded devices is limited by computational resources. Therefore, effective model compression is required for lightweight deployment. In this study, a joint compression strategy combining channel pruning and feature distillation is proposed for the YOLOv7-Tiny model. Specifically, the Group_slim method is first applied to perform channel-level structural compression, which removes redundant channels and reduces model complexity. Then, the original uncompressed model is used as the teacher network. The student network is guided using the Channel-Wise Distillation (CWD) method to learn key feature representations after pruning. Extensive experiments on the MVTecAD-Screw industrial defect dataset show that the proposed strategy reduces model parameters and computational cost while maintaining high accuracy. A good balance between inference speed and precision is achieved. Compared with other single pruning or knowledge distillation methods, the Group_slim + CWD scheme demonstrates superior overall performance across multiple evaluation metrics.