arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2211.09406cs.LGcs.AI

Personalized Federated Learning for Multi-task Fault Diagnosis of Rotating Machinery

Digital Research Institute, ENN Group(新奥集团数字研究院)

Sheng Guo, Zengxiang Li, Hui Liu, Shubao Zhao, Cheng Hao Jin

英文摘要:

Intelligent fault diagnosis is essential to safe operation of machinery. However, due to scarce fault samples and data heterogeneity in field machinery, deep learning based diagnosis methods are prone to over-fitting with poor generalization ability. To solve the problem, this paper proposes a personalized federated learning framework, enabling multi-task fault diagnosis method across multiple factories in a privacypreserving manner. Firstly, rotating machines from different factories with similar vibration feature data are categorized into machine groups using a federated clustering method. Then, a multi-task deep learning model based on convolutional neural network is constructed to diagnose the multiple faults of machinery with heterogeneous information fusion. Finally, a personalized federated learning framework is proposed to solve data heterogeneity across different machines using adaptive hierarchical aggregation strategy. The case study on collected data from real machines verifies the effectiveness of the proposed framework. The result shows that the diagnosis accuracy could be improved significantly using the proposed personalized federated learning, especially for those machines with scarce fault samples.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑