Dilated dynamic supervised contrastive learning framework for fault diagnosis under imbalanced dataset conditions

Rui Yang,Shi Jia,Houliang Wang,Yinglong Yan,Pengfei Zhang,Zenghui An
DOI: https://doi.org/10.1177/09544062241300062
2024-12-09
Proceedings of the Institution of Mechanical Engineers Part C Journal of Mechanical Engineering Science
Abstract:Proceedings of the Institution of Mechanical Engineers, Part C: Journal of Mechanical Engineering Science, Ahead of Print. Deep learning-based fault diagnosis methods have achieved significant results. However, under actual working conditions, the inability of equipment to operate for long periods in a fault state results in a much smaller amount of collected fault data compared to normal data. This leads to existing fault diagnosis models being able to learn only limited diagnostic knowledge, causing a substantial decline in diagnostic accuracy. Therefore, a dilated dynamic supervised contrast learning framework for imbalanced data scenarios is proposed. In this framework, a dilated convolution network is designed as a feature extractor to enhance the feature extraction ability by changing the dilation rate. Then, the dynamic supervision loss function is designed, and the sample compensation factor of each category is given according to the frequency of sample used in the training process to enhance the contribution of the minority class and optimize the feature extractor. Finally, a dynamic cross-entropy loss function is designed to train the network and classifier to achieve accurate classification of faults. Experiments on two open-source datasets show the effectiveness of the proposed model framework.
engineering, mechanical
What problem does this paper attempt to address?