Bearing Faulty Prediction Method Based on Federated Transfer Learning and Knowledge Distillation

In this paper, a novel bearing faulty prediction method based on federated transfer learning and knowledge distillation is proposed with three stages: (1) a “signal to image” conversion method based on the continuous wavelet transform is used as the data pre-processing method to satisfy the input ch...

Full description

Bibliographic Details
Main Authors: Yiqing Zhou, Jian Wang, Zeru Wang
Format: Article
Language:English
Published: MDPI AG 2022-05-01
Series:Machines
Subjects:
Online Access:https://www.mdpi.com/2075-1702/10/5/376
_version_ 1797498421719859200
author Yiqing Zhou
Jian Wang
Zeru Wang
author_facet Yiqing Zhou
Jian Wang
Zeru Wang
author_sort Yiqing Zhou
collection DOAJ
description In this paper, a novel bearing faulty prediction method based on federated transfer learning and knowledge distillation is proposed with three stages: (1) a “signal to image” conversion method based on the continuous wavelet transform is used as the data pre-processing method to satisfy the input characteristic of the proposed faulty prediction model; (2) a novel multi-source based federated transfer learning method is introduced to acquire knowledge from multiple different but related areas, enhancing the generalization ability of the proposed model; and (3) a novel multi-teacher-based knowledge distillation is introduced as the knowledge transference way to transfer multi-source knowledge with dynamic importance weighting, releasing the target data requirement and the target model parameter size, which makes it possible for the edge-computing based deployment. The effectiveness of the proposed bearing faulty prediction approach is evaluated on two case studies of two public datasets offered by the Case Western Reserve University and the Paderborn University, respectively. The evaluation result shows that the proposed approach outperforms other state-of-the-art faulty prediction approaches in terms of higher accuracy and lower parameter size with limited labeled target data.
first_indexed 2024-03-10T03:33:09Z
format Article
id doaj.art-c0e8347630f34eaaad068e19faa145b2
institution Directory Open Access Journal
issn 2075-1702
language English
last_indexed 2024-03-10T03:33:09Z
publishDate 2022-05-01
publisher MDPI AG
record_format Article
series Machines
spelling doaj.art-c0e8347630f34eaaad068e19faa145b22023-11-23T11:53:05ZengMDPI AGMachines2075-17022022-05-0110537610.3390/machines10050376Bearing Faulty Prediction Method Based on Federated Transfer Learning and Knowledge DistillationYiqing Zhou0Jian Wang1Zeru Wang2Computer Integrated Manufacturing System (CIMS) Research Center, College of Electronics and Information Engineering, Tongji University, Shanghai 201804, ChinaComputer Integrated Manufacturing System (CIMS) Research Center, College of Electronics and Information Engineering, Tongji University, Shanghai 201804, ChinaComputer Aided Design (CAD) Research Center, College of Electronics and Information Engineering, Tongji University, Shanghai 201804, ChinaIn this paper, a novel bearing faulty prediction method based on federated transfer learning and knowledge distillation is proposed with three stages: (1) a “signal to image” conversion method based on the continuous wavelet transform is used as the data pre-processing method to satisfy the input characteristic of the proposed faulty prediction model; (2) a novel multi-source based federated transfer learning method is introduced to acquire knowledge from multiple different but related areas, enhancing the generalization ability of the proposed model; and (3) a novel multi-teacher-based knowledge distillation is introduced as the knowledge transference way to transfer multi-source knowledge with dynamic importance weighting, releasing the target data requirement and the target model parameter size, which makes it possible for the edge-computing based deployment. The effectiveness of the proposed bearing faulty prediction approach is evaluated on two case studies of two public datasets offered by the Case Western Reserve University and the Paderborn University, respectively. The evaluation result shows that the proposed approach outperforms other state-of-the-art faulty prediction approaches in terms of higher accuracy and lower parameter size with limited labeled target data.https://www.mdpi.com/2075-1702/10/5/376knowledge distillationfederated transfer learningparameter sizeknowledge transferenceedge-computing deployment
spellingShingle Yiqing Zhou
Jian Wang
Zeru Wang
Bearing Faulty Prediction Method Based on Federated Transfer Learning and Knowledge Distillation
Machines
knowledge distillation
federated transfer learning
parameter size
knowledge transference
edge-computing deployment
title Bearing Faulty Prediction Method Based on Federated Transfer Learning and Knowledge Distillation
title_full Bearing Faulty Prediction Method Based on Federated Transfer Learning and Knowledge Distillation
title_fullStr Bearing Faulty Prediction Method Based on Federated Transfer Learning and Knowledge Distillation
title_full_unstemmed Bearing Faulty Prediction Method Based on Federated Transfer Learning and Knowledge Distillation
title_short Bearing Faulty Prediction Method Based on Federated Transfer Learning and Knowledge Distillation
title_sort bearing faulty prediction method based on federated transfer learning and knowledge distillation
topic knowledge distillation
federated transfer learning
parameter size
knowledge transference
edge-computing deployment
url https://www.mdpi.com/2075-1702/10/5/376
work_keys_str_mv AT yiqingzhou bearingfaultypredictionmethodbasedonfederatedtransferlearningandknowledgedistillation
AT jianwang bearingfaultypredictionmethodbasedonfederatedtransferlearningandknowledgedistillation
AT zeruwang bearingfaultypredictionmethodbasedonfederatedtransferlearningandknowledgedistillation