FedHKD: A Hierarchical Federated Learning Approach Integrating Clustering and Knowledge Distillation for Non-IID Data
摘要
Federated learning allows decentralized model training while preserving data privacy. However, Non-IID data poses significant challenges, leading to performance degradation and increased communication overhead. This paper introduces FedHKD, a hierarchical FL algorithm that integrates client clustering and knowledge distillation to address Non-IID challenges. By clustering clients with similar data distributions and transferring knowledge across clusters, FedHKD improves model accuracy while reducing communication costs. Experimental results on Fashion-MNIST and CIFAR-10 datasets demonstrate up to 14.24% improvement in accuracy and a substantial reduction in communication overhead, proving its effectiveness in Non-IID environments.