.png){kind=icon}
返回
Collaborative filter pruning for efficient automatic surface defect detection
.png){kind=icon}
PRE
AI
DOI：10.1109/TII.2023.3288208.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
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.
Keyword:
Defect detection
convolutional neural network (CNN)
model compression
network pruning
collaborative pruning
期刊
IF:
9.9
论文数:
8.6K
被引数:
6.0W
机构
引用论文
分享收藏An End-to-End Steel Surface Classification Approach Based on EDCGAN and MobileNet V2.png)
SENSORS
OAAI
SENSORSIF3.5
OAAI分享收藏An End-to-End Steel Surface Defect Detection Approach via Fusing Multiple Hierarchical Features融合多个层次特征的端到端钢表面缺陷检测方法OAAI
OAAI分享收藏Fan Fault Diagnosis Based on Lightweight Multiscale Multiattention Feature Fusion Network基于轻量级多尺度多注意力特征融合网络的风机故障诊断
PREAI
分享收藏Deep Learning-Based Solar-Cell Manufacturing Defect Detection With Complementary Attention Network
PREAI
分享收藏HDCB-Net: A Neural Network With the Hybrid Dilated Convolution for Pixel-Level Crack Detection on Concrete BridgesHdcb-net: 具有混合扩张卷积的神经网络，用于混凝土桥梁的像素级裂缝检测
PREAI
分享收藏
查看更多内容