.png){kind=icon}
Return
Collaborative filter pruning for efficient automatic surface defect detection
.png){kind=icon}
PRE
AI
DOI：10.1109/TII.2023.3288208.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Surface defect detection is a critical task in industrial production, and numerous methods have been proposed to achieve high detection accuracy. Although deep learning-based approaches have achieved state-of-the-art performances, their vast computational cost and high memory footprint prevent their deployment in resourceconstrained environments. To address this problem, we propose a collaborative filter pruning method for the defect detection model, which significantly reduces the number of required calculations and parameters while maintaining high performance, even in cases with tasks suffering from the class imbalance problem. Our method aims to obtain lightweight pruned models by removing unimportant filters according to their importance evaluated by both structural similarity and detail richness of corresponding feature maps. Moreover, to improve the performance of pruned models, we propose a knowledge-fused fine-tuning approach that fuses the knowledge derived from two teacher networks to look after both representation learning and classifier learning, alleviating the class imbalance problem. Experimental results on four public datasets demonstrate that the proposed approach performs favorably relative to the state-of-the-art methods. In particular, the proposed method achieves 39x and 59x parameter compression for VGG-16 and ResNet-50, respectively, on the NEU-CLS dataset, with a very small detection accuracy loss (< 0.2%). The code is available at https://github.com/HXuanWang/CFP.
Keywords:
Defect detection
convolutional neural network (CNN)
model compression
network pruning
collaborative pruning
Journal
IF:
9.9
Papers:
8.6K
Citations:
6.0W
Organization
Cited Papers
ShareSaveAn End-to-End Steel Surface Classification Approach Based on EDCGAN and MobileNet V2.png)
SENSORS
OAAI
SENSORSIF3.5
OAAIShareSaveShareSaveShareSaveDeep Learning-Based Solar-Cell Manufacturing Defect Detection With Complementary Attention Network
PREAI
ShareSaveShareSaveShareSave
View more