Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Journal articles
  4. Federated learning with uncertainty-based client clustering for fleet-wide fault diagnosis
 
research article

Federated learning with uncertainty-based client clustering for fleet-wide fault diagnosis

Lu, Hao  
•
Thelen, Adam
•
Fink, Olga  
Show more
2024
Mechanical Systems and Signal Processing

Operators from various industries have been pushing the adoption of wireless sensing nodes for industrial monitoring, and such efforts have produced sizeable condition monitoring datasets that can be used to build diagnosis algorithms capable of warning maintenance engineers of impending failure or identifying current system health conditions. However, single operators may not have sufficiently large fleets of systems or component units to collect sufficient data to develop data-driven algorithms. One potential solution to overcome the challenge of having limited representative datasets is to merge datasets from multiple operators with the same type of assets. However, directly sharing data across the company’s borders yields privacy concerns. Federated learning (FL) has emerged as a promising solution to leverage datasets from multiple operators to train a decentralized asset fault diagnosis model while maintaining data confidentiality. However, the performance of traditional FL algorithms degrades when local clients’ datasets are heterogeneous. The dataset heterogeneity is particularly prevalent in fault diagnosis applications due to the high diversity of operating conditions and system configurations. To address this challenge, this paper proposes a novel clustering-based FL algorithm where clients are clustered based on their dataset similarity. Estimating dataset similarity between clients without explicitly sharing data is achieved by training probabilistic deep learning models and having each client examine the predictive uncertainty of the other clients’ models on its local dataset. Clients are then clustered for FL based on relative prediction accuracy and uncertainty. Experiments on three bearing fault datasets, two publicly available and one newly collected for this work, show that our algorithm significantly outperforms FedAvg and a cosine similarity-based algorithm by 5.1% and 30.7% on average over the three datasets. Further, using a probabilistic classification model has the additional advantage of accurately quantifying its predictive uncertainty, which we show it does exceptionally well.

  • Details
  • Metrics
Type
research article
DOI
10.1016/j.ymssp.2023.111068
Author(s)
Lu, Hao  
Thelen, Adam
Fink, Olga  
Hu, Chao
Laflamme, Simon
Date Issued

2024

Published in
Mechanical Systems and Signal Processing
Volume

210

Article Number

111068

Subjects

Federated learning

•

Fault diagnosis

•

Deep learning

•

Uncertainty quantification

•

Bearings

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
IMOS  
Available on Infoscience
January 30, 2024
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/203305
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés