一种基于协同训练的旋转部件故障诊断方法

本发明公开了一种基于协同训练的旋转部件故障诊断方法，属于特定计算模型技术领域，包括：根据故障类型得到第一故障类型组及第二故障类型组，结合模型训练数据集，得到第一元训练集及第二元训练集；基于预置的模型参数，根据第一元训练集，通过辅助二元分类器得到辅助二元分类损失，通过交叉熵函数得到多分类器损失，结合得到整体损失；根据整体损失，更新模型参数，结合第二元训练集，得到整体损失，根据第一元训练集及第二元训练集得到的整体损失得到目标函数，以用于旋转部件故障诊断模型的训练。本发明通过元训练集与协同梯度策略，使模型无需目标工况数据即可适应未知工况，结合辅助分类器抑制未知干扰，显著提升泛化能力并降低数据成本。