91 / 2021-07-21 16:41:14
Federated Learning for Bearing Fault Diagnosis with Dynamic Weighted Averaging
终稿
Junbin Chen / South China University of Technology
Jipu Li / South China University of Technology
Ruyi Huang / South China University of Technology
Ke Yue / South China University of Technology
Zhuyun Chen / South China University of Technology
Weihua Li / South China University of Technology
Data-driven methods for machinery fault diagnosis have been widely investigated and developed in recent years. Generally, the performance of the data-driven-based diagnosis models deeply depends on abundant high-quality data, which is unrealistic to a specific machine owner in the actual industry. Therefore, multiple data owners are motivated to unite to train a good global model with the aggregated big dataset. However, due to legal regulations on data privacy or potential conflict of interest, it is difficult to build a trusted and safe central database to aggregate the local datasets. To address this issue, a federated learning scheme with dynamic weighted averaging is proposed. In the training process, models are updated with local data. Then, the model parameters are uploaded to the server and aggregated, followed by distributing the updated global model back to the federated learning participants. In this case, data from different participants is not exposed during the training, which ensures data privacy. The proposed dynamic weighted averaging algorithm further considers the imbalance of distributed data, which can effectively reduce the influence of bad local data. The experiments show that the proposed federated learning scheme outperforms the traditional one and offers a promising perspective in privacy-preserving data mining.
重要日期
  • 会议日期

    10月21日

    2021

    10月23日

    2021

  • 10月26日 2021

    注册截止日期

主办单位
Southeast University, China
移动端
在手机上打开
小程序
打开微信小程序
客服
扫码或点此咨询