Two-head classifier guided domain adversarial learning for universal domain adaptation in intelligent fault diagnosis

被引:1
|
作者
Zhang, Jiyang [1 ]
Wang, Xiangxiang [1 ]
Su, Zhiheng [1 ]
Lian, Penglong [1 ]
Xu, Hongbing [1 ]
Zou, Jianxiao [1 ,2 ]
Fan, Shicai [1 ,2 ]
机构
[1] Univ Elect Sci & Technol China, Sch Automat Engn, Chengdu 611731, Sichuan, Peoples R China
[2] Univ Elect Sci & Technol China, Shenzhen Inst Adv Study, Shenzhen 518110, Guangdong, Peoples R China
关键词
Intelligent fault diagnosis; Cross-domain fault diagnosis; Transfer learning; Universal domain adaptation; Two-head classifier; Domain adversarial learning; NETWORK;
D O I
10.1016/j.ress.2024.110708
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Domain adaptation (DA) methods have been widely used in cross-domain fault diagnosis to mitigate the distribution discrepancy between data from different working conditions. However, traditional DA methods are designed for specific one known category shift between domains. When prior knowledge about relationships between source and target label sets is unknown, the applicability of these methods is limited. To address this issue, a universal domain adaptation method named two-head classifier guided domain adversarial learning (THC-DAN) is proposed, which can handle all category shift scenarios in DA, including closed-set, partial-set, open-set, and open-partial-set. Specifically, we develop a domain adversarial network with an elegantly designed two-head classifier and adapt it to target domain. During adaptation, we first introduce an informative consistency score based on the two-head classifier to distinguish target private samples. Then, the consistency separation loss is proposed to push these samples away from classification boundaries. Finally, to realize the safe alignment on common classes between domains, the weighted adversarial learning based on the two-head classifier's prediction probability is presented to weaken effects of source private samples. Experiments under all DA scenarios on datasets from Case Western Reserve University, Paderborn University, and our own Drivetrain Prognostics Simulator demonstrate the effectiveness of THC-DAN.
引用
收藏
页数:18
相关论文
共 50 条
  • [31] Internal adversarial guided unsupervised multi-domain adaptation network for collaborative fault diagnosis of bearing
    Shao H.
    Chen X.
    Cao H.
    Jiang H.
    Zhongguo Kexue Jishu Kexue/Scientia Sinica Technologica, 2023, 53 (07): : 1229 - 1240
  • [32] Improved Conditional Domain Adversarial Networks for Intelligent Transfer Fault Diagnosis
    Qin, Haihua
    Pan, Jiafang
    Li, Jian
    Huang, Faguo
    MATHEMATICS, 2024, 12 (03)
  • [33] Cross-Domain Fault Diagnosis of Rolling Bearings Using Domain Adaptation with Classifier Discrepancy
    Zhang Y.-C.
    Li Q.
    Ren Z.-H.
    Zhou S.-H.
    Dongbei Daxue Xuebao/Journal of Northeastern University, 2021, 42 (03): : 367 - 372
  • [34] A fault diagnosis method for rolling bearings in open-set domain adaptation with adversarial learning
    Lei, Tongfei
    Pan, Feng
    Hu, Jiabei
    He, Xu
    Li, Bing
    SCIENTIFIC REPORTS, 2025, 15 (01):
  • [35] Two-dimensional adversarial domain adaptation graph contrastive learning for fault diagnosis of limited similar batch process
    Gao, Xingke
    Zhu, Jinlin
    Gao, Furong
    Zhang, Zheng
    PROCESS SAFETY AND ENVIRONMENTAL PROTECTION, 2025, 197
  • [36] Domain Adaptation with Multilayer Adversarial Learning for Fault Diagnosis of Gearbox under Multiple Operating Conditions
    Zhang, Ming
    Lu, Weining
    Yang, Jun
    Wang, Duo
    Bin, Liang
    2019 PROGNOSTICS AND SYSTEM HEALTH MANAGEMENT CONFERENCE (PHM-QINGDAO), 2019,
  • [37] Intelligent fault diagnosis via unsupervised domain adaptation: The role of intermediate domain construction
    Cao, Peng
    Yang, Jun
    Jia, Jinyin
    Chen, Junfan
    Fan, Anfei
    KNOWLEDGE-BASED SYSTEMS, 2025, 310
  • [38] Partial adversarial domain adaptation by dual-domain alignment for fault diagnosis of rotating machines
    Wang, Xuan
    She, Bo
    Shi, Zhangsong
    Sun, Shiyan
    Qin, Fenqi
    ISA TRANSACTIONS, 2023, 136 : 455 - 467
  • [39] Bearing fault diagnosis based on partial domain adaptation adversarial network
    Zhou, Huafeng
    Cheng, Peiyuan
    Shao, Siyu
    Zhao, Yuwei
    Yang, Xinyu
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2022, 33 (12)
  • [40] Conditional Adversarial Domain Adaptation With Discrimination Embedding for Locomotive Fault Diagnosis
    Yu, Xiaolei
    Zhao, Zhibin
    Zhang, Xingwu
    Sun, Chuang
    Gong, Baogui
    Yan, Ruqiang
    Chen, Xuefeng
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2021, 70