University of Limerick
Browse

Neural architecture search for bearing fault classification

Download (3.4 MB)
conference contribution
posted on 2024-03-13, 09:38 authored by Edicson Santiago Bonilla Diaz, Enrique Naredo, Nicolas Francisco Mateo Díaz, Douglas Mota DiasDouglas Mota Dias, Maria Alejandra Bonilla Diaz, Susan Harnett, Conor RyanConor Ryan

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.


History

Publication

Proceedings of the 16th International Conference on Agents and Artificial Intelligence - Volume 2: ICAART

Publisher

Scitepress

Sustainable development goals

  • (4) Quality Education

Department or School

  • Computer Science & Information Systems
  • School of Engineering

Usage metrics

    University of Limerick

    Categories

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC