A novel method based on UNET for bearing fault diagnosis

Dileep Kumar Soother, Imtiaz Hussain Kalwar, Tanweer Hussain, Bhawani Shankar Chowdhry, Sanaullah Mehran Ujjan, Tayab Din Memon

Research output: Contribution to journalArticlepeer-review

6 Citations (Scopus)


Reliability of rotating machines is highly dependent on the smooth rolling of bearings. Thus, it is very essential for reliable operation of rotating machines to monitor the working condition of bearings using suitable fault diagnosis and condition monitoring approach. In the recent past, Deep Learning (DL) has become applicable in condition monitoring of rotating machines owing to its performance. This paper proposes a novel bearing fault diagnosis method based on the processing and analysis of the vibration images. The proposed method is the UNET model that is a recent development in DL models. The model is applied to the 2D vibration images obtained by transforming normalized amplitudes of the time-series vibration data samples into the corresponding vibration images. The UNET model performs pixel-level feature learning using the vibration images owing to its unique architecture. The results demonstrate that the model can perform dense predictions without any loss of label information, generally caused by the sliding window labelling method. The comparative analysis with other DL models confirmed the superiority of the UNET model which has achieved maximum accuracy of 98.91% and F1-Score of 99%.

Original languageEnglish
Pages (from-to)393-408
Number of pages16
JournalComputers, Materials and Continua
Issue number1
Publication statusPublished - 2021
Externally publishedYes


  • Condition monitoring
  • Deep learning
  • Fault diagnosis
  • Rotating machines
  • Vibration


Dive into the research topics of 'A novel method based on UNET for bearing fault diagnosis'. Together they form a unique fingerprint.

Cite this