Fedmp: Federated learning through adaptive model pruning in heterogeneous edge computing

Z Jiang, Y Xu, H Xu, Z Wang, C Qiao… - 2022 IEEE 38th …, 2022 - ieeexplore.ieee.org
2022 IEEE 38th International Conference on Data Engineering (ICDE), 2022ieeexplore.ieee.org
Federated learning (FL) has been widely adopted to train machine learning models over
massive distributed data sources in edge computing. However, the existing FL frameworks
usually suffer from the difficulties of resource limitation and edge heterogeneity. Herein, we
design and implement FedMP, an efficient FL framework through adaptive model pruning.
We theoretically analyze the impact of pruning ratio on model training performance, and
propose to employ a Multi-Armed Bandit based online learning algorithm to adaptively …
Federated learning (FL) has been widely adopted to train machine learning models over massive distributed data sources in edge computing. However, the existing FL frameworks usually suffer from the difficulties of resource limitation and edge heterogeneity. Herein, we design and implement FedMP, an efficient FL framework through adaptive model pruning. We theoretically analyze the impact of pruning ratio on model training performance, and propose to employ a Multi-Armed Bandit based online learning algorithm to adaptively determine different pruning ratios for heterogeneous edge nodes, even without any prior knowledge of their computation and communication capabilities. With adaptive model pruning, FedMP can not only reduce resource consumption but also achieve promising accuracy. To prevent the diverse structures of pruned models from affecting the training convergence, we further present a new parameter synchronization scheme, called Residual Recovery Synchronous Parallel (R2SP), and provide a theoretical convergence guarantee. Extensive experiments on the classical models and datasets demonstrate that FedMP is effective for different heterogeneous scenarios and data distributions, and can provide up to 4.1× speedup compared to the existing FL methods.
ieeexplore.ieee.org
以上显示的是最相近的搜索结果。 查看全部搜索结果