Federated temporal-context contrastive learning for fault diagnosis using multiple datasets with insufficient labels

Haowen Zheng,Hui Liu,Zhenyu Liu,Jianrong Tan
DOI: https://doi.org/10.1016/j.aei.2024.102432
IF: 8.8
2024-04-01
Advanced Engineering Informatics
Abstract:Insufficient data, lack of labelled data, and limited data sharing hinder deep-learning-based fault-diagnosis methods. Most of the existing methods focus on addressing only one of these issues and consequently lack practical applicability. The method proposed in this study addresses these issues simultaneously. First, a temporal-context contrastive learning method is proposed that combines the concepts of few-shot and self-supervised learning. This method assists the feature extractor in learning fault data representations from unlabelled data through a specially designed loss function, thereby enabling the training of deep-learning models on small-scale, label-deficient datasets. Next, a federated learning framework is introduced to train a global model from multiple datasets without data sharing. In addition, a novel client contrastive loss function is proposed to address the issue of model performance degradation caused by different distributions among client datasets. Finally, experimental evaluations are conducted on public datasets, and the results demonstrate the effectiveness and superiority of the proposed method.
engineering, multidisciplinary,computer science, artificial intelligence
What problem does this paper attempt to address?