河南理工大学学报（自然科学版）2025，Vol.44Issue(5)：1-8,8.DOI:10.16186/j.cnki.1673-9787.2024080007
基于动态聚类联邦学习的轴承故障诊断方法
Bearing fault diagnosis method based on dynamic clustering federated learning
摘要
Abstract
Objectives In industrial scenarios,bearing fault data held by clients are typically non-independent and identically distributed(Non-IID),which adversely affects the performance of federated learning(FL)systems.This study aimed to address the performance degradation of FL under such condi-tions.Methods This paper proposed a bearing fault diagnosis method based on dynamic clustering federated learning(DCFed).The method grouped clients with similar data distributions into clusters to mitigate model divergence caused by data heterogeneity.It comprised three main stages:parameter selection,clus-tering,and aggregation.In the parameter selection stage,optimal model parameters were provided to the clients.During clustering,K-means clustering was performed based on cosine similarity calculated from cli-ent inference results within the current training round,encouraging clients within each cluster to approxi-mate an independent and identically distributed(IID)pattern.In the aggregation stage,local models were first averaged within clusters and then aggregated across clusters to form an efficient and personalized global model.Results The proposed DCFed method was evaluated on the Case Western Reserve University(CWRU)bearing dataset.Under the Non-IID scenario,the global model achieved an accuracy of 95.23%,which was close to that of centralized training methods.DCFed converged within just 150 training rounds,demonstrating faster convergence and higher accuracy compared to other methods such as FedProx and Fe-dora.Further experiments examining key parameters and clustering methods confirmed the robustness and stability of the proposed approach.Conclusions Under Non-IID conditions,the proposed method effectively mitigated the impact of data heterogeneity on federated learning and accelerates convergence.Experimental results demonstrated that the method performed excellently in bearing fault diagnosis tasks.关键词
数据异构/聚类联邦学习/故障诊断/深度学习/数据隐私Key words
data heterogeneity/clustering federated learning/fault diagnosis/deep learning/data privacy分类
信息技术与安全科学引用本文复制引用
马新娜,李豪..基于动态聚类联邦学习的轴承故障诊断方法[J].河南理工大学学报（自然科学版）,2025,44(5):1-8,8.基金项目
国家自然科学基金资助项目(12172234) （12172234）
河北省自然科学基金资助项目(A2021210022) （A2021210022）
河北省三三三人才科研项目(A202101018) （A202101018）
