학술논문

DecFFD: A Personalized Federated Learning Framework for Cross-Location Fault Diagnosis
Document Type
Periodical
Source
IEEE Transactions on Industrial Informatics IEEE Trans. Ind. Inf. Industrial Informatics, IEEE Transactions on. 20(5):7082-7091 May, 2024
Subject
Power, Energy and Industry Applications
Signal Processing and Analysis
Computing and Processing
Communication, Networking and Broadcast Technologies
Feature extraction
Fault diagnosis
Monitoring
Training
Servers
Optimization
Data models
feature decoupling
nonidentically and independently distributed (Non-IID) data
personalized federated learning (FL)
Language
ISSN
1551-3203
1941-0050
Abstract
Federated learning has emerged as a promising approach for fault diagnosis, as its ability to learn from decentralized data while preserving client privacy for industry. Yet, it also brings the problem of nonidentically and independently distributed (Non-IID) data, which can result in model convergence delay and performance degradation. Recent research aims to alleviate the problem caused by cross-domain without considering by cross-location. However, it is common in industrial production to have devices across different monitoring locations. Furthermore, experimental results indicate that the diagnostic models' performance of the latest techniques is significantly affected. To address the cross-location Non-IID data problem, we propose DecFFD, a personalized federated fault diagnosis framework that decouples global and personalized features. In DecFFD, we design a reconstructor for each client that acts as a supervisor and decoupler to disentangle global and personalized features. We then present a client alignment algorithm to eliminate the differences in global features among clients. In addition, we provide a theoretical analysis of fairness and generalization capability, offering a theoretical guarantee for model convergence. Finally, extensive experiments are conducted on two real-world datasets. Experimental results show that the accuracy of DecFFD outperforms the accuracy that of the state-of-the-art approach by 14.67% and converges at a faster rate.