NestFL: Enhancing Federated Learning Through Nested Multi-Capacity Model Pruning in Heterogeneous Edge Computing

Xiaomao Zhou,Qingmin Jia,Yujiao Hu,Renchao Xie
DOI: https://doi.org/10.1145/3694908.3696174
2024-01-01
Abstract:Federated learning (FL) has been explored as a promising solution for distributed machine learning at the edge. However, the limited capacity and heterogeneity of edge devices usually bring FL with various critical challenges, such as Non-IID data, communication bottleneck, learning inefficiency, etc. In addition, learning a single global model can hardly suffice to work well on all participating devices. In this paper, we present NestFL, a learning-efficient FL framework for edge computing, which can jointly improve the training efficiency and achieve personalization. Specifically, NestFL takes the runtime resources of the edge devices into consideration and assigns each device a sparse-structured subnetwork by progressively performing the structured pruning. During training, only the updates of these subnetworks are transmitted to the central server. Additionally, these generated subnetworks adopt a structure- and parametersharing mechanism, making themselves nested inside a multi-capacity global model. In doing so, the overall communication and computation costs can be significantly reduced, and each device can learn a personalized model without introducing extra parameters. Furthermore, a weighted aggregation mechanism is designed to improve the training performance and maximally preserve personalization. Experimental results in different applications demonstrate the superiority of NestFL over the baseline approaches in terms of model accuracy, convergence speed, and personalization preservation.
What problem does this paper attempt to address?