A Tool Wear Prediction Model Based on Attention Mechanism and IndRNN

被引:2
作者
Lu, Shaofei [1 ]
Zhu, Yajun [1 ]
Liu, Shen [1 ]
She, Jingke [1 ]
机构
[1] Hunan Univ, Coll Comp Sci & Elect Engn, Changsha, Peoples R China
来源
2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN) | 2022年
关键词
Attention Mechanism; Element-wise-Attention Gate (EleAttG); Independent Recurrent Neural Network (IndRNN); Whale Optimization Algorithm (WOA); Tool Condition Monitoring;
D O I
10.1109/IJCNN55064.2022.9889794
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Machining equipment often faces health monitoring problems during long-term use. As an essential piece of equipment for industrial processing, the state of the cutter tool is directly related to the quality of machined parts. Therefore, tool wear prediction plays an important role in improving the quality of parts and achieving intelligent management of equipment health. In real industrial scenarios, some complex tool milling needs to change the machining parameters, and the changes in these machining parameters will directly affect the tool wear status. In addition, some redundant data collected by sensors can also affect the model's training speed and prediction accuracy. To solve the above problems, a CNN-AIndRNN dual-input model is proposed in this paper. The method empowers the IndRNN attention regulation by introducing the attention mechanism EleAttG to reduce the influence of redundant information on the model. Meanwhile, CNN and AIndRNN are used to extract local and temporal features of the data to avoid information loss. For the problem of multiple working conditions, both sensor signals and machining parameters are used as model inputs in this paper to emphasize the influence of machining parameters on tool wear. Finally, the hyperparameters of the model are optimally selected using the whale optimization algorithm. The proposed model is validated on the NASA milling dataset. The experimental results show that the proposed model has a smaller RMSE and MAE than other baseline models. Machining parameters as a model input can effectively improve the performance of the model, and WOA can also optimize the model for model prediction accuracy.
引用
收藏
页数:7
相关论文
共 50 条
  • [41] A watershed water quality prediction model based on attention mechanism and Bi-LSTM
    Qiang Zhang
    Ruiqi Wang
    Ying Qi
    Fei Wen
    Environmental Science and Pollution Research, 2022, 29 : 75664 - 75680
  • [42] Traffic flow prediction based on MSCNN and attention mechanism
    Zhang, Xijun
    Si, Yong
    PROCEEDINGS OF THE INSTITUTION OF MECHANICAL ENGINEERS PART D-JOURNAL OF AUTOMOBILE ENGINEERING, 2025,
  • [43] Electricity consumption prediction based on LSTM with attention mechanism
    Lin, Zhifeng
    Cheng, Lianglun
    Huang, Guoheng
    IEEJ TRANSACTIONS ON ELECTRICAL AND ELECTRONIC ENGINEERING, 2020, 15 (04) : 556 - 562
  • [44] Confrontational flight trajectory prediction based on attention mechanism
    Sun, Yao
    Wang, Dong
    Wang, Wei
    Xiong, Lei
    Yang, Xingyu
    2020 INTERNATIONAL CONFERENCE ON BIG DATA & ARTIFICIAL INTELLIGENCE & SOFTWARE ENGINEERING (ICBASE 2020), 2020, : 211 - 214
  • [45] Vehicle motion trajectory prediction based on attention mechanism
    Liu C.
    Liang J.
    Zhejiang Daxue Xuebao (Gongxue Ban)/Journal of Zhejiang University (Engineering Science), 2020, 54 (06): : 1156 - 1163
  • [46] LSTM WASTEWATER QUALITY PREDICTION BASED ON ATTENTION MECHANISM
    Wang, Xiao-Feng
    Wei, Sheng-Nan
    Xu, Li-Xiang
    Pan, Jun
    Wu, Zhi-Ze
    Kwong, Timothy C. H.
    Tang, Yuan-Yan
    PROCEEDINGS OF 2021 INTERNATIONAL CONFERENCE ON WAVELET ANALYSIS AND PATTERN RECOGNITION (ICWAPR), 2021, : 48 - 53
  • [47] A CNN-BiLSTM model with attention mechanism for earthquake prediction
    Parisa Kavianpour
    Mohammadreza Kavianpour
    Ehsan Jahani
    Amin Ramezani
    The Journal of Supercomputing, 2023, 79 : 19194 - 19226
  • [48] A CNN-BiLSTM model with attention mechanism for earthquake prediction
    Kavianpour, Parisa
    Kavianpour, Mohammadreza
    Jahani, Ehsan
    Ramezani, Amin
    JOURNAL OF SUPERCOMPUTING, 2023, 79 (17) : 19194 - 19226
  • [49] An Integrated Graph Model for Spatial-Temporal Urban Crime Prediction Based on Attention Mechanism
    Hou, Miaomiao
    Hu, Xiaofeng
    Cai, Jitao
    Han, Xinge
    Yuan, Shuaiqi
    ISPRS INTERNATIONAL JOURNAL OF GEO-INFORMATION, 2022, 11 (05)
  • [50] Click-through Rate Prediction Model Based on Dynamic Graph Attention Mechanism Network
    Yuan, Houchuan
    He, Chengwan
    2021 4TH INTERNATIONAL CONFERENCE ON ROBOTICS, CONTROL AND AUTOMATION ENGINEERING (RCAE 2021), 2021, : 99 - 104