摘要详情

ID / 提交时间

14 / 2024-08-07 10:46:25

标题

An Improved Causal Disentanglement Single-Source Domain Generalization Method for Bearing Fault Diagnosis

关键字

causal disentanglement; single-source domain generalization; bearing; fault diagnosis; triplet loss

主题及专题

lAI Applications in Engineering

状态

全文录用

作者

WangHongqi / Harbin University of Science and Technology；Heilongjiang Province Key Laboratory of Pattern Recognition and Information Perception

WangYujing / Harbin University of Science and Technology；Heilongjiang Province Key Laboratory of Pattern Recognition and Information Perception

KangShouqiang / Harbin University of Science and Technology；Heilongjiang Province Key Laboratory of Pattern Recognition and Information Perception

WangQingyan / Harbin University of Science and Technology；Heilongjiang Province Key Laboratory of Pattern Recognition and Information Perception

摘要

In the field of domain generalization for fault diagnosis, most methods concentrate on extracting domain-invariant features from multi-source domains. However, collecting samples from multi-source domains is extremely difficult, and the data typically originate from a single-source domain. To tackle the issue of inadequate generalization capability in unknown target domains when trained on only a single source domain, an improved prototype-guided causal disentanglement domain generalization network (ICDDG) is proposed for mechanical fault diagnosis. This network combines feature mean, similarity, and triplet loss to construct an improved prototype-based triplet loss function, which reduces the influence of outlier samples and achieves more effective prototype learning. The improved triplet loss function effectively guides the causal disentanglement network to separate causal features from non-causal features better, enhancing the model's adaptability and robustness when encountering unseen domains. Diagnostic experiments performed using two bearing datasets substantiate the efficacy of the ICDDG method.