A Privacy-Preserving Federated Learning Framework with Adaptive Client Clustering for Healthcare AI
摘要
This paper proposes FedCLUS, a privacy-preserving FL framework with dynamic client clustering to address data heterogeneity and data privacy challenges in medical domains. By leveraging pseudo-data generated through a Conditional WGAN-GP, the server estimates model similarity and clusters clients accordingly. This enables adaptive and personalized model aggregation without sharing sensitive patient data. The approach improves clustering accuracy, model robustness and performance under non-IID conditions. Experimental results on a sleep posture classification task demonstrate that FedCLUS outperforms existing FL methods in both accuracy and privacy protection across varying privacy budgets and gradient clipping settings, offering a scalable and secure solution for healthcare applications.