基于域泛化的轴承故障诊断方法

本发明提出了基于域泛化的轴承故障诊断方法，由已知工况的数据训练出的模型进行未知工况故障诊断。本发明通过多尺度融合与注意力机制来提取原始信号数据的有效特征片段，经过共享特征提取器和私有特征提取器的深度提取，再通过互信息损失、差异性损失、相似性损失、MMD损失函数的计算叠加权重参数求和，通过反向传播和梯度优化的方式更新模型参数，最后在训练过程中保存最优的模型，将未知工况的数据集进行测试并可视化的反馈模型精度，更加直观的反应出模型的泛化能力。其步骤如下：(1)风机轴承与齿轮数据采集；(2)原始数据采样及切片；(3)搭建网络提取数据特征；(4)损失函数处理；(5)模型训练与测试；(6)模型结果比较，将结果汇总成可视化图像。