CFL-SparseMed: Communication-Efficient Federated Learning for Medical Imaging with Top-k Sparse Updates
This work addresses communication efficiency and data heterogeneity issues in federated learning for medical imaging, offering a domain-specific incremental improvement.
The paper tackles the problem of high communication costs and data heterogeneity in federated learning for medical imaging by proposing CFL-SparseMed, which uses Top-k sparsification to reduce communication overhead while maintaining model accuracy, achieving a 60% reduction in communication costs with competitive accuracy on benchmark datasets.
Secure and reliable medical image classification is crucial for effective patient treatment, but centralized models face challenges due to data and privacy concerns. Federated Learning (FL) enables privacy-preserving collaborations but struggles with heterogeneous, non-IID data and high communication costs, especially in large networks. We propose \textbf{CFL-SparseMed}, an FL approach that uses Top-k Sparsification to reduce communication overhead by transmitting only the top k gradients. This unified solution effectively addresses data heterogeneity while maintaining model accuracy. It enhances FL efficiency, preserves privacy, and improves diagnostic accuracy and patient care in non-IID medical imaging settings. The reproducibility source code is available on \href{https://github.com/Aniket2241/APK_contruct}{Github}.