SFDA-T: A novel source-free domain adaptation method with strong generalization ability for fault diagnosis

被引:2
|
作者
Wang, Jie [1 ]
Shao, Haidong [1 ]
Xiao, Yiming [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
关键词
Source-free domain adaptation; Transformer; Generalization ability; Fault diagnosis;
D O I
10.1016/j.aei.2024.102903
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Currently, source free domain adaptation (SFDA) methods are employed to address the issue of inaccessible source domain data (SDD) in transfer learning. However, existing SFDA methods often suffer from overfitting to specific domains, leading to poor generalization ability in the target domain. To address these challenges, this paper proposes a novel SFDA method named SFDA-T for fault diagnosis. Specifically, a Transformer-CNN-based feature extractor is constructed, to mine the transferable feature knowledge of faults in the SDD. The approach reduces the overfitting of the model to domain-specific information and improves model's generalization ability. In addition, the feature attention loss is designed to calculate attention weights of the sample features to increase the model's attention to the crucial feature regions in the target domain. A source similarity guided exponential loss is developed to guide target samples based on the decision boundaries of the source domain, facilitating cluster alignment of target sample categories and expanding distances between different categories. Furthermore, a self-training pseudo-labeling constraint is employed to reduce the effect of incorrect label matching and further constrain the model. The results of the experiments on gearboxes and bearings indicate that the proposed method achieves high fault diagnosis accuracy while effectively decoupling from SDD.
引用
收藏
页数:14
相关论文
共 50 条
  • [41] A domain adaptation method for bearing fault diagnosis using multiple incomplete source data
    Qibin Wang
    Yuanbing Xu
    Shengkang Yang
    Jiantao Chang
    Jingang Zhang
    Xianguang Kong
    Journal of Intelligent Manufacturing, 2024, 35 : 777 - 791
  • [42] A domain adaptation method for bearing fault diagnosis using multiple incomplete source data
    Wang, Qibin
    Xu, Yuanbing
    Yang, Shengkang
    Chang, Jiantao
    Zhang, Jingang
    Kong, Xianguang
    JOURNAL OF INTELLIGENT MANUFACTURING, 2024, 35 (02) : 777 - 791
  • [43] A multi-source ensemble domain adaptation method for rotary machine fault diagnosis
    Yang, Shengkang
    Kong, Xianguang
    Wang, Qibin
    Li, Zhongquan
    Cheng, Han
    Yu, Linyang
    MEASUREMENT, 2021, 186
  • [44] Reply to Comment on 'A novel transfer learning bearing fault diagnosis method based on multiple-source domain adaptation'
    Zhao, Jing
    Yang, Shaopu
    Li, Qiang
    Liu, Yongqiang
    Wang, Jiujian
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2022, 33 (09)
  • [45] Federated adversarial domain generalization network: A novel machinery fault diagnosis method with data privacy
    Wang, Rui
    Huang, Weiguo
    Shi, Mingkuan
    Wang, Jun
    Shen, Changqing
    Zhu, Zhongkui
    KNOWLEDGE-BASED SYSTEMS, 2022, 256
  • [46] A novel bearing fault diagnosis method based joint attention adversarial domain adaptation
    Chen, Pengfei
    Zhao, Rongzhen
    He, Tianjing
    Wei, Kongyuan
    Yuan, Jianhui
    RELIABILITY ENGINEERING & SYSTEM SAFETY, 2023, 237
  • [47] A novel bearing fault diagnosis method with feature selection and manifold embedded domain adaptation
    Yang, Songyu
    Zheng, Xiaoxia
    PROCEEDINGS OF THE INSTITUTION OF MECHANICAL ENGINEERS PART C-JOURNAL OF MECHANICAL ENGINEERING SCIENCE, 2022, 236 (14) : 8185 - 8197
  • [48] A centroid contrastive multi-source domain adaptation method for fault diagnosis with category shift
    Li, Yufeng
    Xu, Xinghan
    Hu, Lei
    Sun, Kai
    Han, Min
    MEASUREMENT, 2025, 248
  • [49] A New Multiple Source Domain Adaptation Fault Diagnosis Method Between Different Rotating Machines
    Zhu, Jun
    Chen, Nan
    Shen, Changqing
    IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2021, 17 (07) : 4788 - 4797
  • [50] Advancing UWF-SLO Vessel Segmentation with Source-Free Active Domain Adaptation and a Novel Multi-center Dataset
    Wang, Hongqiu
    Luo, Xiangde
    Chen, Wu
    Tang, Qingqing
    Xin, Mei
    Wang, Qiong
    Zhu, Lei
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT IX, 2024, 15009 : 75 - 85