A Coarse-to-Fine Bilevel Adversarial Domain Adaptation Method for Fault Diagnosis of Rolling Bearings

Zhao-Hua Liu,Liang Chen,Hua-Liang Wei,Ying Zhang,Lei Chen,Ming-Yang Lv
DOI: https://doi.org/10.1109/tim.2022.3214624
IF: 5.6
2022-11-02
IEEE Transactions on Instrumentation and Measurement
Abstract:Recently, domain adaptation (DA) has been used to solve the fault diagnosis problem in rolling bearings. However, most of the existing methods only align the distribution of domains and ignore the fine-grained information of the same fault categories in different domains, which leads to the degradation of diagnostic performance. To address such domain difference issues, this article proposes a novel coarse-to-fine bilevel adversarial DA approach (C2FADA) for bearings fault diagnosis. First, a sparse autoencoder (SAE) is used to extract features from raw data (containing both the source and target domains), and a Kullback–Leibler (KL) divergence term is then introduced to measure the discrepancy between the features from the source domain and the target domain. Second, a bilevel adversarial module is established to gradually align different domains at the domain level (with a coarse-grained model) and the class level (with a fine-grained approach) to tackle the domain shift issue and enable the classifier to learn the domain invariant representation features. Third, a spectral norm regularization constraint term is introduced to improve the stability of the adversarial training process by mitigating the effect of adversarial perturbations. The results show that the classification performance of the proposed C2FADA method is better than the compared existing peer methods.
engineering, electrical & electronic,instruments & instrumentation
What problem does this paper attempt to address?