A novel sub-label learning mechanism for enhanced cross-domain fault diagnosis of rotating machinery

Abstract

Deep Domain Adaptation (DDA), which transfers the knowledge learned in the source domain to the target domain, has made remarkable achievements in intelligent fault diagnosis. However, the existing DDA technology mainly focuses on eliminating cross-domain distribution discrepancies, while ignoring the exploration of intra-domain distribution characteristics, resulting in unsatisfactory performance in complex scenarios. To overcome this drawback, a novel sub-label learning mechanism (SLLM) is proposed in this paper, which exploits the structural connectivity of the original sample space to guide distribution alignment, thereby enhancing domain adaptability. Specifically, SLLM consists of two parts. First, the unsupervised target domain is annotated with sub-labels according to the probability distribution of the sample space, so that similar data can be recognized. Then, the intra-domain connectivity of the associated data is preserved during feature matching. In this way, samples belonging to the same category can be aggregated together in the feature space, and mismatches can be effectively alleviated. Extensive experiments on two datasets indicate that the proposed SLLM can significantly improve the domain adaptability of traditional DDA methods.