Federated Two-stage Learning with Sign-based Voting

no code implementations • 10 Dec 2021 • Zichen Ma, Zihan Lu, Yu Lu, Wenye Li, JinFeng Yi, Shuguang Cui

In this paper, we design a federated two-stage learning framework that augments prototypical federated learning with a cut layer on devices and uses sign-based stochastic gradient descent with the majority vote method on model updates.

BIG-bench Machine Learning Federated Learning +2

Towards Heterogeneous Clients with Elastic Federated Learning

no code implementations • 17 Jun 2021 • Zichen Ma, Yu Lu, Zihan Lu, Wenye Li, JinFeng Yi, Shuguang Cui

Training in heterogeneous and potentially massive networks introduces bias into the system, which is originated from the non-IID data and the low participation rate in reality.

Federated Learning