arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2110.03450cs.LG

Efficient and Private Federated Learning with Partially Trainable Networks

Google(谷歌)

Hakim Sidahmed, Zheng Xu, Ankush Garg, Yuan Cao, Mingqing Chen

英文摘要:

Federated learning is used for decentralized training of machine learning models on a large number (millions) of edge mobile devices. It is challenging because mobile devices often have limited communication bandwidth and local computation resources. Therefore, improving the efficiency of federated learning is critical for scalability and usability. In this paper, we propose to leverage partially trainable neural networks, which freeze a portion of the model parameters during the entire training process, to reduce the communication cost with little implications on model performance. Through extensive experiments, we empirically show that Federated learning of Partially Trainable neural networks (FedPT) can result in superior communication-accuracy trade-offs, with up to $46\times$ reduction in communication cost, at a small accuracy cost. Our approach also enables faster training, with a smaller memory footprint, and better utility for strong differential privacy guarantees. The proposed FedPT method can be particularly interesting for pushing the limitations of over-parameterization in on-device learning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑