摘要
arXiv:2505.07315v1 宣告类型: 新
摘要: 由于工业数据稀缺,个体设备用户,尤其是初创企业,在独立训练全面故障诊断模型时面临困难;联邦学习使得在保持数据隐私的前提下进行合作训练成为可能,因此它是一个理想的选择。然而,工作条件的多样性导致了故障模式的差异,从而在不同客户端之间产生了不一致的标签空间。在联邦诊断场景中,标签空间的不一致性导致本地模型专注于各自的故障模式,并导致来自不同客户端的本地模型将不同的故障模式映射到相似的特征表示,从而削弱了聚合的全局模型的一般性。为了解决这一问题,本文提出了一种称为 FedIFL(联邦不变特征学习)的联邦跨域诊断框架。在客户端内训练中,原型对比学习减轻了客户端内域移位,随后,特征生成确保本地模型可以以隐私友好的方式访问其他客户端的分布。此外,在客户端间训练中,引入了一种特征去纠缠机制以减轻客户端间域移位。具体而言,设计了一种实例级别联邦实例一致性损失,以确保不同客户端的不变特征在实例级别的一致性,进一步构建了联邦实例个性化损失和正交损失以区分特定的特征和不变特征。最终,聚合模型在全局标签空间中实现了良好的泛化,使得在不同标签空间的目标客户端的电机驱动系统(MDSs)中实现准确的故障诊断成为可能。在实际MDS上的实验验证了FedIFL在具有不一致故障模式的联邦跨域诊断中的有效性和优越性。