Intelligent fault diagnosis of multi-source cross-machine bearings based on center-weighted optimal transport and class-level alignment domain adaptation

Abstract

Abstract Most of the current domain adaptation research primarily focuses on the single-source or multi-source domain transfer constructed under different working conditions of the same machine. However, when faced with cross-machine tasks with significant domain discrepancies, forcing the direct feature alignment between source and target domain samples may lead to negative transfer, thereby reducing the model’s diagnostic performance. To overcome the above limitations, this paper proposes a multi-source deep transfer model based on center-weighted optimal transport (CWOT) and class-level alignment domain adaptation. Firstly, to enhance the representation capability of deep features, a multi-structure feature representation network is constructed to enrich the information capacity embedded within the deep features, thereby achieving better domain adaptation capabilities. Then, the local maximum mean discrepancy is introduced to fully exploit fine-grained information and discriminative features among different source domains, minimizing the distribution differences among the source domains to the greatest extent, thus capturing reliable and highly generalized multi-source domain invariant features. On this basis, a CWOT strategy is designed, which comprehensively considers the transport cost of intra-domain uncertainty and inter-domain correlation among samples, establishing a more effective transport between source and target domains, alleviating the problem of sample negative transfer, and improving the model’s cross-machine diagnostic performance. Finally, instance studies are conducted through multiple cross-machine transfer diagnostic tasks, demonstrating that the proposed method outperforms existing domain adaptation methods in terms of diagnostic accuracy and fault transfer capability. This research provides a reliable fault diagnosis method for detecting the health status of rotating machinery equipment, promoting the application of domain adaptation technology in practical industry.