Research and Application of Semi-Supervised Category Dictionary Model Based on Transfer Learning

Abstract

Classification tasks are pivotal across diverse applications, yet the burgeoning amount of data, coupled with complicating factors such as noise, exacerbates the challenge of classifying complex data. For algorithms that require a large amount of data, the annotation work for datasets is also exceptionally complex and tedious. Drawing upon existing research, this paper first introduces a novel semi-supervised category dictionary model based on transfer learning (SSDT). This model is designed to construct a more representative category dictionary and to delineate the associations of information across different domains, utilizing the lens of conditional probability distribution. This approach is particularly apt for semi-supervised transfer learning scenarios. Subsequently, the proposed method is applied to the domain of bearing fault diagnosis. This model is suitable for transfer scenarios; moreover, its semi-supervised characteristic eliminates the need for labeling the entire input dataset, significantly reducing manual workload. Experimental results attest to the model’s practical utility. When benchmarked against other 6 models, the SSDT model demonstrates enhanced generalization performance.