Attention Features Selection Oversampling Technique (AFS-O) for Rolling Bearing Fault Diagnosis with Class Imbalance

Zhongze Han,Haoran Wang,Chen Shen,Xuewei Song,Longchao Cao,Lianqing Yu
DOI: https://doi.org/10.1088/1361-6501/ad0e9d
IF: 2.398
2023-11-21
Measurement Science and Technology
Abstract:Abstract In the field of data-driven fault diagnosis of mechanical rotating parts, such as gears and bearings, the presence of class imbalance in life cycle data collected by sensors poses a significant challenge. While Synthetic Minority Over Sampling Technique (SMOTE) is a common method to address class imbalance, it has been found to generate synthetic samples with redundant features due to its blind sampling in the feature space. To overcome this limitation, an Attention Features Selection Oversampling Technique (AFS-O) is proposed in this paper. The approach utilizes Attention Features Selection (AFS) to eliminate redundant features and assign appropriate weights to features before performing oversampling. Experimental results indicate that the AFS-O method outperforms other fusion algorithms, indicating its effectiveness in improving the classification performance of classifiers used for fault diagnosis of mechanical rotating parts.
engineering, multidisciplinary,instruments & instrumentation
What problem does this paper attempt to address?