Log In Sign Up

A Wavelet Transform and self-supervised learning-based framework for bearing fault diagnosis with limited labeled data

by   Yuhong Jin, et al.

Traditional supervised bearing fault diagnosis methods rely on massive labelled data, yet annotations may be very time-consuming or infeasible. The fault diagnosis approach that utilizes limited labelled data is becoming increasingly popular. In this paper, a Wavelet Transform (WT) and self-supervised learning-based bearing fault diagnosis framework is proposed to address the lack of supervised samples issue. Adopting the WT and cubic spline interpolation technique, original measured vibration signals are converted to the time-frequency maps (TFMs) with a fixed scale as inputs. The Vision Transformer (ViT) is employed as the encoder for feature extraction, and the self-distillation with no labels (DINO) algorithm is introduced in the proposed framework for self-supervised learning with limited labelled data and sufficient unlabeled data. Two rolling bearing fault datasets are used for validations. In the case of both datasets only containing 1 utilizing the feature vectors extracted by the trained encoder without fine-tuning, over 90% average diagnosis accuracy can be obtained based on the simple K-Nearest Neighbor (KNN) classifier. Furthermore, the superiority of the proposed method is demonstrated in comparison with other self-supervised fault diagnosis methods.


page 4

page 5

page 8

page 10

page 11

page 14

page 17

page 18


A novel Time-frequency Transformer and its Application in Fault Diagnosis of Rolling Bearings

The scope of data-driven fault diagnosis models is greatly improved thro...

Improving Streaming Transformer Based ASR Under a Framework of Self-supervised Learning

Recently self-supervised learning has emerged as an effective approach t...

Action-based Early Autism Diagnosis Using Contrastive Feature Learning

Autism, also known as Autism Spectrum Disorder (or ASD), is a neurologic...

Learning from Pseudo Lesion: A Self-supervised Framework for COVID-19 Diagnosis

The Coronavirus disease 2019 (COVID-19) has rapidly spread all over the ...

Self-Supervised Encoder for Fault Prediction in Electrochemical Cells

Predicting faults before they occur helps to avoid potential safety haza...

Acoustic Signal based Non-Contact Ball Bearing Fault Diagnosis Using Adaptive Wavelet Denoising

This paper presents a non-contact fault diagnostic method for ball beari...