A Hierarchical Training-Convolutional Neural Network for Imbalanced Fault Diagnosis in Complex Equipment

Yiping Gao,Liang Gao,Xinyu Li,Siyu Cao
DOI: https://doi.org/10.1109/tii.2022.3177662
IF: 12.3
2022-09-21
IEEE Transactions on Industrial Informatics
Abstract:Complex equipment is important in industry and fault diagnosis is a key technology for maintenance. Convolutional neural network (CNN) is a common manner for fault diagnosis. But due to the imbalanced data, it cannot be applied to complex equipment directly. Generally, CNNs require balanced data, and imbalanced data will mislead the models and ignore actual faults. In complex equipment, since fault occurs rarely, the fault data are imbalanced. This impedes the application of CNNs greatly. To overcome this problem, a hierarchical training-CNN is proposed in this article. The proposed method uses an effective number-resampling to balance fault data, which avoids invalid samples, and introduces a magnet-loss pretraining to address the overlap of features between different faults. Based on these improvements, the proposed method achieves good diagnosis performances with an average accuracy of 96.56%, and has been developed into a real-world case successfully with an accuracy of 94.28%.
automation & control systems,computer science, interdisciplinary applications,engineering, industrial
What problem does this paper attempt to address?