Tool wear state recognition under imbalanced data based on WGAN-GP and lightweight neural network ShuffleNet

Wen Hou,Hong Guo,Bingnan Yan,Zhuang Xu,Chao Yuan,Yuan Mao
DOI: https://doi.org/10.1007/s12206-022-0912-1
IF: 1.81
2022-10-05
Journal of Mechanical Science and Technology
Abstract:The tool is an important part of machining, and its condition determines the operational safety of the equipment and the quality of the workpiece. Therefore, tool condition monitoring (TCM) is of great significance. To address the imbalance of the tool monitoring signal and achieve a lightweight model, a TCM method based on WGAN-GP and ShuffleNet is proposed in this paper. The tool monitoring data are enhanced and balanced using WGAN-GP, and the 1D signal data are converted into 2D grayscale images. The existing ShuffleNet is improved by adding a channel attention mechanism to construct the entire model. The tool wear state is recognized through experimental validation of the milling dataset and compared with those through other models. Results show that the proposed model achieves an accuracy of 99.78 % in recognizing the wear state of tools under imbalanced data while ensuring a light weight, showing the superiority of the method.
engineering, mechanical
What problem does this paper attempt to address?