Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
Performance Analysis and Comparison of Pre-Trained CNN in Bearing Fault Diagnostics
Carleton University, Canada.
Carleton University, Canada.
GasTOPS Ltd, Canada.
RISE Research Institutes of Sweden, Digital Systems, Smart Hardware.ORCID iD: 0000-0001-8278-8601
2024 (English)In: Proceedings - 2024 Prognostics and System Health Management Conference, PHM 2024, Institute of Electrical and Electronics Engineers Inc. , 2024, p. 422-427Conference paper, Published paper (Refereed)
Abstract [en]

As deep learning methodologies progress, convolutional neural networks (CNNs) are seeing growing application in image recognition, especially within the domain of bearing fault diagnosis. Utilizing CNN for the automatic identification of features in vibration-related images can enhance both accuracy and efficiency in bearing fault recognition. However, the industry has developed a number of pre trained CNN models and makes it demanding to select models for specific tasks of bearing fault diagnosis. Therefore, this paper aims to compare three of the main-trend CNN models’ prediction accuracy and computational performance comprehensively. First, three pretrained CNN models-VGG16, ResNet and SqueezeNet, were trained to classify the bearing vibration signal dataset which had been converted to 2-D scalogram images. Transfer learning was applied to all models in this process. Then, the prediction accuracy and training time were examined and compared. Results showed that the accuracy of all CNN models were acceptable but SqueezeNet has the lowest runtime and highest accuracy. ResNet showed slightly lower performance and VGG16 experienced overfitting and produced the lowest accuracy with the longest runtime. Overall, for similar tasks of image-classification-based bearing fault diagnostics, SqueezeNet is a better candidate compared to other main stream pretrained CNN models, which can bring great accuracy accompanied with better efficiency. The findings of this work are good reference for future model selection in bearing fault diagnosis and related tasks.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers Inc. , 2024. p. 422-427
Keywords [en]
Automatic identification; Convolution; Deep neural networks; Image enhancement; Transfer learning; Bearing fault diagnosis; Bearing fault diagnostics; Convolutional neural network; Neural network model; Performance comparison; Prediction accuracy; Runtimes; Scalogram; Squeeze net; VGG net; Convolutional neural networks
National Category
Electrical Engineering, Electronic Engineering, Information Engineering
Identifiers
URN: urn:nbn:se:ri:diva-76471DOI: 10.1109/PHM61473.2024.00080Scopus ID: 2-s2.0-85214650826OAI: oai:DiVA.org:ri-76471DiVA, id: diva2:1932078
Conference
Prognostics and System Health Management Conference, PHM 2024. Stockholm, Sweden. 28 May 2024 through 31 May 2024
Note

This project was financially supported by Natural Sciences and Engineering Research Council (NSERC) of Canada, and GasTOPs Ltd., Ottawa, Canada

Available from: 2025-01-28 Created: 2025-01-28 Last updated: 2025-09-23Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full textScopus

Authority records

Mishra, Madhav

Search in DiVA

By author/editor
Mishra, Madhav
By organisation
Smart Hardware
Electrical Engineering, Electronic Engineering, Information Engineering

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 50 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf