Adaptive data pruning for efficient federated learning on heterogeneous clients
摘要
Federated learning (FL) involves clients independently collecting and training data. Discrepancies in client computational capabilities and data volume can result in waiting time, hindering convergence. Intuitively, employing pruning to optimize sample quantity can enhance training efficiency. However, existing works often overlook this efficiency improvement due to the impact of the stage of pruning execution and the amount of pruned data on performance. To address this, we propose an Adaptive Data Pruning (AdaDP) method to dynamically adjust the sample size for heterogeneous clients. The key insight of AdaDP is that the initial FL phase is crucial for achieving optimal performance, as most samples can be learned quickly and then disregarded without affecting the model. Therefore, we formulate adaptive pruning as an optimization problem constrained by the initially obtained client response times to calculate the sample size. Experimental results demonstrate that AdaDP effectively balances efficiency and model performance, outperforming other methods.