Cloud-edge collaborative transfer fault diagnosis of rotating machinery via federated fine-tuning and target self-adaptation
Rui Wang,Weiguo Huang,Yixiang Lu,Jun Wang,Chuancang Ding,Yi Liao,Juanjuan Shi
DOI: https://doi.org/10.1016/j.eswa.2024.123859
IF: 8.5
2024-04-11
Expert Systems with Applications
Abstract:The data-driven fault diagnostic methods have made significant advances and breakthroughs in the past decades. However, due to the huge time and labor costs, single user is hard to collect sufficient monitoring data for model training and multiple users are preferred for collaborative modeling with distributed data. Considering the concerns of data privacy or conflicts of interest, the local data of different users cannot be directly shared with others for centralized training. Moreover, they are usually sampled from different working environments, leading to data island and distribution shift issues. To solve the two issues, this paper proposes a novel cloud-edge collaborative transfer fault diagnosis approach for rotating machinery. The federated learning scheme is introduced for collaborative modeling of multiple users while ensuring the data privacy of participants. To bridge the distribution gap across clients, the source training module is constructed to conduct local model training on insufficient labeled data from different users, and a federated fine-tuning module is then developed to remove distribution discrepancies between different source users, thus building accurate source models. Finally, a target self-adaptation module is designed to reduce prediction differences of source client models on target samples and achieve effective adaptation of the target client to source clients. Comprehensive comparative analysis shows that the proposed approach achieves a performance gain of 0.45% to the traditional centralized transfer learning method, demonstrating its potential in addressing the data island and distribution shift issues in practical applications.
computer science, artificial intelligence,engineering, electrical & electronic,operations research & management science