A clustered federated learning framework for collaborative fault diagnosis of wind turbines

Abstract

Data-driven approaches demonstrate significant potential in accurately diagnosing faults in wind turbines. To enhance diagnostic performance and reduce communication costs in federated learning with data heterogeneity among different clients, we introduce a clustered federated learning framework to wind turbine fault diagnosis. Initially, a lightweight multiscale separable residual network (LMSRN) model is proposed for each local client. The LMSRN model integrates a multiscale spatial feature derivation unit and a depthwise separable feature extraction unit. Subsequently, to tackle data heterogeneity among clients, canonical correlation coefficients of representations are extracted from the intermediate layers of local LMSRN models, and a representational canonical correlation clustering (RCCC) method is proposed to assess the similarity of local LMSRN models and group them into clusters. Finally, a global model is trained for each cluster. Real-world wind turbine data experiments showcase the superior performance of the proposed clustered federated learning framework over traditional methods in terms of diagnostic accuracy and computational speed. Additionally, the optimal choice of the number of clusters is also discussed.