Domain Knowledge-Guided Contrastive Learning Framework Based on Complementary Views for Fault Diagnosis With Limited Labeled Data
Yu Yao, Jian Feng, Yue Liu
Abstract
Intelligent fault diagnosis has attracted much attention in industrial processes. The difficulty of collecting fault samples and high price of labeling data, has led to a relative scarcity of labeled data for deep learning tasks in the field. To address this gap, we propose a domain knowledge-guided contrastive learning framework based on complementary data views for fault diagnosis with limited data. Seven data views of either time- or frequency-domains are introduced and designed first. Then, the framework extracts task-specific features by 1) considering complementary information provided by multiple data views to each other, and 2) embedding a domain knowledge-involved space as the guide for the learning process. The results on two bearing datasets show the proposed framework can produce diagnosis accuracies of 96.60% and 94.24% when just 5% of samples have labels. This study determines two pairs of complementary data views that can boost the performance of the proposed framework.