Efficient Federated Learning with Encrypted Data Sharing for Data-Heterogeneous Edge Devices

TL;DR

This paper introduces FedEDS, a federated learning scheme that encrypts and shares data among edge devices to improve training speed and handle data heterogeneity, addressing network and physical constraints.

Contribution

FedEDS is a novel federated learning approach that uses encrypted data sharing and stochastic layers to enhance convergence and robustness on data-heterogeneous edge devices.

Findings

FedEDS accelerates model convergence.

It mitigates negative effects of data heterogeneity.

Experiments confirm improved model performance.

Abstract

As privacy protection gains increasing importance, more models are being trained on edge devices and subsequently merged into the central server through Federated Learning (FL). However, current research overlooks the impact of network topology, physical distance, and data heterogeneity on edge devices, leading to issues such as increased latency and degraded model performance. To address these issues, we propose a new federated learning scheme on edge devices that called Federated Learning with Encrypted Data Sharing(FedEDS). FedEDS uses the client model and the model's stochastic layer to train the data encryptor. The data encryptor generates encrypted data and shares it with other clients. The client uses the corresponding client's stochastic layer and encrypted data to train and adjust the local model. FedEDS uses the client's local private data and encrypted shared data from other clients to train the model. This approach accelerates the convergence speed of federated learning training and mitigates the negative impact of data heterogeneity, making it suitable for application services deployed on edge devices requiring rapid convergence. Experiments results show the efficacy of FedEDS in promoting model performance.