arXiv:2503.08749cs.LGcs.AI2025-03被引 2

不依赖源数据,利用可靠与不可靠伪标签提升轴承故障诊断跨域性能

Source-free domain adaptation based on label reliability for cross-domain bearing fault diagnosis

  • 通过数据增强投票区分目标样本的伪标签可靠性
  • 用可靠伪标签作为真实标签建模特征与标签关系,提升判别力
  • 最大化不可靠伪标签熵,缓解负迁移,适合无源数据场景

无源域适应(SFDA)被用于跨域轴承故障诊断,无需访问源数据。现有方法仅选取具有可靠伪标签的目标样本进行模型适配,忽略了其他样本。本文认为每个目标样本都可贡献于模型适配,提出一种新方法,同时利用可靠与不可靠伪标签。设计基于数据增强的标签投票策略,将目标样本分为可靠与不可靠两类。利用可靠伪标签作为真实标签,挖掘特征空间与标签空间的内在关联;同时通过最大化不可靠伪标签的熵,缓解负迁移。所提方法在判别力与多样性间取得良好平衡。在两个轴承故障基准数据集上开展大量实验,结果表明,该方法显著优于现有SFDA方法。代码已开源:https://github.com/BdLab405/SDALR。

原文摘要 · Abstract (English)

Source-free domain adaptation (SFDA) has been exploited for cross-domain bearing fault diagnosis without access to source data. Current methods select partial target samples with reliable pseudo-labels for model adaptation, which is sub-optimal due to the ignored target samples. We argue that every target sample can contribute to model adaptation, and accordingly propose in this paper a novel SFDA-based approach for bearing fault diagnosis that exploits both reliable and unreliable pseudo-labels. We develop a data-augmentation-based label voting strategy to divide the target samples into reliable and unreliable ones. We propose to explore the underlying relation between feature space and label space by using the reliable pseudo-labels as ground-truth labels, meanwhile, alleviating negative transfer by maximizing the entropy of the unreliable pseudo-labels. The proposed method achieves well-balance between discriminability and diversity by taking advantage of reliable and unreliable pseudo-labels. Extensive experiments are conducted on two bearing fault benchmarks, demonstrating that our approach achieves significant performance improvements against existing SFDA-based bearing fault diagnosis methods. Our code is available at https://github.com/BdLab405/SDALR.

域适应故障诊断伪标签轴承

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。