A Weakly Supervised Learning-Based Oversampling Framework for Class-Imbalanced Fault Diagnosis

Abstract

With the lack of failure data, class imbalance has become a common challenge in the fault diagnosis of industrial systems. The oversampling methods can tackle the class-imbalanced problem by generating the minority samples to balance the training set. However, one of the main challenges of the existing oversampling methods is how to generate high-quality minority samples. Traditional oversampling methods regard all synthetic samples as minority ones to be added to the training set without filtering. The low-quality synthetic samples would distort the distribution of the dataset and worsen the classification performance. In this article, we propose a weakly supervised oversampling method that treats all synthetic samples as unlabeled samples and develops a graph semisupervised learning algorithm to select high-quality synthetic samples, adding into the final training set as minority samples. To improve the quality of synthetic samples, we propose a cost-sensitive neighborhood component analysis dimensionality reduction method to enhance domain information validity in high-dimensional datasets. Finally, combining a boosting-based ensemble framework, we propose a new imbalanced learning framework suitable for high dimensionality and highly imbalanced fault diagnosis in industrial systems. The experimental validation is performed on five real-world wind turbine blade cracking failure datasets and compared to 15 benchmark methods. The experimental results show that average performances and robustness of the proposed framework are significantly better than those of the benchmark methods.