Tool Wear Prediction Combining Global Feature Attention and Long Short-Term Memory Network

Wanzhen Wang; Sze Song Ngu; Miaomiao Xin; Xiaomei Ni; Beibei Kong; Kaiyuan Wu; Ruyue Han

doi:10.46604/peti.2024.14201

Authors

Wanzhen Wang Faculty of Engineering, Universiti Malaysia Sarawak, Kota Samarahan, Malaysia
Sze Song Ngu Faculty of Engineering, Universiti Malaysia Sarawak, Kota Samarahan, Malaysia
Miaomiao Xin School of Intelligent Manufacturing and Control Engineering, Qilu Institute of Technology, Jinan, China
Xiaomei Ni School of Intelligent Manufacturing and Control Engineering, Qilu Institute of Technology, Jinan, China
Beibei Kong School of Intelligent Manufacturing and Control Engineering, Qilu Institute of Technology, Jinan, China
Kaiyuan Wu School of Intelligent Manufacturing and Control Engineering, Qilu Institute of Technology, Jinan, China
Ruyue Han School of Intelligent Manufacturing and Control Engineering, Qilu Institute of Technology, Jinan, China

DOI:

https://doi.org/10.46604/peti.2024.14201

Keywords:

tool wear, LSTM, global feature attention, feature selection

Abstract

This study aims to accurately predict tool flank wear in milling and simplify the complexity of feature selection. A hybrid approach is proposed to eclectically integrate the advantages between the long short-term memory (LSTM) network and the global feature attention (GFA) module. First, the feature matrix is calculated using the multi-domain feature extraction method. Subsequently, a parallel network is employed to achieve feature fusion. The stacked LSTM network extracts the temporal dependencies between features and the GFA module is used to adaptively complement key features representing global information of samples. Finally, the output features are concatenated, and tool wear prediction is achieved through a fully connected layer. Experiments demonstrate the optimal performance in predicting tool flank wear. Specifically, using the proposed GFA-LSTM framework reduces the mean absolute error (MAE) by 36.9%, 17.7%, and 25.2% in three experiments compared to the simple LSTM, validating the effectiveness of the proposed method.

References

Y. Hu, X. Miao, Y. Si, E. Pan, and E. Zio, “Prognostics and Health Management: A Review From the Perspectives of Design, Development and Decision,” Reliability Engineering & System Safety, vol. 217, article no. 108063, January 2022.

X. Li, B. S. Lim, J. H. Zhou, and S. Huang, “Fuzzy Neural Network Modelling for Tool Wear Estimation in Dry Milling Operation,” Annual Conference of the Prognostics and Health Management Society, vol. 1, no. 1, pp. 1-11, 2009.

K. Goebel and A. Agogino, “Milling Data Set,” NASA Prognostics Data Repository, NASA Ames Research Center, Moffett Field, CA, 2007.

L. De Pauw, T. Jacobs, and T. Goedemé, “Matwi: A Multimodal Automatic Tool Wear Inspection Dataset and Baseline Algorithms,” Computer Vision Systems – 14th International Conference, pp. 255-269, September 2023.

H. Truchan, E. Naumov, R. Abedin, G. Palmer, and Z. Ahmadi, “Multimodal Isotropic Neural Architecture With Patch Embedding,” Neural Information Processing – 30th International Conference, pp. 173-187, November 2023.

B. Cardoz, H. N. E. A. Shaikh, S. M. Mulani, A. Kumar, and S. G. Rajasekharan, “Random Forests Based Classification of Tool Wear Using Vibration Signals and Wear Area Estimation From Tool Image Data,” The International Journal of Advanced Manufacturing Technology, vol. 126, no. 7-8, pp. 3069-3081, June 2023.

L. Cao, T. Huang, X. M. Zhang, and H. Ding, “Generative Adversarial Network for Prediction of Workpiece Surface Topography in Machining Stage,” IEEE/ASME Transactions on Mechatronics, vol. 26, no. 1, pp. 480-490, February 2021.

S. Sayyad, S. Kumar, A. Bongale, K. Kotecha, G. Selvachandran, and P. N. Suganthan, “Tool Wear Prediction Using Long Short-Term Memory Variants and Hybrid Feature Selection Techniques,” The International Journal of Advanced Manufacturing Technology, vol. 121, no. 9-10, pp. 6611-6633, August 2022.

V. Warke, S. Kumar, A. Bongale, and K. Kotecha, “Robust Tool Wear Prediction Using Multi-Sensor Fusion and Time-Domain Features for the Milling Process Using Instance-Based Domain Adaptation,” Knowledge-Based Systems, vol. 288, article no. 111454, March 2024.

M. S. Babu and T. B. Rao, “An In-Process Tool Wear Assessment Using Bayesian Optimized Machine Learning Algorithm,” International Journal on Interactive Design and Manufacturing, vol. 17, no. 4, pp. 1823-1845, August 2023.

K. M. Li and Y. Y. Lin, “Tool Wear Classification in Milling for Varied Cutting Conditions: With Emphasis on Data Pre-Processing,” The International Journal of Advanced Manufacturing Technology, vol. 125, no. 1-2, pp. 341-355, March 2023.

D. K. Dhaked, P. Kumar, and S. Ganguly, “Development of Data Driven Model for Proton Exchange Membrane Fuel Cell Using Machine Learning Approaches,” IEEE 3rd International Conference on Control, Instrumentation, Energy & Communication, pp. 67-72, January 2024.

M. Marani, M. Zeinali, J. Kouam, V. Songmene, and C. K. Mechefske, “Prediction of Cutting Tool Wear During a Turning Process Using Artificial Intelligence Techniques,” The International Journal of Advanced Manufacturing Technology, vol. 111, no. 1-2, pp. 505-515, November 2020.

S. Wang, Z. Yu, G. Xu, and F. Zhao, “Research on Tool Remaining Life Prediction Method Based on CNN-LSTM-PSO,” IEEE Access, vol. 11, pp. 80448-80464, 2023.

V. Lakshmi Narayanan, D. Kumar Dhaked, and R. Sitharthan, “Improved Machine Learning-Based Pitch Controller for Rated Power Generation in Large-Scale Wind Turbine,” Renewable Energy Focus, vol. 50, article no. 100603, September 2024.

B. Li, Z. Lu, X. Jin, and L. Zhao, “Tool Wear Prediction in Milling CFRP With Different Fiber Orientations Based on Multi-Channel 1DCNN-LSTM,” Journal of Intelligent Manufacturing, vol. 35, no. 6, pp. 2547-2566, August 2024.

M. Marani, M. Zeinali, V. Songmene, and C. K. Mechefske, “Tool Wear Prediction in High-Speed Turning of a Steel Alloy Using Long Short-Term Memory Modelling,” Measurement, vol. 177, article no. 109329, June 2021.

R. Wang, Q. Song, Y. Peng, P. Jin, Z. Liu, and Z. Liu, “A Milling Tool Wear Monitoring Method With Sensing Generalization Capability,” Journal of Manufacturing Systems, vol. 68, pp. 25-41, June 2023.

W. Wang, S. S. Ngu, M. Xin, R. Liu, Q. Wang, M. Qiu, et al., “Tool Wear Prediction Based on Adaptive Feature and Temporal Attention With Long Short-Term Memory Model,” International Journal of Engineering and Technology Innovation, vol. 14, no. 3, pp. 271-284, July 2024.

J. Duan, X. Zhang, and T. Shi, “A Hybrid Attention-Based Paralleled Deep Learning Model for Tool Wear Prediction,” Expert Systems with Applications, vol. 211, article no. 118548, January 2023.

S. Hochreiter and J. Schmidhuber, “Long Short-Term Memory,” Neural Computation, vol. 9, no. 8, pp. 1735-1780, November 1997.

J. Hu, L. Shen, and G. Sun, “Squeeze-and-Excitation Networks,” IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 7132-7141, June 2018.

F. C. Zegarra, J. Vargas-Machuca, and A. M. Coronado, “Comparison of CNN and CNN-LSTM Architectures for Tool Wear Estimation,” IEEE Engineering International Research Conference, pp. 1-4, October 2021.