Joint distribution Bures–Wasserstein distance based multi-source student teacher network for rotating machinery fault diagnosis

Abstract

Current research in across working conditions transfer fault diagnosis predominantly relies on single source domain adaptation, neglecting the extensive and diverse diagnostic data available from multiple domains in real-world applications. Furthermore, the joint distribution between fault features and classes is often overlooked in existing multi-source studies, resulting in model failures under varying operational conditions. To address these challenges, a novel multi-source domain diagnostic framework is proposed, leveraging optimal transport theory within a student-teacher learning network. Firstly, the joint distribution Bures–Wasserstein distance is formulated based on the second-order statistic cross-covariance operator, which explicitly models the mapping between fault features and fault labels while also constraining the distribution across different domains. Secondly, a student-teacher network is constructed, with the joint distribution Bures–Wasserstein distance successfully embedded to mitigate distributional discrepancies between domains, while a high-confidence pseudo-labeling strategy is devised to minimize the negative transferability of diagnostic knowledge. The effectiveness of the proposed method is validated using the parallel shaft gearbox and the bearing datasets, the results show that the proposed method has high diagnostic accuracy and robustness.