Unsupervised and Semisupervised Machine Learning Frameworks for Multiclass Tool Wear Recognition

Abstract

Tool condition monitoring (TCM) is crucial to ensure good quality products and avoid downtime. Machine learning has proven to be vital for TCM. However, existing works are predominately based on supervised learning, which hinders their applicability in real-world manufacturing settings, where data labeling is cumbersome and costly with in-service machines. Additionally, the existing unsupervised solutions mostly handle binary decision-based TCM which is unable to fully reflect the dynamics of tool wear progression. To address these issues, we propose different unsupervised and semisupervised five-class tool wear recognition frameworks to handle fully unlabeled and partially labeled data, respectively. The underlying methods include Laplacian score, sparse autoencoder (SAE), stacked SAE (SSAE), self-organizing map, Softmax, support vector machine, and random forest. For the semisupervised frameworks, we considered designs where labeled data influence only feature learning, classifier building, or both. We also investigated different training configurations of SSAE regarding the supervision level. We applied the frameworks on two run-to-failure datasets of milling tools, recorded using a microphone and an accelerometer. Single sensor and multisensor data under different percentages of labeled training data were considered in the evaluation. The results showed which of the frameworks led to the best predictive performance under which data settings, and highlighted the significance of sensor fusion and discriminative feature representations in combating the unavailability and scarcity of labels, among other findings. The highest macro-F1 achieved for the two datasets with fully unlabeled data reached 87.52% and 75.80%, respectively, and over 90% when only 25% of the training observations were labeled.