A hybrid machine learning model based on ensemble methods for devices fault prediction in the wood industry

Abstract

In manufacturing industries, including the wood industry, devices, and equipment are considered the basic elements and the main capital for production. That is why managers are trying to maintain and use these devices and equipment optimally. On the other hand, repurchasing device parts or repairing equipment in case of major damage can cause more damage than planned costs. Therefore, a model that can determine the fault class based on the signs seen in the equipment would prevent major damage to the device and save on repair costs. In this regard, using the registered features for equipment and with the help of machine learning algorithms, a model can be created that can classify devices in the appropriate class based on their observed features. The present study uses nine machine learning algorithms to make this model, trains each model on three sets of selected features, and finally compares them. It is worth mentioning that after evaluating the models, based on the features selected from the embedded techniques, permutation feature importance methods, and genetic algorithm, the best models are considered as categorical boosting with the training and testing accuracy of 0.895 and 0.909, random forest with the training and testing accuracy of 0.905 and 0.893, and extreme gradient boosting with the training and testing accuracy of 0.884 and 0.885.