Combining the theoretical bound and deep adversarial network for machinery open-set diagnosis transfer

被引:77
作者
Deng, Yafei [1 ,2 ]
Lv, Jun [3 ]
Huang, Delin [4 ]
Du, Shichang [1 ,2 ]
机构
[1] Shanghai Jiao Tong Univ, Sch Mech Engn, State Key Lab Mech Syst & Vibrat, 800 Dongchuan Rd, Shanghai 200240, Peoples R China
[2] Shanghai Jiao Tong Univ, Sch Mech Engn, Dept Ind Engn & Management, 800 Dongchuan Rd, Shanghai 200240, Peoples R China
[3] East China Normal Univ, Fac Econ & Management, Shanghai 200241, Peoples R China
[4] Shanghai Polytech Univ, Coll Intelligent Mfg & Control Engn, Shanghai, Peoples R China
基金
中国国家自然科学基金;
关键词
Intelligent fault diagnosis; Deep transfer learning; Open-set domain adaptation; Adversarial learning; INTELLIGENT FAULT-DIAGNOSIS;
D O I
10.1016/j.neucom.2023.126391
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recently, deep transfer learning-based intelligent machine diagnosis has been well investigated, and the source and the target domain are commonly assumed to share the same fault categories, which can be called as the closed-set diagnosis transfer (CSDT). However, this assumption is hard to cover real engi-neering scenarios because some unknown new fault may occur unexpectedly due to the uncertainty and complexity of machinery components, which is called as the open-set diagnosis transfer (OSDT). To solve this challenging but more realistic problem, a Theory-guided Progressive Transfer Learning Network (TPTLN) is proposed in this paper. First, the upper bound of transfer learning model under open-set setting is thoroughly analyzed, which provides a theoretical insight to guide the model opti-mization. Second, a two-stage module is designed to carry out distracting unknown target samples and attracting known samples through progressive learning, which could effectively promote inter-class separability and intra-class compactness. The performance of proposed TPTLN is evaluated in two OSDT cases, where the diagnosis knowledge is transferred across bearings and gearbox running under different working conditions. Comparative results show that the proposed method achieves better robustness and diagnostic performance under different degrees of domain shift and openness variance. The source codes and links to the data can be found in the following GitHub repository: https://github.-com/phoenixdyf/Theory-guided-Progressive-Transfer-LearningNetwork.& COPY; 2023 Elsevier B.V. All rights reserved.
引用
收藏
页数:22
相关论文
共 29 条
[1]  
[Anonymous], BEAR DATACENTER
[2]   Open Set Domain Adaptation [J].
Busto, Pau Panareda ;
Gall, Juergen .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :754-763
[3]   Wasserstein distance based deep adversarial transfer learning for intelligent fault diagnosis with unlabeled or insufficient labeled data [J].
Cheng, Cheng ;
Zhou, Beitong ;
Ma, Guijun ;
Wu, Dongrui ;
Yuan, Ye .
NEUROCOMPUTING, 2020, 409 (409) :35-45
[4]   A double-layer attention based adversarial network for partial transfer learning in machinery fault diagnosis [J].
Deng, Yafei ;
Huang, Delin ;
Du, Shichang ;
Li, Guilong ;
Zhao, Chen ;
Lv, Jun .
COMPUTERS IN INDUSTRY, 2021, 127
[5]   Learning to Detect Open Classes for Universal Domain Adaptation [J].
Fu, Bo ;
Cao, Zhangjie ;
Long, Mingsheng ;
Wang, Jianmin .
COMPUTER VISION - ECCV 2020, PT XV, 2020, 12360 :567-583
[6]   Generative Adversarial Networks [J].
Goodfellow, Ian ;
Pouget-Abadie, Jean ;
Mirza, Mehdi ;
Xu, Bing ;
Warde-Farley, David ;
Ozair, Sherjil ;
Courville, Aaron ;
Bengio, Yoshua .
COMMUNICATIONS OF THE ACM, 2020, 63 (11) :139-144
[7]   Joint distribution adaptation with diverse feature aggregation: A new transfer learning framework for bearing diagnosis across different machines [J].
Jia, Shiyao ;
Deng, Yafei ;
Lv, Jun ;
Du, Shichang ;
Xie, Zhiyuan .
MEASUREMENT, 2022, 187
[8]   A systematic review of deep transfer learning for machinery fault diagnosis [J].
Li, Chuan ;
Zhang, Shaohui ;
Qin, Yi ;
Estupinan, Edgar .
NEUROCOMPUTING, 2020, 407 :121-135
[9]   Domain Consensus Clustering for Universal Domain Adaptation [J].
Li, Guangrui ;
Kang, Guoliang ;
Zhu, Yi ;
Wei, Yunchao ;
Yang, Yi .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :9752-9761
[10]   A Deep Adversarial Transfer Learning Network for Machinery Emerging Fault Detection [J].
Li, Jipu ;
Huang, Ruyi ;
He, Guolin ;
Wang, Shuhua ;
Li, Guanghui ;
Li, Weihua .
IEEE SENSORS JOURNAL, 2020, 20 (15) :8413-8422