1. A novel data augmentation approach to fault diagnosis with class-imbalance problem.
- Author
-
Tian, Jilun, Jiang, Yuchen, Zhang, Jiusi, Luo, Hao, and Yin, Shen
- Abstract
Data-driven fault diagnosis techniques are frequently applied to ensure the reliability and safety of industrial systems. However, as a common challenge, the class-imbalance problem reduces the performance of data-driven methods due to the lack of data information. We propose a weighted modified conditional variational auto-encoder (WM-CVAE) as a novel data augmentation technique to tackle the issue. The modified structure can alleviate the existing Kullback–Leibler (KL) divergence vanishing by an adaptive loss. Meanwhile, kernel mean matching (KMM) is proposed on weight computation to reduce the negative effect of dissimilar generated samples. Constructing the WM-CVAE data augmentation framework can effectively improve the data quality and learning capability in class-imbalance fault diagnosis. To validate the proposed WM-CVAE model, three real-world industrial datasets are used as study objects, and the random forest is used as the base learner in the fault classification tasks. The diagnostic results demonstrate that the proposed WM-CVAE data augmentation framework can improve learning results in class-imbalance fault diagnosis. • A general adaptive loss-based learning algorithm to solve the KL divergence vanishing problem widely existing in the original CVAE generators. • An adaptive weight case is proposed as an example to be analyzed and validated on real-world class-imbalance fault diagnosis datasets. • Kernel mean matching is applied to realize weight assignment by metric learning. • Research results show the WM-CVAE data augmentation approach is excellent for class-imbalance fault diagnosis tasks. [ABSTRACT FROM AUTHOR]
- Published
- 2024
- Full Text
- View/download PDF