A novel unsupervised deep transfer learning method based on contrast pre-training for fault diagnosis

Lingli Jiang,Jianyu Long,Yunwei Huang,Zhe Yang,Chuan Li,Jianwen Guo,Jungang Cao
DOI: https://doi.org/10.1109/ONCON60463.2023.10431100
2023-12-08
Abstract:Unsupervised transfer learning has become a research hotspot in the field of mechanical equipment fault diagnosis. It typically assumes a large amount of labeled data is available for constructing a diagnosis model in the source domain, and aims at applying the model in the target domain by adjusting marginal and conditional data distributions using unlabeled data of the target domain. However, the labeling capability is often extremely limited for large data, and automatically balancing the confrontation of adjusting both marginal and conditional distribution is difficult. To this end, this paper proposes a deep transfer adversarial learning method based on contrast pre-training (DTALC). A fault diagnosis model can be contrasted under the few-label constraint in source domain, where a feature extractor is pre-trained unsupervised by contrast coding, based on which a satisfactory classifier can be built using a few labeled data. And an adversarial factor is designed to automatically adapt the weights between the global discriminator and the local discriminator to release the marginal and conditional distribution adaptation problem of unsupervised transfer learning. The proposed DTALC is verified in On publicly available datasets and its diagnosis accuracy outperforms other state-of-the-art methods considering multiple combinations of source and target domains
Computer Science,Engineering
What problem does this paper attempt to address?