Relation Awareness Network for Few-Shot Fine-Grained Fault Diagnosis

Few-shot fine-grained fault diagnosis aims at identifying faults at a fine-grained level with limited training samples, which is challenged by subtle category differences inherent in fine-grained fault diagnosis. To address this limitation, we introduce a relation awareness network (RAN) to explore multilevel semantic relationships in time-frequency images. RAN integrates two novel strategies: contextual relation learning (CRL) and relation collaboration optimization (RCO). CRL strategy analyzes internal variations within samples and the correlations among different samples, effectively capturing the unique characteristics of each sample but also identifying the patterns shared across various samples. Concurrently, the RCO strategy enhances the model's ability to discriminate between classes by optimizing the interclass relationships. Experiments on two public and one lab-built bearing dataset demonstrate RAN's effectiveness. Quantitative results show that compared with existing methods, RAN achieves a diagnosis accuracy improvement of up to 2.27% and 1.01% on the Paderborn University (PU) bearing dataset in the ten-way one-shot and ten-way five-shot settings, respectively.