Histogram-Based Gradient Boosting Tree: A Federated Learning Approach for Collaborative Fault Diagnosis

Liqiao Xia, Pai Zheng, Jinjie Li, Xiao Huang, Robert X. Gao

Research output: Journal article publicationJournal articleAcademic researchpeer-review

1 Citation (Scopus)


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.

Original languageEnglish
Pages (from-to)1-12
Number of pages12
JournalIEEE/ASME Transactions on Mechatronics
Publication statusPublished - 5 Dec 2023


  • Collaboration
  • Data privacy
  • Fault diagnosis
  • fault diagnosis
  • federated learning (FL)
  • Manufacturing
  • Modeling
  • predictive maintenance
  • rotating machines
  • Task analysis
  • Training

ASJC Scopus subject areas

  • Control and Systems Engineering
  • Computer Science Applications
  • Electrical and Electronic Engineering


Dive into the research topics of 'Histogram-Based Gradient Boosting Tree: A Federated Learning Approach for Collaborative Fault Diagnosis'. Together they form a unique fingerprint.

Cite this