Manifold-Contrastive Broad Learning System for Wheelset Bearing Fault Diagnosis
Manifold-Contrastive Broad Learning System for Wheelset Bearing Fault Diagnosis
复制标题
DOI:
10.1109/tits.2023.3274256
复制
发表时间:
2023-09
影响因子:
8.5
通讯作者:
Ning Wang;L. Jia;Huiyue Zhang;Yong Qin;Xuejun Zhao;Zhipeng Wang
中科院分区:
文献类型:
--
作者:
Ning Wang;L. Jia;Huiyue Zhang;Yong Qin;Xuejun Zhao;Zhipeng Wang
Newly deployed trains have massive normal data and scarce faulty data for training, which limits the diagnosis accuracy with class imbalance problem of small samples. Considering that there are a lot unutilized information hidden in the abundant unlabeled monitoring data, this paper proposes a novel method named manifold-contrastive broad learning system, which utilizes the online updating approach for dealing with the class imbalance problem of small samples. This method constructs a novel one-class broad-learning classifier based on an inherency-guided comparison mechanism, which can classify and annotate unlabeled data online. This classifier employs contrastive manifold matrices to maintain the inherent structures, which is not affected to the overfitting caused by imbalanced samples. Secondly, inspired by the active learning, this classifier proposes the minimum-error strategy to annotate the samples by classifying the modes, which solves the problem of insufficient training data. Thirdly, this method applies an incremental learning strategy that continuously absorbs the newly annotated data to update the model online, which improves the model accuracy under the data imbalanced condition. Finally, the feasibility and effectiveness of the proposed method are verified by wheelset bearing data collected from a test rig of a Chinese rolling stock company.