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

被引:3
作者
Liu, Zhengwu [1 ]
Zhong, Xiang [1 ]
Shao, Haidong [1 ]
Yan, Shen [1 ]
Liu, Bin [2 ]
机构
[1] Hunan Univ, Coll Mech & Vehicle Engn, Changsha 410082, Peoples R China
[2] Univ Strathclyde, Dept Management Sci, Glasgow G1 1XQ, Scotland
基金
中国国家自然科学基金;
关键词
Fault diagnosis; Graph neural networks; Multi-source domain adaptation; Graph feature alignment; Transfer learning;
D O I
10.1016/j.knosys.2024.112606
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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.
引用
收藏
页数:15
相关论文
共 32 条
[1]   Self-learning transferable neural network for intelligent fault diagnosis of rotating machinery with unlabeled and imbalanced data [J].
An, Zenghui ;
Jiang, Xingxing ;
Cao, Jing ;
Yang, Rui ;
Li, Xuegang .
KNOWLEDGE-BASED SYSTEMS, 2021, 230
[2]   Data Augmentation and Intelligent Fault Diagnosis of Planetary Gearbox Using ILoFGAN Under Extremely Limited Samples [J].
Chen, Mingzhi ;
Shao, Haidong ;
Dou, Haoxuan ;
Li, Wei ;
Liu, Bin .
IEEE TRANSACTIONS ON RELIABILITY, 2023, 72 (03) :1029-1037
[3]   Collaborative fault diagnosis of rotating machinery via dual adversarial guided unsupervised multi-domain adaptation network [J].
Chen, Xingkai ;
Shao, Haidong ;
Xiao, Yiming ;
Yan, Shen ;
Cai, Baoping ;
Liu, Bin .
MECHANICAL SYSTEMS AND SIGNAL PROCESSING, 2023, 198
[4]   Compound fault diagnosis of diesel engines by combining generative adversarial networks and transfer learning [J].
Cui, Zhiquan ;
Lu, Yanlin ;
Yan, Xu ;
Cui, Shuya .
EXPERT SYSTEMS WITH APPLICATIONS, 2024, 251
[5]  
Defferrard M, 2016, ADV NEUR IN, V29
[6]  
Ganin Y, 2016, J MACH LEARN RES, V17
[7]  
Gretton A, 2012, J MACH LEARN RES, V13, P723
[8]   Domain adaptation with label-aligned sampling (DALAS) for cross-domain fault diagnosis of rotating machinery under class imbalance [J].
Lee, Jinwook ;
Ko, Jin Uk ;
Kim, Taehun ;
Kim, Yong Chae ;
Ha Jung, Joon ;
Youn, Byeng D. .
EXPERT SYSTEMS WITH APPLICATIONS, 2024, 243
[9]   Attention-based deep meta-transfer learning for few-shot fine-grained fault diagnosis [J].
Li, Chuanjiang ;
Li, Shaobo ;
Wang, Huan ;
Gu, Fengshou ;
Ball, Andrew D. .
KNOWLEDGE-BASED SYSTEMS, 2023, 264
[10]   The emerging graph neural networks for intelligent fault diagnostics and prognostics: A guideline and a benchmark study [J].
Li, Tianfu ;
Zhou, Zheng ;
Li, Sinan ;
Sun, Chuang ;
Yan, Rucliang ;
Chen, Xuefeng .
MECHANICAL SYSTEMS AND SIGNAL PROCESSING, 2022, 168