Handling Data Imbalance in Predictive Maintenance for Machines using SMOTE-based Oversampling
Sashank Sridhar, Sowmya Sanagavarapu
Abstract
The identification of failures and defects in industrial machines has proven to be a challenge to gauge their warranty and performance. Depreciation in industrial machines occurs due to several factors, most commonly- tool wear, strain, heat and power failure. In this paper, the development of machine learning algorithms for the prediction of machine failures is done. A synthesized dataset was used in the predictive maintenance model, that reflects real-time failures encountered in the industries. The class data imbalance hinders the performance of machine learning algorithms and this is handled by evaluating SMOTE-based oversampling techniques. By using SMOTE technique, a 7.83 % increase in the AUC score is observed, thereby improving the performance of the Random Forest classifier in distinguishing the instances of non-failure and machine failures.