基于无源域自适应的滚动轴承故障诊断策略研究

Source Free Domain Adaptation Strategy for Rolling Bearing Fault Diagnosis

摘要: 本文提出了一种基于无源域自适应的滚动轴承故障诊断策略，旨在解决源域数据缺失时的领域自适应问题。该策略通过源域模型的预训练与目标域数据的伪标签生成与过滤相结合，实现了无源域条件下的有效领域适应。具体地，利用源域模型对目标域数据进行伪标签生成，通过信息熵计算和阈值过滤筛选出可靠的故障原型，进而计算每个目标域样本与故障原型的余弦相似度，从而生成噪声较低的伪标签。实验结果表明，本文方法在多个迁移任务中均表现出优异的性能，平均准确率达95.03%，尤其在处理源域和目标域数据分布差异较大的情况下，能够有效提高故障诊断的准确率。与传统的深度学习和领域自适应方法相比，本文方法在无源域自适应故障诊断中具有显著优势。

Abstract: This paper proposes a source free domain adaptation strategy for rolling bearing fault diagnosis, aiming to address the domain adaptation problem when source domain data is missing. The method combines pretraining a source domain model with pseudo-label generation and filtering for target domain data to achieve effective domain adaptation under source-free conditions. Specifically, the source domain model is used to generate pseudo-labels for the target domain data. Reliable fault prototypes are selected through information entropy calculation and threshold filtering. Then, the cosine similarity between each target domain sample and the fault prototypes is computed to generate pseudo-labels with reduced noise. Experimental results demonstrate that the proposed method performs excellently across multiple transfer tasks, with an average accuracy of 95.03%. In particular, it significantly improves fault diagnosis accuracy when dealing with large distribution differences between source and target domain data. Compared with traditional deep learning and domain adaptation methods, the proposed method shows significant advantages in source free domain adaptation for fault diagnosis.