Transfer graph feature alignment guided multi-source domain adaptation network for machinery fault diagnosis

Tools

Liu, Zhengwu and Zhong, Xiang and Shao, Haidong and Yan, Shen and Liu, Bin (2024) Transfer graph feature alignment guided multi-source domain adaptation network for machinery fault diagnosis. Knowledge-Based Systems, 305. 112606. ISSN 0950-7051 (https://doi.org/10.1016/j.knosys.2024.112606)

[thumbnail of Liu-etal-KBS-2024-Transfer-graph-feature-alignment-guided-multi-source]
Preview
 Text. Filename: Liu-etal-KBS-2024-Transfer-graph-feature-alignment-guided-multi-source.pdf
Accepted Author Manuscript
License: Creative Commons Attribution 4.0 logo
Download (2MB)| Preview

Abstract

In recent years, the application of unsupervised multi-source domain adaptation (MSDA) techniques for fault diagnosis has gained significant traction. Current research typically overlooks or fails to effectively capture critical data structure information during feature extraction. Another challenge is optimising the integration of information from multiple source domains to diagnose the target domain while avoiding negative transfer. To address these challenges, this study proposes a transfer graph feature alignment guided multi-source domain adaptation network (MDTGAL). In the proposed method, a transfer graph sample generator module (GSG) is constructed to model the data structure between source and target domains, and multiple graph feature extractors are employed to learn the data structure information from different domain combinations. A regularisation technique is introduced to extract the domain-invariant features by aligning the parameters across multiple independent graph feature extractors. In addition, a weighted soft-voting mechanism based on the polynomial kernel-induced maximum mean difference metric (PK-MMD) is designed to fuse the outputs from multiple classifiers, to comprehensively account for the influence of each source domain. The proposed method was tested on multi-source domain transfer tasks involving various operating conditions of rotating machinery. The experimental results demonstrate that the MDTGAL exhibits superior cross-domain diagnostic performance, outperforming existing mainstream methods. In addition, this study explored the impact of varying numbers of source domains on the diagnostic accuracy of the target domain, providing insights into selecting the correct number of source domains for specific MSDA tasks.

ORCID iDs

Liu, Zhengwu, Zhong, Xiang, Shao, Haidong, Yan, Shen and Liu, Bin ORCID logoORCID: https://orcid.org/0000-0002-3946-8124;
CORE (COnnecting REpositories)