Tool Wear Prediction Method Based on Attention Mechanism

被引：0
|
作者：
Dong J. [1 ]
Wu X. [1 ]
Gao Y. [1 ]
Su D. [1 ]
机构：
[1] School of Mechanical Engineering, Tianjin University, Tianjin
来源：
Tianjin Daxue Xuebao (Ziran Kexue yu Gongcheng Jishu Ban)/Journal of Tianjin University Science and Technology | 2024年 / 57卷 / 04期
基金：
中国国家自然科学基金;
关键词：
attention mechanism; gated recurrent unit(GRU); multiscale convolutional neural network; tool wear prediction;
D O I：
10.11784/tdxbz202209020
中图分类号：
学科分类号：
摘要：
The wear state of a tool affects the surface quality and processing stability of the workpiece；hence，accurate monitoring of its wear amount has a positive role in ensuring processing reliability and maintaining the continuity of production and processing. To further improve the generalization performance and accuracy of the tool wear prediction model，a tool wear prediction method based on a multiscale convolutional bidirectional gated recurrent unit-attention(MSCBGRU-A) neural network is proposed here，which integrates the attention mechanism and is composed of feature expansion，multiscale convolution，bidirectional gated recurrent unit(GRU)，attention，and regression modules. First，cutting force，acoustic emission，and vibration signals are taken as input signals. Next，these signals generate the tool wear output characteristic maps of multiple scales through the multiscale convolution module. The characteristic maps output by multiple convolution channels are input to the connection layer to connect in two ways，head-to-tail and stacking，to obtain two kinds of output data. Then，the two kinds of output data are input to the bidirectional GRU and attention modules，respectively. In addition，the bidirectional GRU module learns the dynamic changes of the output characteristic graph to obtain the time series characteristics. Next，the attention module assigns weights to the output of the multiscale convolutional neural network to strengthen the characteristics that contribute more to the tool wear prediction results. Finally，the regression module predicts the wear value of the tool. Furthermore，the mixed domain attention mechanism，that is，the convolutional block attention mechanism (CBAM)，is introduced through comparative experiments to obtain the MSCBGRU-CBAM model. It is proved that the attention mechanism can adaptively focus on features more relevant to tool wear by drawing an attention weight map of CBAM. Compared with other deep learning models，the MSCBGRU-CBAM model has the highest prediction accuracy. Compared with MSCBGRU models without the attention mechanism，the root mean square error decreases by 19.3%，MAE decreases by 17.7%，and R2 increases by 2.7%. © 2024 Beijing Institute of Technology. All rights reserved.
引用
收藏
页码：362 / 373
页数：11
相关论文
共 22 条
  • [11] Babu G S，, Zhao P，, Li X L., Deep convolutional neural network based regression approach for estimation of remaining useful life[C], International Conference on Database Systems for Advanced Applications, pp. 214-228, (2016)
  • [12] Zhao C Y，, Huang X Z，, Li Y X，, Et al., A double-channel hybrid deep neural network based on CNN and BiLSTM for remaining useful life prediction[J], Sensors, 20, 24, (2020)
  • [13] He Yan, Ling Junjie, Wang Yulin, Et al., In-process tool wear monitoring model based on LSTM-CNN[J], China Mechanical Engineering, 31, 16, pp. 1959-1967, (2020)
  • [14] Bazi R，, Benkedjouh T，, Habbouche H，, Et al., A hybrid CNN-BiLSTM approach-based variational mode decomposition for tool wear monitoring[J], The International Journal of Advanced Manufacturing Technology, 119, 1, pp. 1-15, (2022)
  • [15] Dong Jingchuan, Tan Zhilan, Wang Taiyong, Et al., Prediction method of tool wear combined with domain adversarial adaptation[J], Mechanical Science and Technology for Aerospace Engineering, 42, 2, pp. 165-172, (2023)
  • [16] Xu H, Hong G S, Et al., Coarse-to-fine tool condition monitoring using multiple gated recurrent units[C], 45th Annual Conference of the IEEE Industrial Electronics Society, pp. 3737-3742, (2019)
  • [17] Song Y, Et al., Distributed attention-based temporal convolutional network for remaining useful life prediction[J], IEEE Internet of Things Journal, 8, 12, pp. 9594-9602, (2020)
  • [18] Liang Lu, Liu Yuanlong, Et al., Research on short-term load forecasting of power system Based on ECA-TCN[J], Proceedings of the CSU-EPSA, 34, 11, pp. 52-57, (2022)
  • [19] Hu J, Shen L, Sun G., Squeeze-and-excitation networks[C], Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7132-7141, (2018)
  • [20] Wang Q L, Wu B G，, Zhu P F，, Et al., ECA-net：Efficient channel attention for deep convolutional neural networks[C], Proceedings of 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 11531-11539, (2020)
123