一种基于三分支去噪的轴承故障诊断方法

本发明公开了一种基于三分支去噪的轴承故障诊断方法，S1：获取轴承故障数据集；S2：建立轴承故障诊断模型，对轴承故障诊断模型进行训练，得到训练后的轴承故障诊断模型；轴承故障诊断模型包括三分支去噪模块和多头注意力模块；三分支去噪模块用于通过三个并行的分支模块对轴承故障数据集进行去噪处理，并将不同分支模块去噪得到的轴承故障特征进行拼接；多头注意力模块用于基于拼接后的轴承故障特征输出轴承故障诊断结果；S3：基于训练后的轴承故障诊断模型和实际采集的轴承运行信号进行轴承故障诊断。本发明采用三分支形式对噪声进行去噪，并实现从全局和局部分别进行特征学习和故障诊断分类，全面提升了故障诊断模型的性能以及诊断结果准确性。