Open Access iconOpen Access



A Novel Method Based on UNET for Bearing Fault Diagnosis

Dileep Kumar1,*, Imtiaz Hussain Kalwar2, Tanweer Hussain1, Bhawani Shankar Chowdhry1, Sanaullah Mehran Ujjan1, Tayab Din Memon3

1 National Centre of Robotics and Automation, HHCMS Lab, Mehran University of Engineering & Technology, Jamshoro, 76020, Sindh, Pakistan
2 Department of Electrical Engineering, DHA SUFFA University, Karachi, Sindh, Pakistan
3 Department of Electronic Engineering, Mehran University of Engineering and Technology, Jamshoro, 76020, Sindh, Pakistan

* Corresponding Author: Dileep Kumar. Email: email

(This article belongs to the Special Issue: Machine Learning for Data Analytics)

Computers, Materials & Continua 2021, 69(1), 393-408.


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%.


Cite This Article

APA Style
Kumar, D., Kalwar, I.H., Hussain, T., Chowdhry, B.S., Ujjan, S.M. et al. (2021). A novel method based on UNET for bearing fault diagnosis. Computers, Materials & Continua, 69(1), 393-408.
Vancouver Style
Kumar D, Kalwar IH, Hussain T, Chowdhry BS, Ujjan SM, Memon TD. A novel method based on UNET for bearing fault diagnosis. Comput Mater Contin. 2021;69(1):393-408
IEEE Style
D. Kumar, I.H. Kalwar, T. Hussain, B.S. Chowdhry, S.M. Ujjan, and T.D. Memon "A Novel Method Based on UNET for Bearing Fault Diagnosis," Comput. Mater. Contin., vol. 69, no. 1, pp. 393-408. 2021.


cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 2780


  • 1638


  • 1


Share Link