Counterfactual Faithful Data Generation Based on Disentangled Representation for Compound Fault Diagnosis of Rolling Bearings

Abstract

Recently, deep learning and human-out-of-the-loop methods enjoy their prosperous applications in mechanical fault diagnosis. Nonetheless, the None-IID(independent and identically distributed) issue radicated in acquired data severely limits the stability and accuracy of compound fault diagnosis of rolling bearings. This paper proposes a sample augmentation method for generating simulated signals based on the concept of counterfactuals. Firstly, disentangled representations and counterfactual faithful theory are applied to classify the original signal into two categories of properties. One is the fault semantics encoded from the original vibration signal. And the other is the sample attribute encoded by the encoder of Variational Autoencoders (VAEs). Secondly, the counterfactual faithful pseudo-samples are conjured through the Generative Adversarial Network(GAN) using the seeds of the “factual” sample attributes and “counterfactual” fault semantics to compensate for the drawback of distribution shift. Finally, the original samples and pseudo-samples are used as the CNN classifier dataset to realize bearing fault diagnosis. Experiments show that this method can generate counterfactual signals that are highly consistent with the original data distribution and can achieve better classification accuracy after balancing the dataset.