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

Shu Sun,Jinchuan Qian,Xinmin Zhang,Zhihuan Song
DOI: https://doi.org/10.1109/ddcls61622.2024.10606790
2024-01-01
Abstract:With the development of big data, data-driven fault diagnosis models have experienced rapid growth. However, as the complexity of models increases, there is a corresponding rise in the demand for data. To obtain a substantial amount of data, traditional approaches involve centralizing datasets from multiple factories for training. However, due to heightened emphasis on data privacy from enterprises and regulatory bodies, this method becomes more challenging to implement and poses risks of data leakage. In response to this challenge, this paper designs a novel federated learning-based fault diagnosis modeling approach. Each local device trains its model on its local dataset and subsequently uploads updated model parameters to a central server for global model aggregation and updating. However, it is worth noting that in industrial processes, there may be concurrent class imbalance and data client imbalance, which brings challenges to fault diagnosis modeling based on federated learning. To address this challenge, we further propose a dynamic client aggregation fault diagnosis method based on category loss in federated learning. Experimental results indicate that the proposed approach not only protects data privacy in imbalanced class and client scenarios but also achieves outstanding fault recognition accuracy and robustness.
What problem does this paper attempt to address?