Research Area:  Machine Learning

Abstract:

Statistical and systematic challenges in collaboratively training machine learning models across distributed networks of mobile devices have been the bottlenecks in the real-world application of federated learning. In this work, we show that meta-learning is a natural choice to handle these issues, and propose a federated meta-learning framework FedMeta, where a parameterized algorithm (or meta-learner) is shared, instead of a global model in previous approaches. We conduct an extensive empirical evaluation on LEAF datasets and a real-world production dataset, and demonstrate that FedMeta achieves a reduction in required communication cost by 2.82-4.33 times with faster convergence, and an increase in accuracy by 3.23%-14.84% as compared to Federated Averaging (FedAvg) which is a leading optimization algorithm in federated learning. Moreover, FedMeta preserves user privacy since only the parameterized algorithm is transmitted between mobile devices and central servers, and no raw data is collected onto the servers.

Keywords:  
Federated Meta-Learning
Machine Learning
Deep Learning

Author(s) Name:  Fei Chen, Mi Luo, Zhenhua Dong, Zhenguo Li, Xiuqiang He

Publisher name:  arXiv:1802.07876

DOI:  10.48550/arXiv.1802.07876

Volume Information:  

Paper Link:   https://arxiv.org/abs/1802.07876