Feature selection and interpretability analysis of compound faults in rolling bearings based on the causal feature weighted network
Chongchong Yu, Mengxiong Li, Zongning Wu, K. Gao, Fei Wang
Abstract
Abstract Feature selection is a crucial step in fault diagnosis. When rolling bearings are susceptible to compound faults, causal relationships are hidden within the signal features. Complex network analysis methods provide a tool for causal relationship modeling and feature importance assessment. Existing studies mainly focus on unweighted networks, overlooking the impact of the strength of causal relationships on feature selection. To address this issue, we propose a compound fault feature selection method based on the causal feature weighted network. First, we construct a weighted network using the incremental association Markov blanket discovery algorithm and Pearson correlation coefficient. Then, we quantify the importance of features by treating node strength as a centrality index and rank them to partition the feature subset. Finally, the optimal feature subset is obtained through a neural network with the accuracy of compound fault diagnosis as the threshold. Analysis of public datasets and comparative experiments demonstrate the advantages of our method. Compared to existing research, our method not only effectively reduces the number of optimal feature subsets to 11 but also improves the accuracy of compound fault diagnosis to 95.2%. Furthermore, we employ the SHapley Additive exPlanations to interpret the contribution of each feature in the optimal subset to the accuracy of compound fault diagnosis. This provides reference from both physical and network perspectives to feature selection and compound fault diagnosis in rolling bearings in practical working conditions.