Maximizing Model Generalization for Machine Condition Monitoring with Self-Supervised Learning and Federated Learning

TL;DR

This paper explores combining self-supervised learning with federated learning to improve the generalization of machine condition monitoring models, especially in unlabeled, diverse, and evolving fault scenarios.

Contribution

It introduces a novel approach that leverages Barlow Twins SSL and federated learning to enhance model robustness and transferability in industrial fault diagnosis.

Findings

Barlow Twins outperforms supervised learning on unlabeled target data.

Federated learning slightly improves generalization by sharing knowledge across devices.

The combined approach enhances fault detection in diverse and evolving conditions.

Abstract

Deep Learning (DL) can diagnose faults and assess machine health from raw condition monitoring data without manually designed statistical features. However, practical manufacturing applications remain extremely difficult for existing DL methods. Machine data is often unlabeled and from very few health conditions (e.g., only normal operating data). Furthermore, models often encounter shifts in domain as process parameters change and new categories of faults emerge. Traditional supervised learning may struggle to learn compact, discriminative representations that generalize to these unseen target domains since it depends on having plentiful classes to partition the feature space with decision boundaries. Transfer Learning (TL) with domain adaptation attempts to adapt these models to unlabeled target domains but assumes similar underlying structure that may not be present if new faults emerge. This study proposes focusing on maximizing the feature generality on the source domain and applying TL via weight transfer to copy the model to the target domain. Specifically, Self-Supervised Learning (SSL) with Barlow Twins may produce more discriminative features for monitoring health condition than supervised learning by focusing on semantic properties of the data. Furthermore, Federated Learning (FL) for distributed training may also improve generalization by efficiently expanding the effective size and diversity of training data by sharing information across multiple client machines. Results show that Barlow Twins outperforms supervised learning in an unlabeled target domain with emerging motor faults when the source training data contains very few distinct categories. Incorporating FL may also provide a slight advantage by diffusing knowledge of health conditions between machines.