arXiv:2505.07315cs.AIcs.LG2025-05被引 3

解决设备故障诊断中标签不一致问题,实现跨域协同建模。

FedIFL: A federated cross-domain diagnostic framework for motor-driven systems with inconsistent fault modes

  • 通过原型对比学习与特征生成,缓解客户端内部域偏移。
  • 引入实例级联邦一致性损失,提升不同客户端间特征一致性。
  • 适合工业界跨设备、异构标签场景的故障诊断应用。

由于工业数据稀缺,尤其是初创企业难以独立训练全面的故障诊断模型;联邦学习可在保障数据隐私的前提下实现协作训练,是理想解决方案。然而,工作条件差异导致故障模式不一致,造成各客户端标签空间不统一。在联邦诊断场景中,标签不一致使本地模型聚焦于特定故障模式,且不同客户端将不同故障映射到相似特征表示,削弱全局模型泛化能力。为此,本文提出联邦跨域诊断框架 FedIFL。在客户端内训练中,原型对比学习缓解内部域偏移,特征生成机制在保护隐私前提下使本地模型可访问其他客户端分布。在跨客户端训练中,引入特征解耦机制,设计实例级联邦一致性损失,确保不同客户端间不变特征的实例级一致性;同时构建联邦个性化损失与正交损失,区分不变特征与特定特征。最终,聚合模型在全局标签空间中实现良好泛化,可准确诊断目标客户端电机驱动系统(MDS)在不一致标签空间下的故障。基于真实 MDS 数据的实验验证了 FedIFL 在异构故障模式下联邦跨域诊断的有效性与优越性。

原文摘要 · Abstract (English)

Due to the scarcity of industrial data, individual equipment users, particularly start-ups, struggle to independently train a comprehensive fault diagnosis model; federated learning enables collaborative training while ensuring data privacy, making it an ideal solution. However, the diversity of working conditions leads to variations in fault modes, resulting in inconsistent label spaces across different clients. In federated diagnostic scenarios, label space inconsistency leads to local models focus on client-specific fault modes and causes local models from different clients to map different failure modes to similar feature representations, which weakens the aggregated global model's generalization. To tackle this issue, this article proposed a federated cross-domain diagnostic framework termed Federated Invariant Features Learning (FedIFL). In intra-client training, prototype contrastive learning mitigates intra-client domain shifts, subsequently, feature generating ensures local models can access distributions of other clients in a privacy-friendly manner. Besides, in cross-client training, a feature disentanglement mechanism is introduced to mitigate cross-client domain shifts, specifically, an instance-level federated instance consistency loss is designed to ensure the instance-level consistency of invariant features between different clients, furthermore, a federated instance personalization loss and an orthogonal loss are constructed to distinguish specific features that from the invariant features. Eventually, the aggregated model achieves promising generalization among global label spaces, enabling accurate fault diagnosis for target clients' Motor Driven Systems (MDSs) with inconsistent label spaces. Experiments on real-world MDSs validate the effectiveness and superiority of FedIFL in federated cross-domain diagnosis with inconsistent fault modes.

联邦学习故障诊断跨域学习电机系统

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。