一种DWA Transformer的轴承智能故障诊断方法

本发明公开了一种DWA Transformer的轴承智能故障诊断方法，属于故障诊断技术领域，包括：首先，获取轴承在不同健康状态下的振动信号；其次，采用傅里叶变换FFT和变分模态分解VMD对振动信号进行预处理制作数据集，将所述数据集按一定比例随机划分为训练集和测试集；接着，建立膨胀感知窗口注意力机制DWA和膨胀感知位移窗口注意力机制DSWA，在此基础上，基于Transformer架构搭建DWA Transformer模型；然后，将训练集输入所述DWA Transformer模型进行训练，获得具有降噪和高精度特征提取能力的诊断模型；最后，将测试集输入已经训练好的模型中，自动识别噪声背景下轴承的故障类型，来解决现有模型对长序列时序信号的全局特征提取能力不强，且在噪声背景下轴承故障诊断准确率低的技术问题。