A novel hierarchical structural pruning-multiscale feature fusion residual network for intelligent fault diagnosis

Yiwei Cheng,Xinnuo Lin,Haiping Zhu,Jun Wu,Haibin Shi,Huafeng Ding
DOI: https://doi.org/10.1016/j.mechmachtheory.2023.105292
IF: 4.93
2023-06-01
Mechanism and Machine Theory
Abstract:Residual learning is a commonly used method in the intelligent fault diagnosis (IFD) field. The existing residual networks usually have a large model volume, whose training process puts forward high requirements for both time consumption and the computing resource configuration. This paper proposes a novel hierarchical structural pruning-multiscale feature fusion residual network (HSP-MFFRN) for IFD. The multiple multi-scale feature extraction modules and feature fusion modules are designed in the proposed HSP-MFFRN to extract, fuse and compress the multi-scale features without changing the size of the convolutional filter. In addition, hierarchical structural pruning is implemented for HSP-MFFRN to delete redundant channels. Two experimental study cases are conducted to verify model diagnosis performance, including a rolling bearing IFD case and an aerostat capsule IFD case. The experimental results show that the proposed model can effectively compress the learnable parameters and model volume after hierarchical structural pruning, and can still achieve superior diagnostic performance compared with other deep learning methods, residual learning methods, and other advanced methods reported in the literature.
engineering, mechanical
What problem does this paper attempt to address?