DC-NNMN: Across Components Fault Diagnosis Based on Deep Few-Shot Learning

In recent years, deep learning has become a popular topic in the intelligent fault diagnosis of industrial equipment. In practical working conditions, how to realize intelligent fault diagnosis in the case of the different mechanical components with a tiny labeled sample is a challenging problem. Th...

Full description

Saved in:
Bibliographic Details
Main Authors: Juan Xu, Pengfei Xu, Zhenchun Wei, Xu Ding, Lei Shi
Format: Article
Language:English
Published: Wiley 2020-01-01
Series:Shock and Vibration
Online Access:http://dx.doi.org/10.1155/2020/3152174
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1832561172243546112
author Juan Xu
Pengfei Xu
Zhenchun Wei
Xu Ding
Lei Shi
author_facet Juan Xu
Pengfei Xu
Zhenchun Wei
Xu Ding
Lei Shi
author_sort Juan Xu
collection DOAJ
description In recent years, deep learning has become a popular topic in the intelligent fault diagnosis of industrial equipment. In practical working conditions, how to realize intelligent fault diagnosis in the case of the different mechanical components with a tiny labeled sample is a challenging problem. That means training with one component sample but testing with another component sample has not been resolved. In this paper, we propose a deep convolutional nearest neighbor matching network (DC-NNMN) based on few-shot learning. The 1D convolution embedding network is constructed to extract the high-dimensional fault feature. The cosine distance is merged into the K-Nearest Neighbor method to model the distance distribution between the unlabeled sample from the query set and labeled sample from the support set in high-dimensional fault features. The multiple few-shot learning fault diagnosis tasks as the testing dataset are constructed, and then the network parameters are optimized through training in multiple tasks. Thus, a robust network model is obtained to classify the unknown fault categories in different components with tiny labeled fault samples. We use the CWRU bearing vibration dataset, the bearing vibration data selected from the Lab-built experimental platform, and another gearing vibration dataset for across components experiment to prove the proposed method. Experimental results show that the proposed method can achieve fault diagnosis accuracy of 82.19% for gearing and 82.63% for bearings with only one sample of each fault category. The proposed DC-NNMN model provides a new approach to solve the across components fault diagnosis in few-shot learning.
format Article
id doaj-art-b7f6401d4b934d1cbf5814dd0c7333c0
institution Kabale University
issn 1070-9622
1875-9203
language English
publishDate 2020-01-01
publisher Wiley
record_format Article
series Shock and Vibration
spelling doaj-art-b7f6401d4b934d1cbf5814dd0c7333c02025-02-03T01:25:46ZengWileyShock and Vibration1070-96221875-92032020-01-01202010.1155/2020/31521743152174DC-NNMN: Across Components Fault Diagnosis Based on Deep Few-Shot LearningJuan Xu0Pengfei Xu1Zhenchun Wei2Xu Ding3Lei Shi4School of Computer and Information Science, Hefei University of Technology, Hefei 230009, ChinaSchool of Computer and Information Science, Hefei University of Technology, Hefei 230009, ChinaSchool of Computer and Information Science, Hefei University of Technology, Hefei 230009, ChinaInstitute of Industry and Equipment Technology, Hefei University of Technology, Hefei 230009, ChinaSchool of Computer and Information Science, Hefei University of Technology, Hefei 230009, ChinaIn recent years, deep learning has become a popular topic in the intelligent fault diagnosis of industrial equipment. In practical working conditions, how to realize intelligent fault diagnosis in the case of the different mechanical components with a tiny labeled sample is a challenging problem. That means training with one component sample but testing with another component sample has not been resolved. In this paper, we propose a deep convolutional nearest neighbor matching network (DC-NNMN) based on few-shot learning. The 1D convolution embedding network is constructed to extract the high-dimensional fault feature. The cosine distance is merged into the K-Nearest Neighbor method to model the distance distribution between the unlabeled sample from the query set and labeled sample from the support set in high-dimensional fault features. The multiple few-shot learning fault diagnosis tasks as the testing dataset are constructed, and then the network parameters are optimized through training in multiple tasks. Thus, a robust network model is obtained to classify the unknown fault categories in different components with tiny labeled fault samples. We use the CWRU bearing vibration dataset, the bearing vibration data selected from the Lab-built experimental platform, and another gearing vibration dataset for across components experiment to prove the proposed method. Experimental results show that the proposed method can achieve fault diagnosis accuracy of 82.19% for gearing and 82.63% for bearings with only one sample of each fault category. The proposed DC-NNMN model provides a new approach to solve the across components fault diagnosis in few-shot learning.http://dx.doi.org/10.1155/2020/3152174
spellingShingle Juan Xu
Pengfei Xu
Zhenchun Wei
Xu Ding
Lei Shi
DC-NNMN: Across Components Fault Diagnosis Based on Deep Few-Shot Learning
Shock and Vibration
title DC-NNMN: Across Components Fault Diagnosis Based on Deep Few-Shot Learning
title_full DC-NNMN: Across Components Fault Diagnosis Based on Deep Few-Shot Learning
title_fullStr DC-NNMN: Across Components Fault Diagnosis Based on Deep Few-Shot Learning
title_full_unstemmed DC-NNMN: Across Components Fault Diagnosis Based on Deep Few-Shot Learning
title_short DC-NNMN: Across Components Fault Diagnosis Based on Deep Few-Shot Learning
title_sort dc nnmn across components fault diagnosis based on deep few shot learning
url http://dx.doi.org/10.1155/2020/3152174
work_keys_str_mv AT juanxu dcnnmnacrosscomponentsfaultdiagnosisbasedondeepfewshotlearning
AT pengfeixu dcnnmnacrosscomponentsfaultdiagnosisbasedondeepfewshotlearning
AT zhenchunwei dcnnmnacrosscomponentsfaultdiagnosisbasedondeepfewshotlearning
AT xuding dcnnmnacrosscomponentsfaultdiagnosisbasedondeepfewshotlearning
AT leishi dcnnmnacrosscomponentsfaultdiagnosisbasedondeepfewshotlearning