Fault Diagnosis of Bearings with Small Sample Size Using Improved Capsule Network and Siamese Neural Network

This paper addresses the challenges of low accuracy and long transfer learning time in small-sample bearing fault diagnosis, which are often caused by limited samples, high noise levels, and poor feature extraction. We propose a method that combines an improved capsule network with a Siamese neural...

Full description

Saved in:
Bibliographic Details
Main Authors: Jarula Yasenjiang, Yang Xiao, Chao He, Luhui Lv, Wenhao Wang
Format: Article
Language:English
Published: MDPI AG 2024-12-01
Series:Sensors
Subjects:
Online Access:https://www.mdpi.com/1424-8220/25/1/92
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1841548938616766464
author Jarula Yasenjiang
Yang Xiao
Chao He
Luhui Lv
Wenhao Wang
author_facet Jarula Yasenjiang
Yang Xiao
Chao He
Luhui Lv
Wenhao Wang
author_sort Jarula Yasenjiang
collection DOAJ
description This paper addresses the challenges of low accuracy and long transfer learning time in small-sample bearing fault diagnosis, which are often caused by limited samples, high noise levels, and poor feature extraction. We propose a method that combines an improved capsule network with a Siamese neural network. Multi-view data partitioning is used to enrich data diversity, and Markov transformation converts one-dimensional vibration signals into two-dimensional images, enhancing the visualization of signal features. The dynamic routing mechanism of the capsule network effectively captures and integrates key fault features, improving the model’s feature representation and robustness. The Siamese network shares weights to optimize feature matching, while SKNet dynamically adjusts feature fusion to enhance generalization performance. By integrating the Siamese neural network with SKNet, we improve transfer efficiency, reduce the number of parameters, and lighten the model to reduce complexity and shorten transfer time. Experimental results demonstrate that this method can accurately identify faults under conditions of limited samples and high noise, thereby improving diagnostic accuracy and reducing transfer time.
format Article
id doaj-art-9f5ab99e37f74be8ac5a8e9879ddf464
institution Kabale University
issn 1424-8220
language English
publishDate 2024-12-01
publisher MDPI AG
record_format Article
series Sensors
spelling doaj-art-9f5ab99e37f74be8ac5a8e9879ddf4642025-01-10T13:20:51ZengMDPI AGSensors1424-82202024-12-012519210.3390/s25010092Fault Diagnosis of Bearings with Small Sample Size Using Improved Capsule Network and Siamese Neural NetworkJarula Yasenjiang0Yang Xiao1Chao He2Luhui Lv3Wenhao Wang4College of Intelligent Manufacturing and Industrial Modernization, Xinjiang University, Urumqi 830017, ChinaCollege of Intelligent Manufacturing and Industrial Modernization, Xinjiang University, Urumqi 830017, ChinaCollege of Intelligent Manufacturing and Industrial Modernization, Xinjiang University, Urumqi 830017, ChinaCollege of Intelligent Manufacturing and Industrial Modernization, Xinjiang University, Urumqi 830017, ChinaCollege of Intelligent Manufacturing and Industrial Modernization, Xinjiang University, Urumqi 830017, ChinaThis paper addresses the challenges of low accuracy and long transfer learning time in small-sample bearing fault diagnosis, which are often caused by limited samples, high noise levels, and poor feature extraction. We propose a method that combines an improved capsule network with a Siamese neural network. Multi-view data partitioning is used to enrich data diversity, and Markov transformation converts one-dimensional vibration signals into two-dimensional images, enhancing the visualization of signal features. The dynamic routing mechanism of the capsule network effectively captures and integrates key fault features, improving the model’s feature representation and robustness. The Siamese network shares weights to optimize feature matching, while SKNet dynamically adjusts feature fusion to enhance generalization performance. By integrating the Siamese neural network with SKNet, we improve transfer efficiency, reduce the number of parameters, and lighten the model to reduce complexity and shorten transfer time. Experimental results demonstrate that this method can accurately identify faults under conditions of limited samples and high noise, thereby improving diagnostic accuracy and reducing transfer time.https://www.mdpi.com/1424-8220/25/1/92small samplecapsule networkSiamese neural networkSKNet
spellingShingle Jarula Yasenjiang
Yang Xiao
Chao He
Luhui Lv
Wenhao Wang
Fault Diagnosis of Bearings with Small Sample Size Using Improved Capsule Network and Siamese Neural Network
Sensors
small sample
capsule network
Siamese neural network
SKNet
title Fault Diagnosis of Bearings with Small Sample Size Using Improved Capsule Network and Siamese Neural Network
title_full Fault Diagnosis of Bearings with Small Sample Size Using Improved Capsule Network and Siamese Neural Network
title_fullStr Fault Diagnosis of Bearings with Small Sample Size Using Improved Capsule Network and Siamese Neural Network
title_full_unstemmed Fault Diagnosis of Bearings with Small Sample Size Using Improved Capsule Network and Siamese Neural Network
title_short Fault Diagnosis of Bearings with Small Sample Size Using Improved Capsule Network and Siamese Neural Network
title_sort fault diagnosis of bearings with small sample size using improved capsule network and siamese neural network
topic small sample
capsule network
Siamese neural network
SKNet
url https://www.mdpi.com/1424-8220/25/1/92
work_keys_str_mv AT jarulayasenjiang faultdiagnosisofbearingswithsmallsamplesizeusingimprovedcapsulenetworkandsiameseneuralnetwork
AT yangxiao faultdiagnosisofbearingswithsmallsamplesizeusingimprovedcapsulenetworkandsiameseneuralnetwork
AT chaohe faultdiagnosisofbearingswithsmallsamplesizeusingimprovedcapsulenetworkandsiameseneuralnetwork
AT luhuilv faultdiagnosisofbearingswithsmallsamplesizeusingimprovedcapsulenetworkandsiameseneuralnetwork
AT wenhaowang faultdiagnosisofbearingswithsmallsamplesizeusingimprovedcapsulenetworkandsiameseneuralnetwork