acceptodds
Under review as a conference paper at ICLR 2027

Hybrid Framework for Hyperparameter Optimization in Federated Learning.

Abstract

Federated Learning (FL) enables collaborative model training across decentralized clients while preserving data privacy, yet optimizing its hyperparameters remains a critical challenge due to data heterogeneity, dynamic resource constraints, and heavy communication overheads. Existing hyperparameter optimization techniques often fail to jointly tune system-level variables, execution dynamics, and internal algorithm parameters. In this paper, we propose PSO-REALF, a novel three-tier hybrid optimization framework designed to maximize efficiency and model accuracy in non-IID federated environments. The architecture leverages a Particle Swarm Optimization (PSO) at the intermediate layer to dynamically adjust core FL hyperparameters, and Resource-Efficient Accelerating Federated Learning (REAFL) at the execution layer to optimize local iteration assignments () under fixed time budgets. Evaluated on non-IID partitions of the MNIST and CIFAR-10 datasets using Convolutional Neural Networks, PSO-REALF demonstrates superior convergence speed and accuracy compared to state-of-the-art baselines like FedEx and Auto-FedRL.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.