Multi-Scale Adaptive-Routing Capsule Contrastive Network-Based Intelligent Fault Diagnosis Method for Rotating Machinery under Noisy Environment and Labels

Yucheng Xiong,Zhiwen Liu,Jiyong Tan,Liang Hao
DOI: https://doi.org/10.1016/j.aei.2024.102712
IF: 8.8
2024-01-01
Advanced Engineering Informatics
Abstract:The robustness and accuracy of fault diagnosis are affected by noisy environment and labels from manual annotation errors in vibration signal data collected during the operation of mechanical equipment. Therefore, this study develops a multi-scale adaptive-routing capsule contrastive network-based intelligent fault diagnosis method for rotating machinery under noisy environment and labels. Firstly, multi-scale residual blocks (MSRB) are introduced into the capsule network to utilize receptive fields of different scales, comprehensively capturing multi-scale feature information. Meanwhile, the structure of the capsule network is further enhanced by employing a non-iterative adaptive-routing algorithm, preventing the amplification of irrelevant features due to the sparsification of coupling coefficients and improving the accuracy of the classification model. Secondly, a supervised contrastive learning module is added to enhance the model’s discriminative ability for multi-scale features, improving the model’s representation capabilities. Finally, the proposed method achieves “end-to-end” intelligent recognition of multiple faults by optimizing the model parameters via the established loss function and input data. Experimental results demonstrate that the proposed method significantly reduces the impact of noisy environment and labels, achieving high-precision classification. Compared to other classification methods, the proposed method has a clear advantage in noise resistance and stronger robustness.
What problem does this paper attempt to address?