Edge-Assisted Federated Learning: An Empirical Study from Software Decomposition Perspective

Yimin Shi, Haihan Duan, Yuanfang Chi, Keke Gai, Wei Cai

Abstract: Federated learning is considered to be a privacy-preserving collaborative machine learning training method. However, due to the general limitation of the computing ability of the terminal device, the training efficiency becomes an issue when training some complex deep neural network models. On the other hand, edges, the nearby stationary devices with higher computational capacity, might serve as a help. This paper presents the design of a component-based federated learning framework, which facilitates the offloading of training layers to nearby edge devices while preserving the users’ privacy. We conduct an empirical study on a classic convolutional neural network to validate our framework. Experiments show that this method can effectively shorten the time cost for mobile terminals to perform local training in the federated learning process.