Manifold-Contrastive Broad Learning System for Wheelset Bearing Fault Diagnosis

Abstract

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.