Federated Learning (FL) is a client-based distributed machine learning framework, which aim to train a central- ized model with protecting data privacy. However, the decentralized datasets pose a challenge on the traditional FL, as they are non-independent and identical distributed (non-IID). Non-IID settings can result in client result gradient biases, which may decrease the accuracy of the model. To address this issue, we propose Federated Learning with Auxiliary Generator(FedGen), which keeps the consistent of data distribution between clients leveraging the auxiliary generator, and the gradient become more accurate. To demonstrate the effectiveness of proposed method, extensive experiments are conducted on the benchmark datasets, including the MNIST and LEAF dataset. The experimental results shows that FedGen converges 1.2 times faster than FedAvg, while the accuracy can be increased.
Federated learning is an emerging machine learning setting, which can train a shared model on large amounts of decentralized data while protecting data privacy. However, the communication cost of federated learning is heavy, especially for mobile devices with higher latency and lower throughput. Although several algorithms have been proposed to reduce the communication cost, they are extremely sensitive to data distribution, even inapplicable to the real client Non-IID data. In this paper, we propose an effective communication strategy for federated learning called FedSAA, which increases the testing performance on Non-IID data by introducing self- attention mechanism. Two major innovations of our paper are presented here. Firstly, we utilize self-attention mechanism to optimize both the server-to-client and the client-to-client parameter divergence during the model aggregation process so as to improve the model robustness for Non-IID data. Secondly, we adopt the sign compression operator to help data transmission between nodes. The experimental results demonstrate that the model accuracy of our communication-efficient strategy for federated learning with Non-IID data is superior to other communication-efficient algorithms.
Access to the requested content is limited to institutions that have purchased or subscribe to SPIE eBooks.
You are receiving this notice because your organization may not have SPIE eBooks access.*
*Shibboleth/Open Athens users─please
sign in
to access your institution's subscriptions.
To obtain this item, you may purchase the complete book in print or electronic format on
SPIE.org.
INSTITUTIONAL Select your institution to access the SPIE Digital Library.
PERSONAL Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.