Abstract:The federated fault diagnosis approach has achieved remarkable results in recent years, which enables multiple clients with similar mechanical devices to collaboratively construct global intelligent diagnostic models while protecting data privacy. However, in practice, the statistical heterogeneity of data collected from different clients, as well as the model heterogeneity due to local model personalization, pose great challenges to federated learning. Meanwhile, using a central server as an information management center to build global models increases additional model parameters and the risk of data privacy leakage. To address these issues, this paper proposes a heterogeneous federated learning framework based on peer-to-peer communication (P2PCHF) for rotating machinery fault diagnosis. To achieve heterogeneous client communication without relying on a central server, sharing unlabeled dataset is utilized in the collaborative updating phase to achieve peer-to-peer communication between clients and to align instance dimensions and clustering dimensions between heterogeneous clients by constructing inter-correlation matrices to achieve feature-level and semantic-level knowledge exchange for better inter-domain generalization capabilities. Joint knowledge distillation based on class labels and class relations is introduced in the local update phase to mitigate forgetting effect in the local update phase of private models and effectively balance multi-domain category knowledge. It is verified in three cases that the proposed P2PCHF can effectively address model heterogeneity and data statistics heterogeneity among clients, and enable locally-privatized models to gain inter-domain generalization capability. The code framework is available at https://github.com/JC952/P2PCHF.

Federated learning for intelligent fault diagnosis based on similarity collaboration

Class-Imbalance and Client-Imbalance Federated Learning for Fault Diagnosis

Federated learning for machinery fault diagnosis with dynamic validation and self-supervision

Heterogeneous Federated Learning: Client-side Collaborative Update Inter-Domain Generalization Method for Intelligent Fault Diagnosis

An Efficient Federated Learning Framework for Machinery Fault Diagnosis with Improved Model Aggregation and Local Model Training.

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

A Federated Adversarial Fault Diagnosis Method Driven by Fault Information Discrepancy

A federated learning approach to mixed fault diagnosis in rotating machinery

An Asynchronous Quasi‐Cloud/Edge/Client Collaborative Federated Learning Mechanism for Fault Diagnosis

Fault diagnosis based on federated learning driven by dynamic expansion for model layers of imbalanced client

FedAlign: Federated Model Alignment Via Data-Free Knowledge Distillation for Machine Fault Diagnosis

Data Privacy Preserving Federated Transfer Learning in Machinery Fault Diagnostics Using Prior Distributions

DecFFD: A Personalized Federated Learning Framework for Cross-Location Fault Diagnosis

Federated Learning with Uncertainty-Based Client Clustering for Fleet-Wide Fault Diagnosis

Federated Transfer Learning for Bearing Fault Diagnosis Based on Averaging Shared Layers

A personalized federated meta-learning method for intelligent and privacy-preserving fault diagnosis

A Federated Distillation Domain Generalization Framework for Machinery Fault Diagnosis with Data Privacy

A Clustered Federated Learning Framework for Collaborative Fault Diagnosis of Wind Turbines

Data Privacy Protection: A Novel Federated Transfer Learning Scheme for Bearing Fault Diagnosis

Blockchain-based Decentralized Federated Transfer Learning Methodology for Collaborative Machinery Fault Diagnosis

Federated Domain Generalization With Global Robust Model Aggregation Strategy For Bearing Fault Diagnosis