Federated learning (FL) has attracted growing attention for decentralized training of diagnosis models across clients while preserving privacy. Industrial big data are inherently not independent and identically distributed, which deteriorates the performance of global FL diagnosis model on individual clients. To solve this problem, a novel personalized FL framework, namely federated diverse knowledge learning (FDKL) is proposed in this paper. In the FDKL method, a trainable client-specific knowledge memory module is designed to store the bias of feature representation locally. Meanwhile, the client-general knowledge regularization module is developed to guide local feature encoder learn global unbiased knowledge. Two modules work synergistically and eventually customizes a personalized model for each client. Extensive experiments conducted on a self-build propulsion shaft test rig verify the effectiveness and superiority of the proposed method.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Federated Diverse Knowledge Learning for Machinery Fault Diagnosis with Data Heterogeneity

  • Zongzhen Ye,
  • Jun Wu,
  • Xuesong He,
  • Yifan Yang,
  • Tianjiao Dai

摘要

Federated learning (FL) has attracted growing attention for decentralized training of diagnosis models across clients while preserving privacy. Industrial big data are inherently not independent and identically distributed, which deteriorates the performance of global FL diagnosis model on individual clients. To solve this problem, a novel personalized FL framework, namely federated diverse knowledge learning (FDKL) is proposed in this paper. In the FDKL method, a trainable client-specific knowledge memory module is designed to store the bias of feature representation locally. Meanwhile, the client-general knowledge regularization module is developed to guide local feature encoder learn global unbiased knowledge. Two modules work synergistically and eventually customizes a personalized model for each client. Extensive experiments conducted on a self-build propulsion shaft test rig verify the effectiveness and superiority of the proposed method.