English

Lab-scale Vibration Analysis Dataset and Baseline Methods for Machinery Fault Diagnosis with Machine Learning

Signal Processing 2024-07-25 v1 Machine Learning Systems and Control Systems and Control

Abstract

The monitoring of machine conditions in a plant is crucial for production in manufacturing. A sudden failure of a machine can stop production and cause a loss of revenue. The vibration signal of a machine is a good indicator of its condition. This paper presents a dataset of vibration signals from a lab-scale machine. The dataset contains four different types of machine conditions: normal, unbalance, misalignment, and bearing fault. Three machine learning methods (SVM, KNN, and GNB) evaluated the dataset, and a perfect result was obtained by one of the methods on a 1-fold test. The performance of the algorithms is evaluated using weighted accuracy (WA) since the data is balanced. The results show that the best-performing algorithm is the SVM with a WA of 99.75\% on the 5-fold cross-validations. The dataset is provided in the form of CSV files in an open and free repository at https://zenodo.org/record/7006575.

Keywords

Cite

@article{arxiv.2212.14732,
  title  = {Lab-scale Vibration Analysis Dataset and Baseline Methods for Machinery Fault Diagnosis with Machine Learning},
  author = {Bagus Tris Atmaja and Haris Ihsannur and Suyanto and Dhany Arifianto},
  journal= {arXiv preprint arXiv:2212.14732},
  year   = {2024}
}
R2 v1 2026-06-28T07:57:14.134Z