Abstract:Intelligent mechanical fault diagnosis techniques have been extensively developed in recent years. Owing to the advantage of data privacy protection, federated learning has recently received increasing attention; this approach can utilize monitoring data from multiple local clients to train an optimal global diagnosis model. However, low-quality data are often present for some clients, including mislabeled data and incomplete data that lack some health states. Furthermore, the data distributions are usually different across different clients owing to variations in machine operating conditions. Therefore, the performance of the federated diagnostic model may be limited by low-quality data and data distribution discrepancies. To address this issue, a federated transfer learning method with low-quality knowledge filtering and dynamic model aggregation is proposed. First, a dynamic filter is designed by filtering out low label probabilities predicted by low-quality source domain models to construct high-confidence pseudo-labels for the target domain data. Then, the batch normalized maximum mean discrepancy (BN-MMD) distance metric is introduced into the training loss function to reduce the data distribution discrepancy between the source clients and the target client without private data leakage. While building the global model in each training round, a dynamic model aggregation algorithm is proposed to mitigate the influence of low-quality clients. This algorithm evaluates the weight of each client according to its contribution to the total consensus diagnostic knowledge and then aggregates the local models with adaptive weights. Consequently, it can overcome the drawback of the classical federated averaging (FedAvg) algorithm, where all local clients are assigned the same weight when constructing the global model. Experiments are conducted on three bearing datasets under various loads and speeds. Compared with some existing diagnostic methods, the proposed federated transfer learning method can reduce the impact of low-quality data and achieve higher diagnostic accuracy while preserving data privacy.

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

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

Federated learning for intelligent fault diagnosis based on similarity collaboration

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

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

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

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

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

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

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

A Federated Transfer Learning Method with Low-Quality Knowledge Filtering and Dynamic Model Aggregation for Rolling Bearing Fault Diagnosis

Federated Transfer Learning in Fault Diagnosis under Data Privacy with Target Self-Adaptation

A federated learning approach to mixed fault diagnosis in rotating machinery

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

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

Self-paced decentralized federated transfer framework for rotating machinery fault diagnosis with multiple domains

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

Cloud-edge collaborative transfer fault diagnosis of rotating machinery via federated fine-tuning and target self-adaptation

Federated Transfer Learning Strategy: A Novel Cross-Device Fault Diagnosis Method Based on Repaired Data

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

Intelligent fault diagnosis of rolling bearing based on an active federated local subdomain adaptation method