Open Access Article

Title: One-shot transfer learning with limited data sample for bearing component fault diagnosis

Authors: Wei Ren Sia; Mohd Syahril Ramadhan Mohd Saufi; Muhammad Firdaus Bin Isham; Mohd Salman Leong

Addresses: Faculty of Mechanical Engineering, Universiti Teknologi Malaysia, 81310 UTM Johor Bahru, Johor, Malaysia; Faculty of Mechanical Engineering, Institute of Noise and Vibration, Universiti Teknologi Malaysia, 54100 Kuala Lumpur, Wilayah Persekutuan Kuala Lumpur, Malaysia ' Faculty of Mechanical Engineering, Universiti Teknologi Malaysia, 81310 UTM Johor Bahru, Johor, Malaysia; Faculty of Mechanical Engineering, Institute of Noise and Vibration, Universiti Teknologi Malaysia, 54100 Kuala Lumpur, Wilayah Persekutuan Kuala Lumpur, Malaysia ' Faculty of Mechanical Engineering, Universiti Teknologi Malaysia, 81310 UTM Johor Bahru, Johor, Malaysia; Faculty of Mechanical Engineering, Institute of Noise and Vibration, Universiti Teknologi Malaysia, 54100 Kuala Lumpur, Wilayah Persekutuan Kuala Lumpur, Malaysia ' Faculty of Mechanical Engineering, Universiti Teknologi Malaysia, 81310 UTM Johor Bahru, Johor, Malaysia; Faculty of Mechanical Engineering, Institute of Noise and Vibration, Universiti Teknologi Malaysia, 54100 Kuala Lumpur, Wilayah Persekutuan Kuala Lumpur, Malaysia

Abstract: Rolling element bearings are crucial components in the machine, so it is important to maintain the bearings' health. The classic deep learning model needs bulky quality data for the model to achieve high performance. However, it is difficult for the industry to obtain bulk-quality data due to machinery systems working in harsh conditions and the current one-shot learning model has limited capabilities in transfer learning. Thus, a one-shot learning with rhombus Siamese neural network (RSNN) is proposed for a small data size fault diagnosis. RSNN in this study focuses on a large number of classes with a small sample data size and transfer learning without pre-training the target data. The results proved that the one-shot RSNN has high prediction accuracy for the limited data fault detection and diagnosis (FDD) by achieving 90.63% performance with just four training data per class for the 64 classes for the CWRU dataset.

Keywords: one-shot learning; Siamese neural network; SNN; bearing component diagnosis; transfer learning; cross-domain analysis.

DOI: 10.1504/IJHM.2025.145708

International Journal of Hydromechatronics, 2025 Vol.8 No.6, pp.1 - 29

Received: 21 Nov 2024
Accepted: 26 Feb 2025

Published online: 15 Apr 2025 *