Histogram-Based Gradient Boosting Tree: A Federated Learning Approach for Collaborative Fault Diagnosis
Liqiao Xia, Pai Zheng, Jinjie Li, Xiao Huang, Robert X. Gao
Abstract
Data-driven approaches have been widely applied in fault diagnosis. However, having sufficient labels to effectively develop fault diagnosis models in real-world manufacturing scenarios has remained a challenge. The key issue is to break down data silos and expand the source of data availability while safeguarding data privacy. To address abovementioned gap, a federated learning model using histogram-based gradient boosting tree, is proposed for collaborative fault diagnosis under privacy-preserving. During model training process, only an update of relevant local model parameters is communicated, thus protecting data privacy while contributing to the increased accuracy of the global model. Meanwhile, a data compensation mechanism is introduced to alleviate the disparity problem in data volume across participants. Two case studies are carried out and further compared with traditional situations to reveal the effectiveness of the proposed method. It is envisioned that this developed method can be useful for industrial practitioners to integrate their data for model training while maintaining data privacy, and promoting collaboration in digital manufacturing.