A Weighted Ensemble Domain Generalization Method Via Multi-Level Invariant Feature Learning for Rotating Machinery Fault Diagnosis

Zhiheng Su,Jiyang Zhang,Penglong Lian,Jianxiao Zou,Shicai Fan
DOI: https://doi.org/10.1145/3685073.3685081
2024-01-01
Abstract:Domain generalization aims to address the problem of model transfer to the unknown target domain. However, existing domain generalization based fault diagnosis methods still suffer from insufficient exploration of domain-invariant features and lack consideration of the correlation between target data and different source data when solving the multi-source domain generalization problem. To address these issues, a weighted ensemble domain generalization (WEDG) method via multi-level invariant feature learning is proposed in this paper. By leveraging domain-level domain-invariant feature learning based on higher-order moment matching to adapt the overall data distribution of different source domains, sample-level domain-invariant feature learning is introduced to deeply learn domain-invariant features of different class data across domains through random cross-domain sample mixing. Additionally, considering the different similarities between target data and different source domains, the proposed method separately trains extra classifiers for each source domain and integrates the prediction results of all classifiers based on the similarity between target data and source domains to enhance the final generalization ability of the model. Experimental results on the CWRU and Gearbox datasets demonstrate that the WEDG outperforms other methods on domain generalization fault diagnosis problem in both bearings and gears.
What problem does this paper attempt to address?