Rolling bearing intelligent fault diagnosis towards variable speed and imbalanced samples using multiscale dynamic supervised contrast learning

Yutong Dong, Hongkai Jiang, Renhe Yao, Mingzhe Mu and Qiao Yang

Reliability Engineering and System Safety, 2024, vol. 243, issue C

Abstract: Deep learning-based fault diagnosis methods have already attained remarkable achievements in this field. However, rolling bearing frequently operates under variable speed conditions, and the number of healthy samples collected is often significantly larger than that of failure samples. In this paper, a multiscale dynamic supervised contrast learning (MDSupCon) framework is proposed. First, a multiscale adaptive feature extraction network is designed as the backbone, which utilizes multiple convolutional kernels to enhance feature extraction capabilities under variable speed conditions, and the branch attention mechanism is incorporated to adaptively adjust the weights of various scale branches. Second, the joint channel-space attention mechanism is constructed to enhance the importance of critical features while reducing redundant information, thereby improving fault identification accuracy and interpretability. Third, the dynamic supervised contrast loss function is designed to assign dynamic compensation factors to samples of various categories according to the training results, which reduces the impact of easily classified samples and enhances the contribution of hard-to-classify samples in imbalanced scenarios. Additionally, a dynamic cross-entropy loss is designed to train the backbone and the classifiers. The MDSupCon has achieved superior results of 89.49% and 92.15% on two bearing datasets with an imbalance ratio of 20:1 and variable speeds.

Keywords: Fault diagnosis; variable speed and imbalanced samples; multiscale adaptive feature extraction network; joint channel-space attention mechanism; dynamic supervised contrast learning (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (6)

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832023007196
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:reensy:v:243:y:2024:i:c:s0951832023007196

DOI: 10.1016/j.ress.2023.109805

Access Statistics for this article

Reliability Engineering and System Safety is currently edited by Carlos Guedes Soares

More articles in Reliability Engineering and System Safety from Elsevier
Bibliographic data for series maintained by Catherine Liu ().