arXiv · 2601.12664
Generalizable Hyperparameter Optimization for Federated Learning on Non-IID Cancer Images
Abstract
Deep learning for cancer histopathology training conflicts with privacy constraints in clinical settings. Federated Learning (FL) mitigates this by keeping data local; however, its performance depends on hyperparameter choices under non-independent and identically distributed (non-IID) client datasets. This paper examined whether hyperparameters optimized on one cancer imaging dataset generalized across non-IID federated scenarios. We considered binary histopathology tasks for ovarian and colorectal cancers. We perform centralized Bayesian hyperparameter optimization and transfer dataset-specific optima to the non-IID FL setup. The main contribution of this study is the introduction of a simple cross-dataset aggregation heuristic by combining configurations by averaging the learning rates and considering the modal optimizers and batch sizes. This combined configuration achieves a competitive classification performance.
Explore related subjects
Keep this discovery
Elisa Gonçalves Ribeiro, Rodrigo Moreira, Larissa Ferreira Rodrigues Moreira, André Ricardo Backes. 2026-01-19. Generalizable Hyperparameter Optimization for Federated Learning on Non-IID Cancer Images. https://arxiv.org/abs/2601.12664
Cite the original work for its findings. Save a collection to share your selection of sources.