Neural Architecture Search for Bearing Fault Classification

Edicson Santiago Bonilla Diaz, Enrique Naredo, Nicolas Francisco Mateo Díaz, Douglas Mota Dias, Maria Alejandra Bonilla Diaz, Susan Harnett, Conor Ryan

Research output: Contribution to journalConference articlepeer-review

Abstract

In this research, we address bearing fault classification by evaluating three neural network models: 1D Convolutional Neural Network (1D-CNN), CNN-Visual Geometry Group (CNN-VGG), and Long Short-Term Memory (LSTM). Utilizing vibration data, our approach incorporates data augmentation to address the limited availability of fault class data. A significant aspect of our methodology is the application of neural architecture search (NAS), which automates the evolution of network architectures, including hyperparameter tuning, significantly enhancing model training. Our use of early stopping strategies effectively prevents overfitting, ensuring robust model generalization. The results highlight the potential of integrating advanced machine learning models with NAS in bearing fault classification and suggest possibilities for further improvements, particularly in model differentiation for specific fault classes.

Original languageEnglish
Pages (from-to)288-300
Number of pages13
JournalInternational Conference on Agents and Artificial Intelligence
Volume2
DOIs
Publication statusPublished - 2024
Event16th International Conference on Agents and Artificial Intelligence, ICAART 2024 - Rome, Italy
Duration: 24 Feb 202426 Feb 2024

Keywords

  • Bearing Fault Classification
  • Hyperparameter Optimization
  • Neural Architecture Search
  • Vibration Analysis

Fingerprint

Dive into the research topics of 'Neural Architecture Search for Bearing Fault Classification'. Together they form a unique fingerprint.

Cite this