计算机应用与软件2024，Vol.41Issue(11)：108-116,152,10.DOI:10.3969/j.issn.1000-386x.2024.11.015
基于交互式多头注意力的机械故障诊断方法
A FAULT DIAGNOSIS METHOD BASED ON INTERACTIVE MULTI-HEAD ATTENTION
摘要
Abstract
Aimed at the complex vibration signals of rotating machinery and the difficulty in extracting effective fault features and the problem that the multi head attention fault diagnosis method has high computational complexity,an interactive multi-head attention fault diagnosis method is proposed.By using the method of dividing the input data feature maps into windows,the computational complexity of attention was reduced.The rolled feature maps scheme was used to establish the data interaction between windows and the data interaction between self-attention,which reduced the computational complexity and ensured the classification accuracy.The position information of the input data was integrated into the attention weight matrix to enhance the ability of neural network to distinguish the position of input data.In the experiment part,in order to test the performance of the algorithm,it was applied to the ZHS-2 multi-function motor flexible rotor test bed to verify.Compared with other data-driven fault diagnosis methods,the results show that this method can effectively identify various fault features and realize fault diagnosis.关键词
机器学习/深度学习/故障诊断/多头注意力机制Key words
Machine learning/Deep learning/Fault diagnosis/Multi-head attention分类
信息技术与安全科学引用本文复制引用
冯肖亮,赵广..基于交互式多头注意力的机械故障诊断方法[J].计算机应用与软件,2024,41(11):108-116,152,10.基金项目
国家自然科学基金项目(U1804163). （U1804163）
