Decision Self-Regulating Network for Imbalanced Working Conditions Identification in the Application of Gearbox Intelligent Fault Diagnosis
Minmin Xu,Yaoyao Han,Xiaoxi Ding,Haidong Shao,Yimin Shao
DOI: https://doi.org/10.1109/tim.2023.3292952
IF: 5.6
2023-01-01
IEEE Transactions on Instrumentation and Measurement
Abstract:Due to its powerful capability of extracting high dimensional nonlinear features, deep learning has proven to be an effective tool for gearbox fault diagnosis. It can recognize various faults through training sufficient samples. However, the available samples in practice are class imbalanced, such as various imbalanced fault data under different working conditions, which affects the update of weights and leads to low identification accuracy. To address this problem, this article proposes a network named decision self-regulating network (DSRN) to improve identification accuracy through self-regulation. The proposed DSRN consists of multiple classifier units, score units, and self-regulation units. Each classifier is established based on a convolutional neural network (CNN) to identify different faults under a known working condition, where the identification result is later output as training data. The scoring unit aims to acquire a comprehensive recognition result with the fused training data. The self-regulation unit further realizes the final identification which weighted the result of each classifier with the corresponding score unit in a self-regulation way. Especially, the sore unit will be updated to separate the influence of different working conditions during the training process. To verify the effectiveness of the proposed network, a gearbox test rig was designed, and datasets under different speeds and five health conditions were obtained. Experimental results show that owing to the effect of self-regulation, the proposed DSRN shows good performance on gearbox fault classification under both single-speed and multispeed class imbalance. In most cases, DSRN outperforms other conventional fault identification network models, including CNN, MK-ResCNN, WDCNN, and ensemble convnet (ECN).