Debunking Optimization Myths in Federated Learning for Medical Image Classification
By: Youngjoon Lee , Hyukjoon Lee , Jinu Gong and more
Potential Business Impact:
Makes AI learn better from private health pictures.
Federated Learning (FL) is a collaborative learning method that enables decentralized model training while preserving data privacy. Despite its promise in medical imaging, recent FL methods are often sensitive to local factors such as optimizers and learning rates, limiting their robustness in practical deployments. In this work, we revisit vanilla FL to clarify the impact of edge device configurations, benchmarking recent FL methods on colorectal pathology and blood cell classification task. We numerically show that the choice of local optimizer and learning rate has a greater effect on performance than the specific FL method. Moreover, we find that increasing local training epochs can either enhance or impair convergence, depending on the FL method. These findings indicate that appropriate edge-specific configuration is more crucial than algorithmic complexity for achieving effective FL.
Similar Papers
Federated Learning: A Survey on Privacy-Preserving Collaborative Intelligence
Machine Learning (CS)
Trains computers together without sharing private info.
Optimization Methods and Software for Federated Learning
Machine Learning (CS)
Helps many phones learn together safely.
Federated learning, ethics, and the double black box problem in medical AI
Machine Learning (CS)
Keeps patient data private while training AI.