首页|基于元学习与多头注意力的刀具磨损值预测

基于元学习与多头注意力的刀具磨损值预测

扫码查看
为了采用基于神经网络方法预测刀具磨损值时在多工况条件下进行有效迁移,提出一种结合模型无关元学习(MAML)与多头注意力(MHA)模型的方法,预测多工况下刀具的磨损值.首先通过多传感器收集刀具切削过程中的多维信息,从中提取时域频域和时频域的特征信息,以构建时频特征矩阵;其次将MHA模型作为基模型,利用历史工况信息构建的时频特征样本,通过MAML方法训练该模型,获得最优的MHA模型初始化参数;然后最优的初始化MHA模型通过少量的初始磨损样本迭代训练几次来适应新工况,从而预测新工况的刀具磨损值;最后通过相关实验验证,该方法在多工况下的刀具磨损值预测中能够有效进行模型迁移.
Prediction of tool wear value based on meta-learning and multi-head attention
To achieve effective transfer of tool wear prediction values based on neural network methods under multiple working conditions,a method combining Model Agnostic Meta-Learning(MAML)and Multi-Head Attention(MHA)was proposed to realize tool wear prediction under multiple working conditions.The multi-dimensional in-formation in the cutting process of the tool was collected by multiple sensors,and the time-frequency feature matrix was constructed by extracting time-domain,frequency-domain and time-frequency domain features.The MHA model was utilized as the base model,and the time-frequency feature samples constructed from historical working condition information were leveraged to train this model via the MAML approach to acquire the optimal initialization parameters for the MHA model.In the new working condition,the optimal initialization MHA model was iteratively trained with a small number of initial wear samples several times to adapt to the new condition,thereby predicting the tool wear value for the new working conditions.Finally,relevant experiments demonstrated that the proposed method could achieve effective model transfer in tool wear prediction under multiple working conditions.

tool wearmeta-learningattention mechanismmulti-sensor

卢志强、朱海平、吴军

展开 >

华中科技大学机械科学与工程学院,湖北 武汉 430074

华中科技大学船舶与海洋工程学院,湖北 武汉 430074

刀具磨损 元学习 注意力机制 多传感器

国家自然科学基金资助项目

52075202

2024

计算机集成制造系统
中国兵器工业集团第210研究所

计算机集成制造系统

CSTPCD北大核心
影响因子:1.092
ISSN:1006-5911
年,卷(期):2024.30(10)