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 条
  • [21] A novel source-free domain adaptation network for intelligent diagnosis of bearings under unknown faults
    Jia, Feng
    Xu, Xiang
    Wang, Yuanfei
    Shen, Jianjun
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2025, 36 (04)
  • [22] A robust source-free unsupervised domain adaptation method based on uncertainty measure and adaptive calibration for rotating machinery fault diagnosis
    Lin, Yanzhuo
    Wang, Yu
    Zhang, Mingquan
    Zhao, Ming
    RELIABILITY ENGINEERING & SYSTEM SAFETY, 2025, 253
  • [23] Semi-supervised source-free domain adaptation method via diffusive label propagation for rotating machinery fault diagnosis
    Su, Zhiheng
    Lian, Penglong
    Shang, Penghui
    Zhang, Jiyang
    Xu, Hongbing
    Zou, Jianxiao
    Fan, Shicai
    RELIABILITY ENGINEERING & SYSTEM SAFETY, 2024, 252
  • [24] Asymmetric Source-Free Unsupervised Domain Adaptation for Medical Image Diagnosis
    Zhang, Yajie
    Huang, Zhi-An
    Wu, Jibin
    Tan, Kay Chen
    2024 IEEE CONFERENCE ON ARTIFICIAL INTELLIGENCE, CAI 2024, 2024, : 234 - 239
  • [25] A Source-Free Unsupervised Domain Adaptation Method Based on Feature Consistency
    Lee, JoonHo
    Lee, Gyemin
    INTERNATIONAL WORKSHOP ON ADVANCED IMAGING TECHNOLOGY, IWAIT 2023, 2023, 12592
  • [26] UC-SFDA: Source-free domain adaptation via uncertainty prediction and evidence-based contrastive learning
    Chen, Dong
    Zhu, Hongqing
    Yang, Suyi
    KNOWLEDGE-BASED SYSTEMS, 2023, 275
  • [27] Source-Free Cluster Adaptation for Privacy-Preserving Machinery Fault Diagnosis
    Zhu, Mengliang
    Zeng, Xiangyu
    Liu, Jie
    Yang, Chaoying
    Zhou, Kaibo
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2023, 72
  • [28] Generation, augmentation, and alignment: a pseudo-source domain based method for source-free domain adaptation
    Yuntao Du
    Haiyang Yang
    Mingcai Chen
    Hongtao Luo
    Juan Jiang
    Yi Xin
    Chongjun Wang
    Machine Learning, 2024, 113 : 3611 - 3631
  • [29] Generation, augmentation, and alignment: a pseudo-source domain based method for source-free domain adaptation
    Du, Yuntao
    Yang, Haiyang
    Chen, Mingcai
    Luo, Hongtao
    Jiang, Juan
    Xin, Yi
    Wang, Chongjun
    MACHINE LEARNING, 2024, 113 (06) : 3611 - 3631
  • [30] Generation, division and training: A promising method for source-free unsupervised domain adaptation
    Tian, Qing
    Zhao, Mengna
    NEURAL NETWORKS, 2024, 172