Deep convolutional neural network based on self-distillation for tool wear recognition
Yi Pan,Ling Hao,Jianliang He,Kun Ding,Qiang Yu,Yulin Wang
DOI: https://doi.org/10.1016/j.engappai.2024.107851
IF: 8
2024-06-01
Engineering Applications of Artificial Intelligence
Abstract:In the ever-evolving Industry 4.0 landscape, traditional manufacturing constantly merges with cutting-edge AI technology. In manufacturing, Computer Numerical Control (CNC) machine tools heavily depend on cutting tools. Most existing researches on tool fault diagnosis center on machines customized for visual monitoring, overlooking the specific challenges posed by unmodifiable production line equipment. Mobile inspection robots, though versatile, are constrained by resources, limiting the use of large deep learning models. In view of this, we aim to introduce a tailored tool wear recognition solution for resource-constrained mobile inspection robots. To achieve this goal, a lightweight deep learning model with a unique single-stage self-distillation approach is presented. This method efficiently transfers knowledge from deep networks to shallow networks, optimizing performance while minimizing training costs. From our results, we achieve a promising top-1 accuracy of 97.66% after 50 training epochs within 1 h and 20 min of training time. This surpasses the student model obtained through offline distillation by 4.30%, saving 59.39% of training time, and improves recognition accuracy by 2.43% compared to online distillation, while reducing training time by 44.83%. Compared with the state-of-the-art method TinyVit, 12.09% of training time is saved with an accuracy improvement of 1.48%. In conclusion, our model manifests a desirable balance between efficiency and accuracy, rendering it highly suitable for deployment on mobile terminals. Furthermore, ablation experiments confirm the robustness of our method’s internal structure in influencing final performance. This research addresses tool fault diagnosis in unmodifiable production line tools and is expected to provide a practical, efficient solution with broad implications for Industry 4.0.
automation & control systems,computer science, artificial intelligence,engineering, electrical & electronic, multidisciplinary