TY - JOUR
T1 - Neural Architecture Search for Bearing Fault Classification
AU - Diaz, Edicson Santiago Bonilla
AU - Naredo, Enrique
AU - Díaz, Nicolas Francisco Mateo
AU - Dias, Douglas Mota
AU - Diaz, Maria Alejandra Bonilla
AU - Harnett, Susan
AU - Ryan, Conor
N1 - Publisher Copyright:
© 2024 by SCITEPRESS – Science and Technology Publications, Lda.
PY - 2024
Y1 - 2024
N2 - 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.
AB - 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.
KW - Bearing Fault Classification
KW - Hyperparameter Optimization
KW - Neural Architecture Search
KW - Vibration Analysis
UR - http://www.scopus.com/inward/record.url?scp=85190677191&partnerID=8YFLogxK
U2 - 10.5220/0012373100003636
DO - 10.5220/0012373100003636
M3 - Conference article
AN - SCOPUS:85190677191
SN - 2184-3589
VL - 2
SP - 288
EP - 300
JO - International Conference on Agents and Artificial Intelligence
JF - International Conference on Agents and Artificial Intelligence
T2 - 16th International Conference on Agents and Artificial Intelligence, ICAART 2024
Y2 - 24 February 2024 through 26 February 2024
ER -