Self-attention Metric Learning Based on Multi-Scale Feature Fusion for Few-shot Fault Diagnosis

• Published in: IEEE sensors journal 2023-07, p.1-1
• Main Authors: Xie, Jingsong, Liu, Jie, Ding, Tianqi, Wang, Tiantian, Yu, Tianjian
• Format: Article
• Language: English
• Subjects: Data models; Deep learning; Fault diagnosis; Feature extraction; Few-shot fault diagnosis; Measurement; metric learning; multi-scale feature fusion; self-attention mechanism; Task analysis; Training
• ISSN: 1530-437X
• DOI: 10.1109/JSEN.2023.3296750

The intelligent diagnosis model based on deep learning method can effectively and accurately diagnose the health state of bearings widely used in various mechanical equipment. However, in engineering practice, it is difficult to obtain sufficient labeled fault data, which would reduce the diagnosis performance. In order to solve the above problem, this paper proposes a self-attention metric learning based on multi-scale feature fusion to classify the bearing fault with few shot. The proposed few-shot intelligent diagnosis model mainly contains the feature extraction module and metric learning module. First, the vibration data need to be combined in pairs to form sample pairs, which are input into the feature extraction module based on multi-scale feature fusion. This input approach can increase the learning tasks so as to alleviate the problem of insufficient training samples. Then, the extracted features are concatenated and input into the metric learning module whose learning ability is improved by adding the self-attention network, and the input features are eventually converted into the similarity. Finally, the fault type can be identified according to the maximum similarity value between the test sample and samples with different labels. Three experiment cases are conducted to validate the performance of proposed intelligent diagnosis model. The experimental results show that the proposed model can accurately classify the bearing fault with few-shot case. Additionally, the comparison experiments with the traditional models have also validated the advantage of the proposed intelligent diagnosis model for few-shot case.