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

11/17/2022
by   Sheng Guo, et al.
0

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.

READ FULL TEXT
research
02/25/2020

Personalized Federated Learning for Intelligent IoT Applications: A Cloud-Edge based Framework

Internet of Things (IoT) have widely penetrated in different aspects of ...
research
06/18/2022

Motley: Benchmarking Heterogeneity and Personalization in Federated Learning

Personalized federated learning considers learning models unique to each...
research
03/17/2023

Multi-Task Model Personalization for Federated Supervised SVM in Heterogeneous Networks

In this paper, we design an efficient distributed iterative learning met...
research
03/24/2022

FedGradNorm: Personalized Federated Gradient-Normalized Multi-Task Learning

Multi-task learning (MTL) is a novel framework to learn several tasks si...
research
04/01/2022

Federated Learning Framework Coping with Hierarchical Heterogeneity in Cooperative ITS

In this paper, we introduce a federated learning framework coping with H...
research
04/11/2023

Multi-scale Fusion Fault Diagnosis Method Based on Two-Dimensionaliztion Sequence in Complex Scenarios

Rolling bearings are critical components in rotating machinery, and thei...
research
10/31/2012

Learning in the Model Space for Fault Diagnosis

The emergence of large scaled sensor networks facilitates the collection...

Please sign up or login with your details

Forgot password? Click here to reset